package cn.iocoder.yudao.module.im.dal.mysql.message;
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
import cn.iocoder.yudao.module.im.dal.dataobject.message.ImPrivateMessageDO;
|
import cn.iocoder.yudao.module.im.enums.message.ImMessageStatusEnum;
|
import jakarta.annotation.Resource;
|
import org.junit.jupiter.api.Test;
|
|
import java.time.LocalDateTime;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
/**
|
* {@link ImPrivateMessageMapper} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
public class ImPrivateMessageMapperTest extends BaseDbUnitTest {
|
|
@Resource
|
private ImPrivateMessageMapper mapper;
|
|
private static final LocalDateTime FAR_PAST = LocalDateTime.of(2000, 1, 1, 0, 0, 0);
|
|
// ========== selectListByMinId ==========
|
|
@Test
|
public void testSelectListByMinId() {
|
// 准备：用户 1 发给用户 2
|
ImPrivateMessageDO msg1 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg1);
|
// 用户 2 发给用户 1
|
ImPrivateMessageDO msg2 = buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg2);
|
// 用户 3 发给用户 4（不相关）
|
ImPrivateMessageDO msg3 = buildMessage(3L, 4L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg3);
|
|
// 调用：用户 1 从 id=0 拉取
|
List<ImPrivateMessageDO> result = mapper.selectListByMinId(1L, 0L, FAR_PAST, 100);
|
|
// 断言：只包含与用户 1 相关的 2 条
|
assertEquals(2, result.size());
|
assertTrue(result.stream().allMatch(m ->
|
m.getSenderId().equals(1L) || m.getReceiverId().equals(1L)));
|
}
|
|
@Test
|
public void testSelectListByMinId_withMinId() {
|
// 准备
|
ImPrivateMessageDO msg1 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg1);
|
ImPrivateMessageDO msg2 = buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg2);
|
|
// 调用：从 msg1.id 之后拉取
|
List<ImPrivateMessageDO> result = mapper.selectListByMinId(1L, msg1.getId(), FAR_PAST, 100);
|
|
// 断言：只有 msg2
|
assertEquals(1, result.size());
|
assertEquals(msg2.getId(), result.get(0).getId());
|
}
|
|
@Test
|
public void testSelectListByMinId_limitSize() {
|
// 准备：插入 3 条
|
mapper.insert(buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL));
|
mapper.insert(buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL));
|
mapper.insert(buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL));
|
|
// 调用：limit 2
|
List<ImPrivateMessageDO> result = mapper.selectListByMinId(1L, 0L, FAR_PAST, 2);
|
|
// 断言
|
assertEquals(2, result.size());
|
}
|
|
@Test
|
public void testSelectListByMinId_sendTimeFilter() {
|
// 准备：一条落在窗口内，一条落在窗口外
|
ImPrivateMessageDO newMsg = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
newMsg.setSendTime(LocalDateTime.now().minusDays(1));
|
mapper.insert(newMsg);
|
ImPrivateMessageDO oldMsg = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
oldMsg.setSendTime(LocalDateTime.now().minusDays(40));
|
mapper.insert(oldMsg);
|
|
// 调用：窗口起点 = 30 天前
|
List<ImPrivateMessageDO> result = mapper.selectListByMinId(1L, 0L,
|
LocalDateTime.now().minusDays(30), 100);
|
|
// 断言：只返回窗口内的消息
|
assertEquals(1, result.size());
|
assertEquals(newMsg.getId(), result.get(0).getId());
|
}
|
|
// ========== selectHistoryList ==========
|
|
@Test
|
public void testSelectHistoryList_basic() {
|
// 准备：用户 1 <-> 用户 2 的消息
|
ImPrivateMessageDO msg1 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg1);
|
ImPrivateMessageDO msg2 = buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg2);
|
// 用户 1 <-> 用户 3 的消息（不相关）
|
ImPrivateMessageDO msg3 = buildMessage(1L, 3L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg3);
|
|
// 调用
|
List<ImPrivateMessageDO> result = mapper.selectHistoryList(1L, 2L, null, 100);
|
|
// 断言：只有与用户 2 对话的 2 条，按 id 倒序
|
assertEquals(2, result.size());
|
assertTrue(result.get(0).getId() > result.get(1).getId());
|
}
|
|
@Test
|
public void testSelectHistoryList_includeRecall() {
|
// 准备
|
ImPrivateMessageDO msg1 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg1);
|
ImPrivateMessageDO msg2 = buildMessage(1L, 2L, ImMessageStatusEnum.RECALL);
|
mapper.insert(msg2);
|
|
// 调用
|
List<ImPrivateMessageDO> result = mapper.selectHistoryList(1L, 2L, null, 100);
|
|
// 断言：撤回消息一并返回
|
assertEquals(2, result.size());
|
assertEquals(msg2.getId(), result.get(0).getId());
|
assertEquals(msg1.getId(), result.get(1).getId());
|
}
|
|
@Test
|
public void testSelectHistoryList_withMaxId() {
|
// 准备
|
ImPrivateMessageDO msg1 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg1);
|
ImPrivateMessageDO msg2 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg2);
|
ImPrivateMessageDO msg3 = buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL);
|
mapper.insert(msg3);
|
|
// 调用：从 msg3 往前拉
|
List<ImPrivateMessageDO> result = mapper.selectHistoryList(1L, 2L, msg3.getId(), 100);
|
|
// 断言：只有 msg1 和 msg2
|
assertEquals(2, result.size());
|
assertTrue(result.stream().allMatch(m -> m.getId() < msg3.getId()));
|
}
|
|
@Test
|
public void testSelectHistoryList_limit() {
|
// 准备
|
mapper.insert(buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL));
|
mapper.insert(buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL));
|
mapper.insert(buildMessage(1L, 2L, ImMessageStatusEnum.NORMAL));
|
|
// 调用：limit 2
|
List<ImPrivateMessageDO> result = mapper.selectHistoryList(1L, 2L, null, 2);
|
|
// 断言
|
assertEquals(2, result.size());
|
}
|
|
@Test
|
public void testSelectHistoryList_bidirectional() {
|
// 准备：验证双向查询
|
ImPrivateMessageDO msg1 = buildMessage(2L, 1L, ImMessageStatusEnum.NORMAL); // 对方发的
|
mapper.insert(msg1);
|
|
// 调用：以用户 1 的视角查
|
List<ImPrivateMessageDO> result = mapper.selectHistoryList(1L, 2L, null, 100);
|
|
// 断言：能查到对方发来的消息
|
assertEquals(1, result.size());
|
assertEquals(2L, result.get(0).getSenderId());
|
}
|
|
// ========== 工具方法 ==========
|
|
private ImPrivateMessageDO buildMessage(Long senderId, Long receiverId, ImMessageStatusEnum status) {
|
return ImPrivateMessageDO.builder()
|
.clientMessageId("uuid-" + System.nanoTime())
|
.senderId(senderId)
|
.receiverId(receiverId)
|
.type(0)
|
.content("{\"content\":\"test\"}")
|
.status(status.getStatus())
|
.sendTime(LocalDateTime.now())
|
.build();
|
}
|
|
}
|
