|
@@ -2,81 +2,74 @@ package cn.iocoder.yudao.module.im.service.message;
|
|
|
|
|
|
import org.junit.jupiter.api.Disabled;
|
|
|
import org.junit.jupiter.api.Test;
|
|
|
-import org.springframework.boot.test.mock.mockito.MockBean;
|
|
|
|
|
|
import jakarta.annotation.Resource;
|
|
|
|
|
|
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
|
|
|
|
|
|
import cn.iocoder.yudao.module.im.controller.admin.message.vo.*;
|
|
|
-import cn.iocoder.yudao.module.im.dal.dataobject.message.MessageDO;
|
|
|
-import cn.iocoder.yudao.module.im.dal.mysql.message.MessageMapper;
|
|
|
+import cn.iocoder.yudao.module.im.dal.dataobject.message.ImMessageDO;
|
|
|
+import cn.iocoder.yudao.module.im.dal.mysql.message.ImMessageMapper;
|
|
|
import cn.iocoder.yudao.framework.common.pojo.PageResult;
|
|
|
|
|
|
-import jakarta.annotation.Resource;
|
|
|
import org.springframework.context.annotation.Import;
|
|
|
-import java.util.*;
|
|
|
-import java.time.LocalDateTime;
|
|
|
|
|
|
-import static cn.hutool.core.util.RandomUtil.*;
|
|
|
import static cn.iocoder.yudao.module.im.enums.ErrorCodeConstants.*;
|
|
|
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.*;
|
|
|
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
|
|
|
import static cn.iocoder.yudao.framework.common.util.date.LocalDateTimeUtils.*;
|
|
|
import static cn.iocoder.yudao.framework.common.util.object.ObjectUtils.*;
|
|
|
-import static cn.iocoder.yudao.framework.common.util.date.DateUtils.*;
|
|
|
import static org.junit.jupiter.api.Assertions.*;
|
|
|
-import static org.mockito.Mockito.*;
|
|
|
|
|
|
/**
|
|
|
- * {@link MessageServiceImpl} 的单元测试类
|
|
|
+ * {@link ImMessageServiceImpl} 的单元测试类
|
|
|
*
|
|
|
* @author 芋道源码
|
|
|
*/
|
|
|
-@Import(MessageServiceImpl.class)
|
|
|
-public class MessageServiceImplTest extends BaseDbUnitTest {
|
|
|
+@Import(ImMessageServiceImpl.class)
|
|
|
+public class ImMessageServiceImplTest extends BaseDbUnitTest {
|
|
|
|
|
|
@Resource
|
|
|
- private MessageServiceImpl messageService;
|
|
|
+ private ImMessageServiceImpl messageService;
|
|
|
|
|
|
@Resource
|
|
|
- private MessageMapper messageMapper;
|
|
|
+ private ImMessageMapper imMessageMapper;
|
|
|
|
|
|
@Test
|
|
|
public void testCreateMessage_success() {
|
|
|
// 准备参数
|
|
|
- MessageSaveReqVO createReqVO = randomPojo(MessageSaveReqVO.class).setId(null);
|
|
|
+ ImMessageSaveReqVO createReqVO = randomPojo(ImMessageSaveReqVO.class).setId(null);
|
|
|
|
|
|
// 调用
|
|
|
Long messageId = messageService.createMessage(createReqVO);
|
|
|
// 断言
|
|
|
assertNotNull(messageId);
|
|
|
// 校验记录的属性是否正确
|
|
|
- MessageDO message = messageMapper.selectById(messageId);
|
|
|
+ ImMessageDO message = imMessageMapper.selectById(messageId);
|
|
|
assertPojoEquals(createReqVO, message, "id");
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testUpdateMessage_success() {
|
|
|
// mock 数据
|
|
|
- MessageDO dbMessage = randomPojo(MessageDO.class);
|
|
|
- messageMapper.insert(dbMessage);// @Sql: 先插入出一条存在的数据
|
|
|
+ ImMessageDO dbMessage = randomPojo(ImMessageDO.class);
|
|
|
+ imMessageMapper.insert(dbMessage);// @Sql: 先插入出一条存在的数据
|
|
|
// 准备参数
|
|
|
- MessageSaveReqVO updateReqVO = randomPojo(MessageSaveReqVO.class, o -> {
|
|
|
+ ImMessageSaveReqVO updateReqVO = randomPojo(ImMessageSaveReqVO.class, o -> {
|
|
|
o.setId(dbMessage.getId()); // 设置更新的 ID
|
|
|
});
|
|
|
|
|
|
// 调用
|
|
|
messageService.updateMessage(updateReqVO);
|
|
|
// 校验是否更新正确
|
|
|
- MessageDO message = messageMapper.selectById(updateReqVO.getId()); // 获取最新的
|
|
|
+ ImMessageDO message = imMessageMapper.selectById(updateReqVO.getId()); // 获取最新的
|
|
|
assertPojoEquals(updateReqVO, message);
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void testUpdateMessage_notExists() {
|
|
|
// 准备参数
|
|
|
- MessageSaveReqVO updateReqVO = randomPojo(MessageSaveReqVO.class);
|
|
|
+ ImMessageSaveReqVO updateReqVO = randomPojo(ImMessageSaveReqVO.class);
|
|
|
|
|
|
// 调用, 并断言异常
|
|
|
assertServiceException(() -> messageService.updateMessage(updateReqVO), MESSAGE_NOT_EXISTS);
|
|
@@ -85,15 +78,15 @@ public class MessageServiceImplTest extends BaseDbUnitTest {
|
|
|
@Test
|
|
|
public void testDeleteMessage_success() {
|
|
|
// mock 数据
|
|
|
- MessageDO dbMessage = randomPojo(MessageDO.class);
|
|
|
- messageMapper.insert(dbMessage);// @Sql: 先插入出一条存在的数据
|
|
|
+ ImMessageDO dbMessage = randomPojo(ImMessageDO.class);
|
|
|
+ imMessageMapper.insert(dbMessage);// @Sql: 先插入出一条存在的数据
|
|
|
// 准备参数
|
|
|
Long id = dbMessage.getId();
|
|
|
|
|
|
// 调用
|
|
|
messageService.deleteMessage(id);
|
|
|
// 校验数据不存在了
|
|
|
- assertNull(messageMapper.selectById(id));
|
|
|
+ assertNull(imMessageMapper.selectById(id));
|
|
|
}
|
|
|
|
|
|
@Test
|
|
@@ -109,7 +102,7 @@ public class MessageServiceImplTest extends BaseDbUnitTest {
|
|
|
@Disabled // TODO 请修改 null 为需要的值,然后删除 @Disabled 注解
|
|
|
public void testGetMessagePage() {
|
|
|
// mock 数据
|
|
|
- MessageDO dbMessage = randomPojo(MessageDO.class, o -> { // 等会查询到
|
|
|
+ ImMessageDO dbMessage = randomPojo(ImMessageDO.class, o -> { // 等会查询到
|
|
|
o.setClientMessageId(null);
|
|
|
o.setSenderId(null);
|
|
|
o.setReceiverId(null);
|
|
@@ -123,33 +116,33 @@ public class MessageServiceImplTest extends BaseDbUnitTest {
|
|
|
o.setSendFrom(null);
|
|
|
o.setCreateTime(null);
|
|
|
});
|
|
|
- messageMapper.insert(dbMessage);
|
|
|
+ imMessageMapper.insert(dbMessage);
|
|
|
// 测试 clientMessageId 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setClientMessageId(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setClientMessageId(null)));
|
|
|
// 测试 senderId 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderId(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderId(null)));
|
|
|
// 测试 receiverId 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setReceiverId(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setReceiverId(null)));
|
|
|
// 测试 senderNickname 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderNickname(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderNickname(null)));
|
|
|
// 测试 senderAvatar 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderAvatar(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSenderAvatar(null)));
|
|
|
// 测试 conversationType 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setConversationType(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setConversationType(null)));
|
|
|
// 测试 conversationNo 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setConversationNo(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setConversationNo(null)));
|
|
|
// 测试 contentType 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setContentType(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setContentType(null)));
|
|
|
// 测试 content 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setContent(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setContent(null)));
|
|
|
// 测试 sendTime 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSendTime(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSendTime(null)));
|
|
|
// 测试 sendFrom 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSendFrom(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setSendFrom(null)));
|
|
|
// 测试 createTime 不匹配
|
|
|
- messageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setCreateTime(null)));
|
|
|
+ imMessageMapper.insert(cloneIgnoreId(dbMessage, o -> o.setCreateTime(null)));
|
|
|
// 准备参数
|
|
|
- MessagePageReqVO reqVO = new MessagePageReqVO();
|
|
|
+ ImMessagePageReqVO reqVO = new ImMessagePageReqVO();
|
|
|
reqVO.setClientMessageId(null);
|
|
|
reqVO.setSenderId(null);
|
|
|
reqVO.setReceiverId(null);
|
|
@@ -164,7 +157,7 @@ public class MessageServiceImplTest extends BaseDbUnitTest {
|
|
|
reqVO.setCreateTime(buildBetweenTime(2023, 2, 1, 2023, 2, 28));
|
|
|
|
|
|
// 调用
|
|
|
- PageResult<MessageDO> pageResult = messageService.getMessagePage(reqVO);
|
|
|
+ PageResult<ImMessageDO> pageResult = messageService.getMessagePage(reqVO);
|
|
|
// 断言
|
|
|
assertEquals(1, pageResult.getTotal());
|
|
|
assertEquals(1, pageResult.getList().size());
|