7479521546
- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
60 lines
2.3 KiB
C#
60 lines
2.3 KiB
C#
using BuildFlow.Shared.Domain.Errors;
|
||
using BuildFlow.Shared.Domain.Exceptions;
|
||
using FluentAssertions;
|
||
using IM.Application.Chat.Commands;
|
||
using IM.Application.Chat.Handlers;
|
||
using IM.Domain.Enums;
|
||
using IM.Infrastructure.Persistence;
|
||
using Microsoft.EntityFrameworkCore;
|
||
using Xunit;
|
||
|
||
namespace BuildFlow.IM.Tests.Chat;
|
||
|
||
/// <summary>
|
||
/// SetConversationRemarkCommandHandler：成功置备注 / 空白清空备注 / 参与者不存在抛 NotFound。
|
||
/// </summary>
|
||
public class SetConversationRemarkCommandHandlerTests
|
||
{
|
||
private static readonly Guid UserId = Guid.Parse("22222222-2222-2222-2222-222222222222");
|
||
private static readonly Guid ConvId = Guid.Parse("11111111-1111-1111-1111-111111111111");
|
||
|
||
[Fact]
|
||
public async Task Sets_remark_for_participant()
|
||
{
|
||
await using var db = IMTestDb.Create();
|
||
db.Conversations.Add(ImTestData.BuildConversation(ConvId, ConversationTypeEnum.Private, (UserId, 0)));
|
||
await db.SaveChangesAsync();
|
||
|
||
var handler = new SetConversationRemarkCommandHandler(db);
|
||
await handler.Handle(new SetConversationRemarkCommand(ConvId, UserId, "老板"), CancellationToken.None);
|
||
|
||
var p = await db.ConversationParticipants.FirstAsync(x => x.ConversationId == ConvId && x.UserId == UserId);
|
||
p.Remark.Should().Be("老板");
|
||
}
|
||
|
||
[Fact]
|
||
public async Task Clears_remark_when_blank()
|
||
{
|
||
await using var db = IMTestDb.Create();
|
||
db.Conversations.Add(ImTestData.BuildConversation(ConvId, ConversationTypeEnum.Private, (UserId, 0)));
|
||
await db.SaveChangesAsync();
|
||
|
||
var handler = new SetConversationRemarkCommandHandler(db);
|
||
await handler.Handle(new SetConversationRemarkCommand(ConvId, UserId, ""), CancellationToken.None);
|
||
|
||
var p = await db.ConversationParticipants.FirstAsync(x => x.ConversationId == ConvId && x.UserId == UserId);
|
||
p.Remark.Should().BeNull();
|
||
}
|
||
|
||
[Fact]
|
||
public async Task Throws_not_found_when_participant_missing()
|
||
{
|
||
await using var db = IMTestDb.Create();
|
||
var handler = new SetConversationRemarkCommandHandler(db);
|
||
var act = async () => await handler.Handle(new SetConversationRemarkCommand(ConvId, UserId, "x"), CancellationToken.None);
|
||
|
||
var ex = await act.Should().ThrowAsync<NotFoundException>();
|
||
ex.Which.Code.Should().Be(ErrorCode.NotFound);
|
||
}
|
||
}
|