xiangning 7479521546 feat(rag): 实现RAG模块的Hybrid检索和向量索引优化
- 添加HNSW向量索引提升检索性能
- 实现Hybrid检索模式支持向量和关键词双路召回
- 采用RRF算法融合多路召回结果
- 支持ParentChild分块模式的上下文回填
- 优化向量写入采用批量SQL操作提升性能
- 添加RAG PostgreSQL连接字符串配置
- 更新部署脚本增加PGVector集成测试
- 修正表名引用从document_chunks到b_document_chunks
- 重构RagRetrievalService实现完整的检索流程
- 增加重排策略支持关键词和LLM重排
- 修改服务器部署地址配置
2026-08-11 10:59:09 +08:00

105 lines
3.3 KiB
C#

using BuildFlow.Shared.Domain.Errors;
using FluentAssertions;
using Microsoft.EntityFrameworkCore;
using Workflow.Application.BuildFlow.Imports.Handlers;
using Workflow.Application.BuildFlow.Imports.Queries;
using Workflow.Domain.Entities;
using Workflow.Domain.Enums;
using Xunit;
namespace BuildFlow.Workflow.Tests.BuildFlow.Imports;
public class ImportJobQueryHandlerTests
{
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;
}
private static BuildFlowImportJob SeedJob(WorkflowDbContext db, Guid formId, BuildFlowImportJobStatus status)
{
var job = new BuildFlowImportJob
{
Id = Guid.NewGuid(),
FormDefinitionId = formId,
FileUrl = "upload/test.xlsx",
Status = status
};
db.BuildFlowImportJobs.Add(job);
return job;
}
#region GetImportJobList
[Fact]
public async Task GetImportJobList_ReturnsAll()
{
await using var db = CreateDbContext();
var formId = Guid.NewGuid();
SeedJob(db, formId, BuildFlowImportJobStatus.Pending);
SeedJob(db, formId, BuildFlowImportJobStatus.Success);
await db.SaveChangesAsync();
var handler = new GetImportJobListQueryHandler(db);
var result = await handler.Handle(new GetImportJobListQuery(null, null, 1, 10), CancellationToken.None);
result.Items.Should().HaveCount(2);
result.Total.Should().Be(2);
}
[Fact]
public async Task GetImportJobList_FiltersByStatus()
{
await using var db = CreateDbContext();
var formId = Guid.NewGuid();
SeedJob(db, formId, BuildFlowImportJobStatus.Pending);
SeedJob(db, formId, BuildFlowImportJobStatus.Success);
await db.SaveChangesAsync();
var handler = new GetImportJobListQueryHandler(db);
var result = await handler.Handle(
new GetImportJobListQuery(null, (int)BuildFlowImportJobStatus.Pending, 1, 10), CancellationToken.None);
result.Items.Should().HaveCount(1);
result.Items[0].Status.Should().Be(BuildFlowImportJobStatus.Pending);
}
#endregion
#region GetImportJob
[Fact]
public async Task GetImportJob_ReturnsDto()
{
await using var db = CreateDbContext();
var job = SeedJob(db, Guid.NewGuid(), BuildFlowImportJobStatus.Running);
await db.SaveChangesAsync();
var handler = new GetImportJobQueryHandler(db);
var result = await handler.Handle(new GetImportJobQuery(job.Id), CancellationToken.None);
result.Should().NotBeNull();
result.Id.Should().Be(job.Id);
result.Status.Should().Be(BuildFlowImportJobStatus.Running);
}
[Fact]
public async Task GetImportJob_Missing_ThrowsNotFound()
{
await using var db = CreateDbContext();
var handler = new GetImportJobQueryHandler(db);
var act = async () => await handler.Handle(new GetImportJobQuery(Guid.NewGuid()), CancellationToken.None);
var ex = await act.Should().ThrowAsync<NotFoundException>();
ex.Which.Code.Should().Be(ErrorCode.NotFound);
}
#endregion
}