test: 添加 AI 智能体集成测试（28 个用例）
- PromptManager 7 个 Agent 的 System Prompt 验证 - DeepSeekClient 连通性测试 - 5 个 Agent 的端到端 SSE 对话测试: 记数据（血压/心率录入+Tool Calling） 药管家（用药查询） AI 问诊（症状追问） 默认 Agent（自我介绍） - 通过后端 API 真实调用 DeepSeek 验证 Tool Calling 逻辑
This commit is contained in:
MingNian
294
backend/tests/Health.Tests/ai_agent_tests.cs
Normal file
294
backend/tests/Health.Tests/ai_agent_tests.cs
Normal file
@@ -0,0 +1,294 @@
|
||||
using System.Net.Http.Headers;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using Health.Domain.Entities;
|
||||
using Health.Domain.Enums;
|
||||
using Health.Infrastructure.AI;
|
||||
using Health.Infrastructure.Data;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Microsoft.Extensions.Configuration;
|
||||
|
||||
namespace Health.Tests;
|
||||
|
||||
/// <summary>
|
||||
/// AI 智能体集成测试 — 模拟真实用户对话，验证 Tool Calling 与数据库写入
|
||||
/// 运行前需确保后端已启动: dotnet run --project src/Health.WebApi
|
||||
/// </summary>
|
||||
public class AiAgentTests
|
||||
{
|
||||
private static readonly HttpClient Http = new()
|
||||
{
|
||||
BaseAddress = new Uri("http://localhost:5000"),
|
||||
Timeout = TimeSpan.FromSeconds(120)
|
||||
};
|
||||
|
||||
private static readonly JsonSerializerOptions JsonOpts = new()
|
||||
{
|
||||
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
||||
PropertyNameCaseInsensitive = true
|
||||
};
|
||||
|
||||
// ==================== PromptManager 单元测试 ====================
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Default_Should_Contain_HeartKeywords()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Default);
|
||||
Assert.Contains("心脏", prompt);
|
||||
Assert.Contains("阿福", prompt);
|
||||
Assert.Contains("温暖", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Consultation_Should_Contain_TriageRules()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Consultation);
|
||||
Assert.Contains("剧烈胸痛", prompt);
|
||||
Assert.Contains("呼吸困难", prompt);
|
||||
Assert.Contains("160/100", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Health_Should_Contain_NormalRanges()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Health);
|
||||
Assert.Contains("139", prompt); // 收缩压上界
|
||||
Assert.Contains("89", prompt); // 舒张压下界
|
||||
Assert.Contains("100", prompt); // 心率上界
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Diet_Should_Contain_VlmKeywords()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Diet);
|
||||
Assert.Contains("VLM", prompt);
|
||||
Assert.Contains("能不能吃", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Medication_Should_Contain_ParseRules()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Medication);
|
||||
Assert.Contains("药名", prompt);
|
||||
Assert.Contains("剂量", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Report_Should_Contain_Disclaimer()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Report);
|
||||
Assert.Contains("AI预解读", prompt);
|
||||
Assert.Contains("医生确认", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_Exercise_Should_Contain_RehabKeywords()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
var prompt = pm.GetSystemPrompt(AgentType.Exercise);
|
||||
Assert.Contains("心脏康复", prompt);
|
||||
Assert.Contains("循序渐进", prompt);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void PromptManager_All_Agents_Should_Return_NonEmpty()
|
||||
{
|
||||
var pm = new PromptManager();
|
||||
foreach (AgentType agent in Enum.GetValues<AgentType>())
|
||||
{
|
||||
var prompt = pm.GetSystemPrompt(agent);
|
||||
Assert.False(string.IsNullOrWhiteSpace(prompt), $"{agent} 的 SystemPrompt 为空");
|
||||
}
|
||||
}
|
||||
|
||||
// ==================== DeepSeekClient 连通性测试 ====================
|
||||
|
||||
[Fact]
|
||||
public async Task DeepSeekClient_SimpleChat_Should_Return_Response()
|
||||
{
|
||||
var client = CreateDeepSeekClient();
|
||||
var messages = new List<ChatMessage>
|
||||
{
|
||||
new() { Role = "system", Content = "你是一个测试助手。请只回复 'OK'。" },
|
||||
new() { Role = "user", Content = "测试" }
|
||||
};
|
||||
|
||||
var response = await client.ChatAsync(messages);
|
||||
Assert.NotNull(response);
|
||||
Assert.NotEmpty(response.Choices);
|
||||
Assert.Contains("OK", response.Choices.First().Message?.Content ?? "");
|
||||
}
|
||||
|
||||
// ==================== AI 对话 + Tool Calling 集成测试 ====================
|
||||
|
||||
[Fact]
|
||||
public async Task HealthAgent_RecordBloodPressure_Should_SaveToDb()
|
||||
{
|
||||
// 先登录获取 token
|
||||
var token = await LoginAsync("13800000001");
|
||||
|
||||
// 发送对话消息触发 Tool Calling
|
||||
var events = await SendChatMessage(token, "health", "我刚刚测了血压，138/86");
|
||||
var toolResults = events.Where(e => e.Action == "tool_result").ToList();
|
||||
|
||||
Assert.NotEmpty(toolResults);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task HealthAgent_RecordHeartRate_Should_SaveToDb()
|
||||
{
|
||||
var token = await LoginAsync("13800000001");
|
||||
var events = await SendChatMessage(token, "health", "心率72");
|
||||
var toolResults = events.Where(e => e.Action == "tool_result").ToList();
|
||||
|
||||
Assert.NotEmpty(toolResults);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task MedicationAgent_Query_Should_Return_Medications()
|
||||
{
|
||||
var token = await LoginAsync("13800000001");
|
||||
var events = await SendChatMessage(token, "medication", "我现在在吃什么药？");
|
||||
|
||||
var answers = events.Where(e => e.Action == "answer")
|
||||
.Select(e => e.Data?.ToString() ?? "");
|
||||
|
||||
var fullResponse = string.Join("", answers);
|
||||
Assert.NotEmpty(fullResponse);
|
||||
// 应该提到阿司匹林或阿托伐他汀
|
||||
Assert.True(fullResponse.Contains("阿司匹林") || fullResponse.Contains("阿托伐他汀") ||
|
||||
fullResponse.Contains("Aspirin"));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task ConsultationAgent_SymptomCheck_Should_AskFollowUp()
|
||||
{
|
||||
var token = await LoginAsync("13800000001");
|
||||
var events = await SendChatMessage(token, "consultation", "最近胸口有点不舒服");
|
||||
|
||||
var answers = events.Where(e => e.Action == "answer")
|
||||
.Select(e => e.Data?.ToString() ?? "");
|
||||
|
||||
var fullResponse = string.Join("", answers);
|
||||
Assert.NotEmpty(fullResponse);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task DefaultAgent_GeneralQuestion_Should_Respond()
|
||||
{
|
||||
var token = await LoginAsync("13800000001");
|
||||
var events = await SendChatMessage(token, "default", "你好，介绍一下你自己");
|
||||
|
||||
var answers = events.Where(e => e.Action == "answer")
|
||||
.Select(e => e.Data?.ToString() ?? "");
|
||||
|
||||
var fullResponse = string.Join("", answers);
|
||||
Assert.NotEmpty(fullResponse);
|
||||
Assert.True(fullResponse.Contains("阿福") || fullResponse.Contains("健康"), "默认 Agent 自我介绍应包含名称");
|
||||
}
|
||||
|
||||
// ==================== 辅助方法 ====================
|
||||
|
||||
/// <summary>
|
||||
/// 发送验证码 + 登录，返回 accessToken
|
||||
/// </summary>
|
||||
private static async Task<string> LoginAsync(string phone)
|
||||
{
|
||||
// 发送验证码
|
||||
var smsPayload = JsonSerializer.Serialize(new { phone }, JsonOpts);
|
||||
var smsResp = await Http.PostAsync("/api/auth/send-sms",
|
||||
new StringContent(smsPayload, Encoding.UTF8, "application/json"));
|
||||
var smsJson = JsonDocument.Parse(await smsResp.Content.ReadAsStringAsync());
|
||||
var devCode = smsJson.RootElement.GetProperty("data").GetProperty("devCode").GetString()!;
|
||||
|
||||
// 登录
|
||||
var loginPayload = JsonSerializer.Serialize(new { phone, smsCode = devCode }, JsonOpts);
|
||||
var loginResp = await Http.PostAsync("/api/auth/login",
|
||||
new StringContent(loginPayload, Encoding.UTF8, "application/json"));
|
||||
var loginJson = JsonDocument.Parse(await loginResp.Content.ReadAsStringAsync());
|
||||
|
||||
return loginJson.RootElement.GetProperty("data").GetProperty("accessToken").GetString()!;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 向指定 Agent 发送消息，返回所有 SSE 事件
|
||||
/// </summary>
|
||||
private static async Task<List<SseEvent>> SendChatMessage(string token, string agentType, string message)
|
||||
{
|
||||
var url = $"/api/ai/{agentType}/chat?message={Uri.EscapeDataString(message)}&token={Uri.EscapeDataString(token)}";
|
||||
|
||||
Http.DefaultRequestHeaders.Authorization = null; // token 走 query string
|
||||
var response = await Http.GetAsync(url, HttpCompletionOption.ResponseHeadersRead);
|
||||
response.EnsureSuccessStatusCode();
|
||||
|
||||
var events = new List<SseEvent>();
|
||||
using var stream = await response.Content.ReadAsStreamAsync();
|
||||
using var reader = new StreamReader(stream);
|
||||
|
||||
string? line;
|
||||
while ((line = await reader.ReadLineAsync()) != null)
|
||||
{
|
||||
if (string.IsNullOrWhiteSpace(line)) continue;
|
||||
if (!line.StartsWith("data: ")) continue;
|
||||
|
||||
var data = line["data: ".Length..];
|
||||
if (data == "[DONE]") break;
|
||||
|
||||
try
|
||||
{
|
||||
var parsed = JsonSerializer.Deserialize<SseEvent>(data, JsonOpts);
|
||||
if (parsed != null) events.Add(parsed);
|
||||
}
|
||||
catch { /* 跳过无法解析的 chunk */ }
|
||||
}
|
||||
|
||||
return events;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 创建 DeepSeekClient（读取 .env 配置）
|
||||
/// </summary>
|
||||
private static DeepSeekClient CreateDeepSeekClient()
|
||||
{
|
||||
// 从测试输出目录向上 5 级找到 backend/.env
|
||||
// bin/Debug/net10.0 → Health.Tests → tests → backend
|
||||
var baseDir = AppContext.BaseDirectory;
|
||||
var envPath = Path.GetFullPath(Path.Combine(baseDir, "..", "..", "..", "..", "..", ".env"));
|
||||
|
||||
if (File.Exists(envPath))
|
||||
{
|
||||
foreach (var envLine in File.ReadAllLines(envPath))
|
||||
{
|
||||
var trimmed = envLine.Trim();
|
||||
if (string.IsNullOrEmpty(trimmed) || trimmed.StartsWith('#')) continue;
|
||||
var eqIdx = trimmed.IndexOf('=');
|
||||
if (eqIdx <= 0) continue;
|
||||
var key = trimmed[..eqIdx].Trim();
|
||||
var value = trimmed[(eqIdx + 1)..].Trim();
|
||||
Environment.SetEnvironmentVariable(key, value);
|
||||
}
|
||||
}
|
||||
|
||||
var config = new ConfigurationBuilder().AddEnvironmentVariables().Build();
|
||||
var httpClient = new HttpClient
|
||||
{
|
||||
BaseAddress = new Uri((config["DEEPSEEK_BASE_URL"] ?? "https://api.deepseek.com/v1").TrimEnd('/') + "/"),
|
||||
Timeout = TimeSpan.FromSeconds(120)
|
||||
};
|
||||
return new DeepSeekClient(httpClient, config);
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>SSE 事件模型</summary>
|
||||
public class SseEvent
|
||||
{
|
||||
public string? Action { get; set; }
|
||||
public object? Data { get; set; }
|
||||
public string? Message { get; set; }
|
||||
}
|
||||
Reference in New Issue
Block a user