- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
58 lines
1.9 KiB
C#
58 lines
1.9 KiB
C#
using BuildFlow.Shared.Domain.Errors;
|
|
using FluentAssertions;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using Workflow.Application.Features.WorkflowDefinitions.Commands;
|
|
using Workflow.Application.Features.WorkflowDefinitions.Handlers;
|
|
using Workflow.Domain.Entities;
|
|
using Workflow.Domain.Enums;
|
|
using Xunit;
|
|
|
|
namespace BuildFlow.Workflow.Tests.Handlers;
|
|
|
|
public class DisableWorkflowDefinitionHandlerTests
|
|
{
|
|
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 DisableWorkflowDefinition_SetsIsEnabledFalse()
|
|
{
|
|
await using var db = CreateDbContext();
|
|
var id = Guid.NewGuid();
|
|
db.WorkflowDefinitions.Add(new WorkflowDefinition
|
|
{
|
|
Id = id,
|
|
Name = "Def",
|
|
Code = "def",
|
|
Status = DefinitionStatus.Published,
|
|
Version = 1
|
|
});
|
|
await db.SaveChangesAsync();
|
|
|
|
var handler = new DisableWorkflowDefinitionCommandHandler(db);
|
|
await handler.Handle(new DisableWorkflowDefinitionCommand(id), CancellationToken.None);
|
|
|
|
var stored = await db.WorkflowDefinitions.FindAsync(id);
|
|
stored!.IsEnabled.Should().BeFalse();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task DisableWorkflowDefinition_Missing_ThrowsNotFound()
|
|
{
|
|
await using var db = CreateDbContext();
|
|
var handler = new DisableWorkflowDefinitionCommandHandler(db);
|
|
|
|
var act = async () => await handler.Handle(new DisableWorkflowDefinitionCommand(Guid.NewGuid()), CancellationToken.None);
|
|
|
|
var ex = await act.Should().ThrowAsync<NotFoundException>();
|
|
ex.Which.Code.Should().Be(ErrorCode.NotFound);
|
|
}
|
|
}
|