7479521546
- 添加HNSW向量索引提升检索性能 - 实现Hybrid检索模式支持向量和关键词双路召回 - 采用RRF算法融合多路召回结果 - 支持ParentChild分块模式的上下文回填 - 优化向量写入采用批量SQL操作提升性能 - 添加RAG PostgreSQL连接字符串配置 - 更新部署脚本增加PGVector集成测试 - 修正表名引用从document_chunks到b_document_chunks - 重构RagRetrievalService实现完整的检索流程 - 增加重排策略支持关键词和LLM重排 - 修改服务器部署地址配置
54 lines
2.2 KiB
C#
54 lines
2.2 KiB
C#
using BuildFlow.Platform.Application.Users.DTOs;
|
||
using BuildFlow.Platform.Application.Users.Queries;
|
||
using BuildFlow.Shared.Domain.Errors;
|
||
using BuildFlow.Shared.Domain.Exceptions;
|
||
using FluentAssertions;
|
||
using IM.Application.Chat.Handlers;
|
||
using IM.Application.Chat.Queries;
|
||
using IM.Domain.Enums;
|
||
using IM.Infrastructure.Persistence;
|
||
using MediatR;
|
||
using Moq;
|
||
using Xunit;
|
||
|
||
namespace BuildFlow.IM.Tests.Chat;
|
||
|
||
/// <summary>
|
||
/// GetGroupMembersQueryHandler：非群成员拒绝（Forbidden）；群成员返回成员列表含角色。
|
||
/// </summary>
|
||
public class GetGroupMembersQueryHandlerTests
|
||
{
|
||
private static readonly Guid OwnerId = Guid.Parse("22222222-2222-2222-2222-222222222222");
|
||
private static readonly Guid MemberId = Guid.Parse("33333333-3333-3333-3333-333333333333");
|
||
private const string Name = "group-1";
|
||
|
||
[Fact]
|
||
public async Task Throws_forbidden_when_not_member()
|
||
{
|
||
await using var db = IMTestDb.Create();
|
||
var (_, groupId) = await IMTestDb.SeedGroupAsync(db, Name, OwnerId, MemberId);
|
||
var handler = new GetGroupMembersQueryHandler(db, new Mock<IMediator>().Object);
|
||
var act = async () => await handler.Handle(new GetGroupMembersQuery(Guid.NewGuid(), groupId), CancellationToken.None);
|
||
|
||
var ex = await act.Should().ThrowAsync<UnauthorizedException>();
|
||
ex.Which.Code.Should().Be(ErrorCode.Forbidden);
|
||
}
|
||
|
||
[Fact]
|
||
public async Task Returns_members_with_roles()
|
||
{
|
||
await using var db = IMTestDb.Create();
|
||
var (_, groupId) = await IMTestDb.SeedGroupAsync(db, Name, OwnerId, MemberId);
|
||
var mediator = new Mock<IMediator>();
|
||
mediator.Setup(m => m.Send(It.IsAny<GetUserSummariesQuery>(), It.IsAny<CancellationToken>()))
|
||
.ReturnsAsync(new Dictionary<Guid, UserSummaryDto>());
|
||
var handler = new GetGroupMembersQueryHandler(db, mediator.Object);
|
||
|
||
var result = await handler.Handle(new GetGroupMembersQuery(OwnerId, groupId), CancellationToken.None);
|
||
|
||
result.Should().HaveCount(2);
|
||
result.Should().Contain(m => m.UserId == OwnerId && m.Role == GroupMemberRoleEnum.Owner);
|
||
result.Should().Contain(m => m.UserId == MemberId && m.Role == GroupMemberRoleEnum.Member);
|
||
}
|
||
}
|