7479521546
- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
68 lines
2.2 KiB
C#
68 lines
2.2 KiB
C#
using FluentAssertions;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Workflow.Application.Features.WorkflowInstances.DTOs;
|
|
using Workflow.Application.Features.WorkflowInstances.Handlers;
|
|
using Workflow.Application.Features.WorkflowInstances.Queries;
|
|
using Workflow.Domain.Entities;
|
|
using Workflow.Domain.Enums;
|
|
using Xunit;
|
|
|
|
using TaskStatus = global::Workflow.Domain.Enums.TaskStatus;
|
|
|
|
namespace BuildFlow.Workflow.Tests.Handlers;
|
|
|
|
public class MonitorWorkflowInstancesQueryHandlerTests
|
|
{
|
|
private static WorkflowDbContext CreateDbContext()
|
|
{
|
|
var options = new DbContextOptionsBuilder<WorkflowDbContext>()
|
|
.UseInMemoryDatabase(Guid.NewGuid().ToString())
|
|
.Options;
|
|
var db = new WorkflowDbContext(options);
|
|
db.SavingChanges += AuditStub.OnSave;
|
|
return db;
|
|
}
|
|
|
|
[Fact]
|
|
public async Task MonitorWorkflowInstances_AggregatesCounts()
|
|
{
|
|
await using var db = CreateDbContext();
|
|
db.WorkflowInstances.Add(new WorkflowInstance
|
|
{
|
|
Id = Guid.NewGuid(),
|
|
DefinitionId = Guid.NewGuid(),
|
|
Title = "Running",
|
|
Status = InstanceStatus.Running,
|
|
InitiatorId = Guid.NewGuid()
|
|
});
|
|
db.WorkflowInstances.Add(new WorkflowInstance
|
|
{
|
|
Id = Guid.NewGuid(),
|
|
DefinitionId = Guid.NewGuid(),
|
|
Title = "Completed",
|
|
Status = InstanceStatus.Completed,
|
|
InitiatorId = Guid.NewGuid()
|
|
});
|
|
db.WorkflowTasks.Add(new WorkflowTask
|
|
{
|
|
Id = Guid.NewGuid(),
|
|
InstanceId = Guid.NewGuid(),
|
|
TokenId = Guid.NewGuid(),
|
|
NodeId = Guid.NewGuid(),
|
|
Title = "Task",
|
|
AssigneeId = Guid.NewGuid(),
|
|
Status = TaskStatus.Pending
|
|
});
|
|
await db.SaveChangesAsync();
|
|
|
|
var handler = new MonitorWorkflowInstancesQueryHandler(db);
|
|
WorkflowMonitorDto result = await handler.Handle(new MonitorWorkflowInstancesQuery(), CancellationToken.None);
|
|
|
|
result.TotalInstances.Should().Be(2);
|
|
result.RunningInstances.Should().Be(1);
|
|
result.CompletedInstances.Should().Be(1);
|
|
result.PendingTasks.Should().Be(1);
|
|
result.OverdueTasks.Should().Be(0);
|
|
}
|
|
}
|