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.ImGroupMessageDO;
|
import cn.iocoder.yudao.module.im.enums.message.ImMessageReceiptStatusEnum;
|
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 ImGroupMessageMapper} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
public class ImGroupMessageMapperTest extends BaseDbUnitTest {
|
|
@Resource
|
private ImGroupMessageMapper mapper;
|
|
private static final LocalDateTime FAR_PAST = LocalDateTime.of(2000, 1, 1, 0, 0, 0);
|
|
// ========== selectListByMinId ==========
|
|
@Test
|
public void testSelectListByMinId_filterByReceiverSnapshot() {
|
// 准备：群 10 三条消息，快照分别命中 / 不命中用户 1
|
ImGroupMessageDO visible = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(visible);
|
ImGroupMessageDO invisible = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(2L, 3L));
|
mapper.insert(invisible);
|
ImGroupMessageDO selfSend = buildMessage(10L, 1L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(selfSend);
|
|
// 调用：用户 1 拉取群 10
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L, FAR_PAST, 100);
|
|
// 断言：仅返回快照包含用户 1 的消息
|
assertEquals(2, result.size());
|
assertEquals(visible.getId(), result.get(0).getId());
|
assertEquals(selfSend.getId(), result.get(1).getId());
|
}
|
|
@Test
|
public void testSelectListByMinId_excludeNullAndEmptySnapshot() {
|
// 准备：快照为 null / 空字符串的老数据，不应被任何用户拉到
|
ImGroupMessageDO nullSnapshot = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, null);
|
mapper.insert(nullSnapshot);
|
ImGroupMessageDO emptySnapshot = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of());
|
mapper.insert(emptySnapshot);
|
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L, FAR_PAST, 100);
|
|
assertTrue(result.isEmpty());
|
}
|
|
@Test
|
public void testSelectListByMinId_onlyReturnsGroupsInIdList() {
|
// 准备：群 10 + 群 20 各一条，用户 1 都在快照内
|
ImGroupMessageDO msg10 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(msg10);
|
ImGroupMessageDO msg20 = buildMessage(20L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(msg20);
|
|
// 调用：候选群只含群 10
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L, FAR_PAST, 100);
|
|
assertEquals(1, result.size());
|
assertEquals(msg10.getId(), result.get(0).getId());
|
}
|
|
@Test
|
public void testSelectListByMinId_includesRecall() {
|
// 准备：撤回消息也要返回，由客户端按 status 渲染「此消息已撤回」占位
|
ImGroupMessageDO normal = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(normal);
|
ImGroupMessageDO recalled = buildMessage(10L, 2L, ImMessageStatusEnum.RECALL, List.of(1L));
|
mapper.insert(recalled);
|
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L, FAR_PAST, 100);
|
|
assertEquals(2, result.size());
|
}
|
|
@Test
|
public void testSelectListByMinId_sendTimeWindow() {
|
// 准备：一条在窗口内，一条在窗口外
|
ImGroupMessageDO inWindow = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
inWindow.setSendTime(LocalDateTime.now().minusDays(1));
|
mapper.insert(inWindow);
|
ImGroupMessageDO outWindow = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
outWindow.setSendTime(LocalDateTime.now().minusDays(40));
|
mapper.insert(outWindow);
|
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L,
|
LocalDateTime.now().minusDays(30), 100);
|
|
assertEquals(1, result.size());
|
assertEquals(inWindow.getId(), result.get(0).getId());
|
}
|
|
@Test
|
public void testSelectListByMinId_sortAscLimit() {
|
// 准备：插入 3 条可见消息
|
ImGroupMessageDO m1 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m1);
|
ImGroupMessageDO m2 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m2);
|
ImGroupMessageDO m3 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m3);
|
|
// 调用：size=2，返回 id 最小的 2 条
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), 0L, FAR_PAST, 2);
|
|
assertEquals(2, result.size());
|
assertTrue(result.get(0).getId() < result.get(1).getId());
|
assertEquals(m1.getId(), result.get(0).getId());
|
assertEquals(m2.getId(), result.get(1).getId());
|
}
|
|
@Test
|
public void testSelectListByMinId_minIdExclusive() {
|
// 准备：游标之前的消息不返回
|
ImGroupMessageDO m1 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m1);
|
ImGroupMessageDO m2 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m2);
|
|
List<ImGroupMessageDO> result = mapper.selectListByMinId(1L, List.of(10L), m1.getId(), FAR_PAST, 100);
|
|
assertEquals(1, result.size());
|
assertEquals(m2.getId(), result.get(0).getId());
|
}
|
|
// ========== selectHistoryListByUser ==========
|
|
@Test
|
public void testSelectHistoryListByUser_filterByReceiverSnapshot() {
|
// 准备：群 10 三条，用户 1 在前两条快照内、第三条定向给别人
|
ImGroupMessageDO m1 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(m1);
|
ImGroupMessageDO m2 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L, 2L));
|
mapper.insert(m2);
|
ImGroupMessageDO directedOther = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(2L, 3L));
|
mapper.insert(directedOther);
|
// 别的群
|
ImGroupMessageDO other = buildMessage(20L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(other);
|
|
List<ImGroupMessageDO> result = mapper.selectHistoryListByUser(1L, 10L, null, 100);
|
|
// 断言：只返回群 10 中用户 1 可见的，按 id 倒序
|
assertEquals(2, result.size());
|
assertEquals(m2.getId(), result.get(0).getId());
|
assertEquals(m1.getId(), result.get(1).getId());
|
}
|
|
@Test
|
public void testSelectHistoryListByUser_maxIdCursor() {
|
ImGroupMessageDO m1 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m1);
|
ImGroupMessageDO m2 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m2);
|
ImGroupMessageDO m3 = buildMessage(10L, 2L, ImMessageStatusEnum.NORMAL, List.of(1L));
|
mapper.insert(m3);
|
|
List<ImGroupMessageDO> result = mapper.selectHistoryListByUser(1L, 10L, m3.getId(), 100);
|
|
// 断言：只返回 id < m3.id 的 m1、m2
|
assertEquals(2, result.size());
|
assertTrue(result.stream().allMatch(m -> m.getId() < m3.getId()));
|
}
|
|
// ========== 工具方法 ==========
|
|
private ImGroupMessageDO buildMessage(Long groupId, Long senderId, ImMessageStatusEnum status,
|
List<Long> receiverUserIds) {
|
return ImGroupMessageDO.builder()
|
.clientMessageId("uuid-" + System.nanoTime())
|
.senderId(senderId)
|
.groupId(groupId)
|
.type(0)
|
.content("{\"content\":\"test\"}")
|
.status(status.getStatus())
|
.sendTime(LocalDateTime.now())
|
.receiverUserIds(receiverUserIds)
|
.receiptStatus(ImMessageReceiptStatusEnum.NO_RECEIPT.getStatus())
|
.build();
|
}
|
|
}
|
