- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
105 lines
3.3 KiB
C#
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
|
|
}
|