7479521546
- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
69 lines
2.4 KiB
C#
69 lines
2.4 KiB
C#
using BuildFlow.Shared.Domain.Errors;
|
|
using FluentAssertions;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using NSubstitute;
|
|
using Workflow.Application.BuildFlow.Common.Services;
|
|
using Workflow.Application.BuildFlow.Exports.Handlers;
|
|
using Workflow.Application.BuildFlow.Exports.Queries;
|
|
using Workflow.Application.BuildFlow.Imports.DTOs;
|
|
using Workflow.Domain.Entities;
|
|
using Workflow.Domain.Enums;
|
|
using Xunit;
|
|
|
|
namespace BuildFlow.Workflow.Tests.BuildFlow.Exports;
|
|
|
|
public class DownloadImportTemplateQueryHandlerTests
|
|
{
|
|
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 DownloadImportTemplate_BuildsExcelViaService()
|
|
{
|
|
await using var db = CreateDbContext();
|
|
var formId = Guid.NewGuid();
|
|
db.FormDefinitions.Add(new FormDefinition
|
|
{
|
|
Id = formId,
|
|
Name = "报销单",
|
|
Code = "expense",
|
|
Version = 1,
|
|
Status = FormStatus.Published,
|
|
SchemaJson = "{\"type\":\"object\"}"
|
|
});
|
|
await db.SaveChangesAsync();
|
|
|
|
var excel = Substitute.For<IFormSchemaExcelService>();
|
|
excel.BuildTemplate(Arg.Any<string>(), Arg.Any<string>()).Returns(new byte[] { 1, 2, 3 });
|
|
|
|
var handler = new DownloadImportTemplateQueryHandler(db, excel);
|
|
var result = await handler.Handle(new DownloadImportTemplateQuery(formId), CancellationToken.None);
|
|
|
|
result.Should().NotBeNull();
|
|
result.Content.Should().NotBeNull();
|
|
result.Content.Length.Should().Be(3);
|
|
result.FileName.Should().EndWith("导入模版.xlsx");
|
|
excel.Received(1).BuildTemplate("{\"type\":\"object\"}", "报销单");
|
|
}
|
|
|
|
[Fact]
|
|
public async Task DownloadImportTemplate_MissingForm_ThrowsNotFound()
|
|
{
|
|
await using var db = CreateDbContext();
|
|
var excel = Substitute.For<IFormSchemaExcelService>();
|
|
var handler = new DownloadImportTemplateQueryHandler(db, excel);
|
|
|
|
var act = async () => await handler.Handle(new DownloadImportTemplateQuery(Guid.NewGuid()), CancellationToken.None);
|
|
|
|
var ex = await act.Should().ThrowAsync<NotFoundException>();
|
|
ex.Which.Code.Should().Be(ErrorCode.NotFound);
|
|
}
|
|
}
|