后端：
新增好友请求事件和好友已添加事件
This commit is contained in:
@@ -1,25 +1,30 @@
|
||||
using Xunit;
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Moq;
|
||||
using AutoMapper;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
using System;
|
||||
using IM_API.Services;
|
||||
using IM_API.Models;
|
||||
using AutoMapper;
|
||||
using IM_API.Domain.Events;
|
||||
using IM_API.Dtos;
|
||||
using IM_API.Exceptions;
|
||||
using IM_API.Tools;
|
||||
using IM_API.Models;
|
||||
using IM_API.Services;
|
||||
using MassTransit; // 必须引入
|
||||
using Microsoft.EntityFrameworkCore;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Moq;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using Xunit;
|
||||
|
||||
public class FriendServiceTests
|
||||
{
|
||||
private readonly Mock<IPublishEndpoint> _mockEndpoint = new();
|
||||
private readonly Mock<ILogger<FriendService>> _mockLogger = new();
|
||||
|
||||
#region 辅助构造方法
|
||||
private ImContext CreateDbContext()
|
||||
{
|
||||
var options = new DbContextOptionsBuilder<ImContext>()
|
||||
.UseInMemoryDatabase(Guid.NewGuid().ToString())
|
||||
.UseInMemoryDatabase(Guid.NewGuid().ToString()) // 确保每个测试数据库隔离
|
||||
.Options;
|
||||
|
||||
return new ImContext(options);
|
||||
}
|
||||
|
||||
@@ -27,132 +32,115 @@ public class FriendServiceTests
|
||||
{
|
||||
var config = new MapperConfiguration(cfg =>
|
||||
{
|
||||
// 补充你业务中实际需要的映射规则
|
||||
cfg.CreateMap<Friend, FriendInfoDto>();
|
||||
cfg.CreateMap<FriendRequestDto, FriendRequest>();
|
||||
cfg.CreateMap<HandleFriendRequestDto, FriendRequest>();
|
||||
cfg.CreateMap<FriendRequest, Friend>()
|
||||
.ForMember(dest => dest.UserId, opt => opt.MapFrom(src => src.ResponseUser))
|
||||
.ForMember(dest => dest.FriendId, opt => opt.MapFrom(src => src.RequestUser))
|
||||
.ForMember(dest => dest.RemarkName, opt => opt.MapFrom(src => "AutoAdded"));
|
||||
.ForMember(d => d.UserId, o => o.MapFrom(s => s.ResponseUser))
|
||||
.ForMember(d => d.FriendId, o => o.MapFrom(s => s.RequestUser));
|
||||
});
|
||||
|
||||
return config.CreateMapper();
|
||||
}
|
||||
|
||||
private FriendService CreateService(ImContext context)
|
||||
{
|
||||
var logger = new Mock<ILogger<FriendService>>();
|
||||
return new FriendService(context, logger.Object, CreateMapper());
|
||||
// 注入 Mock 对象和真实的 Mapper/Context
|
||||
return new FriendService(context, _mockLogger.Object, CreateMapper(), _mockEndpoint.Object);
|
||||
}
|
||||
#endregion
|
||||
|
||||
// --------------------------- 测试 BlockFriendAsync ---------------------------
|
||||
[Fact]
|
||||
public async Task BlockFriendAsync_Should_Set_Status_To_Blocked()
|
||||
public async Task SendFriendRequestAsync_Success_ShouldSaveAndPublish()
|
||||
{
|
||||
// Arrange
|
||||
var context = CreateDbContext();
|
||||
context.Friends.Add(new Friend
|
||||
{
|
||||
Id = 1,
|
||||
UserId = 10,
|
||||
FriendId = 20,
|
||||
StatusEnum = FriendStatus.Added,
|
||||
RemarkName = "test remark"
|
||||
});
|
||||
context.Users.AddRange(
|
||||
new User { Id = 1, Username = "Sender", Password = "..." },
|
||||
new User { Id = 2, Username = "Receiver", Password = "..." }
|
||||
);
|
||||
await context.SaveChangesAsync();
|
||||
|
||||
var service = CreateService(context);
|
||||
var dto = new FriendRequestDto { ToUserId = 2, Description = "Hello" };
|
||||
|
||||
var result = await service.BlockeFriendAsync(1);
|
||||
|
||||
Assert.True(result);
|
||||
Assert.Equal(FriendStatus.Blocked, context.Friends.Find(1).StatusEnum);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task BlockFriendAsync_Should_Throw_When_NotFound()
|
||||
{
|
||||
var service = CreateService(CreateDbContext());
|
||||
|
||||
await Assert.ThrowsAsync<BaseException>(() => service.BlockeFriendAsync(99));
|
||||
}
|
||||
|
||||
// --------------------------- 删除好友关系 ---------------------------
|
||||
[Fact]
|
||||
public async Task DeleteFriendAsync_Should_Remove_Friend()
|
||||
{
|
||||
var context = CreateDbContext();
|
||||
context.Friends.Add(new Friend
|
||||
{
|
||||
Id = 2,
|
||||
UserId = 1,
|
||||
FriendId = 3,
|
||||
RemarkName = "remark",
|
||||
StatusEnum = FriendStatus.Added
|
||||
});
|
||||
await context.SaveChangesAsync();
|
||||
|
||||
var service = CreateService(context);
|
||||
|
||||
var result = await service.DeleteFriendAsync(2);
|
||||
|
||||
Assert.True(result);
|
||||
Assert.Empty(context.Friends);
|
||||
}
|
||||
|
||||
// --------------------------- 获取好友列表 ---------------------------
|
||||
[Fact]
|
||||
public async Task GetFriendListAsync_Should_Return_Only_Added_Friends()
|
||||
{
|
||||
var context = CreateDbContext();
|
||||
context.Friends.AddRange(new List<Friend>
|
||||
{
|
||||
new Friend{ UserId = 1, FriendId = 2, RemarkName ="a1", StatusEnum = FriendStatus.Added },
|
||||
new Friend{ UserId = 1, FriendId = 3, RemarkName ="a2", StatusEnum = FriendStatus.Blocked }
|
||||
});
|
||||
await context.SaveChangesAsync();
|
||||
|
||||
var service = CreateService(context);
|
||||
|
||||
var result = await service.GetFriendListAsync(1, 1, 10, false);
|
||||
|
||||
Assert.Single(result);
|
||||
}
|
||||
|
||||
// --------------------------- 发起好友请求 ---------------------------
|
||||
[Fact]
|
||||
public async Task SendFriendRequestAsync_Should_Succeed()
|
||||
{
|
||||
var context = CreateDbContext();
|
||||
context.Users.Add(new User { Id = 10, Username = "A", Password = "123" });
|
||||
context.Users.Add(new User { Id = 20, Username = "B", Password = "123" });
|
||||
await context.SaveChangesAsync();
|
||||
|
||||
var service = CreateService(context);
|
||||
|
||||
var result = await service.SendFriendRequestAsync(new FriendRequestDto
|
||||
{
|
||||
FromUserId = 10,
|
||||
ToUserId = 20
|
||||
});
|
||||
// Act
|
||||
var result = await service.SendFriendRequestAsync(dto);
|
||||
|
||||
// Assert
|
||||
Assert.True(result);
|
||||
Assert.Single(context.FriendRequests);
|
||||
Assert.Single(context.Friends);
|
||||
|
||||
// 验证事件是否发布到了 MQ
|
||||
_mockEndpoint.Verify(x => x.Publish(
|
||||
It.Is<RequestFriendEvent>(e => e.FromUserId == 1 && e.ToUserId == 2),
|
||||
It.IsAny<CancellationToken>()),
|
||||
Times.Once);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task SendFriendRequestAsync_Should_Throw_When_User_NotFound()
|
||||
public async Task SendFriendRequestAsync_UserNotFound_ShouldThrow()
|
||||
{
|
||||
// Arrange
|
||||
var context = CreateDbContext();
|
||||
context.Users.Add(new User { Id = 10, Username = "A", Password = "123" });
|
||||
await context.SaveChangesAsync();
|
||||
var service = CreateService(context);
|
||||
var dto = new FriendRequestDto { ToUserId = 99 }; // 不存在的用户
|
||||
|
||||
// Act & Assert
|
||||
await Assert.ThrowsAsync<BaseException>(() => service.SendFriendRequestAsync(dto));
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task SendFriendRequestAsync_AlreadyExists_ShouldThrow()
|
||||
{
|
||||
// Arrange
|
||||
var context = CreateDbContext();
|
||||
context.Users.Add(new User { Id = 2 });
|
||||
context.FriendRequests.Add(new FriendRequest
|
||||
{
|
||||
RequestUser = 1,
|
||||
ResponseUser = 2,
|
||||
State = (sbyte)FriendRequestState.Pending
|
||||
});
|
||||
await context.SaveChangesAsync();
|
||||
var service = CreateService(context);
|
||||
|
||||
await Assert.ThrowsAsync<BaseException>(() => service.SendFriendRequestAsync(new FriendRequestDto
|
||||
{
|
||||
FromUserId = 10,
|
||||
ToUserId = 99
|
||||
}));
|
||||
// Act & Assert
|
||||
await Assert.ThrowsAsync<BaseException>(() => service.SendFriendRequestAsync(new FriendRequestDto { ToUserId = 2 }));
|
||||
}
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task BlockFriendAsync_ValidId_ShouldUpdateStatus()
|
||||
{
|
||||
// Arrange
|
||||
var context = CreateDbContext();
|
||||
var friend = new Friend { Id = 50, UserId = 1, FriendId = 2, StatusEnum = FriendStatus.Added };
|
||||
context.Friends.Add(friend);
|
||||
await context.SaveChangesAsync();
|
||||
var service = CreateService(context);
|
||||
|
||||
// Act
|
||||
await service.BlockeFriendAsync(50);
|
||||
|
||||
// Assert
|
||||
var updated = await context.Friends.FindAsync(50);
|
||||
Assert.Equal(FriendStatus.Blocked, updated.StatusEnum);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task GetFriendListAsync_ShouldFilterByStatus()
|
||||
{
|
||||
// Arrange
|
||||
var context = CreateDbContext();
|
||||
context.Friends.AddRange(
|
||||
new Friend { UserId = 1, FriendId = 2, StatusEnum = FriendStatus.Added },
|
||||
new Friend { UserId = 1, FriendId = 3, StatusEnum = FriendStatus.Blocked }
|
||||
);
|
||||
await context.SaveChangesAsync();
|
||||
var service = CreateService(context);
|
||||
|
||||
// Act
|
||||
var result = await service.GetFriendListAsync(1, 1, 10, false);
|
||||
|
||||
// Assert
|
||||
Assert.Single(result); // 只应该拿到 Added 状态的
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user