package cn.iocoder.yudao.module.im.util;
|
|
import cn.iocoder.yudao.framework.common.util.json.JsonUtils;
|
import cn.iocoder.yudao.module.im.dal.dataobject.message.content.QuoteMessage;
|
import org.junit.jupiter.api.Test;
|
|
import java.util.Map;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertFalse;
|
import static org.junit.jupiter.api.Assertions.assertNull;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
/**
|
* {@link ImMessageUtils} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
public class ImMessageUtilsTest {
|
|
// ========== parseQuoteMessageId ==========
|
|
@Test
|
public void testParseQuoteMessageId_valid() {
|
// 准备：content 含合法 quote 字段
|
String content = "{\"text\":\"你好\",\"quote\":{\"messageId\":1001,\"senderId\":2,\"type\":1}}";
|
|
// 调用 + 断言
|
assertEquals(1001L, ImMessageUtils.parseQuoteMessageId(content));
|
}
|
|
@Test
|
public void testParseQuoteMessageId_invalidJson() {
|
// 准备：content 非合法 JSON
|
// 调用 + 断言：解析失败返回 null
|
assertNull(ImMessageUtils.parseQuoteMessageId("not a json"));
|
}
|
|
@Test
|
public void testParseQuoteMessageId_noQuoteField() {
|
// 准备：content 无 quote 字段
|
String content = "{\"text\":\"你好\"}";
|
|
// 调用 + 断言
|
assertNull(ImMessageUtils.parseQuoteMessageId(content));
|
}
|
|
@Test
|
public void testParseQuoteMessageId_quoteIsNotMap() {
|
// 准备：quote 字段不是对象
|
String content = "{\"text\":\"你好\",\"quote\":\"oops\"}";
|
|
// 调用 + 断言
|
assertNull(ImMessageUtils.parseQuoteMessageId(content));
|
}
|
|
@Test
|
public void testParseQuoteMessageId_messageIdMissing() {
|
// 准备：quote 对象内无 messageId
|
String content = "{\"quote\":{\"senderId\":2}}";
|
|
// 调用 + 断言
|
assertNull(ImMessageUtils.parseQuoteMessageId(content));
|
}
|
|
// ========== appendQuote ==========
|
|
@Test
|
public void testAppendQuote_existingContent() {
|
// 准备：已有 text 字段的 content
|
String content = "{\"text\":\"你好\"}";
|
QuoteMessage quote = new QuoteMessage().setMessageId(1001L).setSenderId(2L).setType(1).setContent("{}");
|
|
// 调用
|
String result = ImMessageUtils.appendQuote(content, quote);
|
|
// 断言：保留原字段，注入 quote
|
Map<String, Object> map = JsonUtils.parseMap(result);
|
assertEquals("你好", map.get("text"));
|
assertTrue(map.get("quote") instanceof Map);
|
assertEquals(1001, ((Map<?, ?>) map.get("quote")).get("messageId"));
|
}
|
|
@Test
|
public void testAppendQuote_blankContent() {
|
// 准备：content 为空字符串，无法解析为 map
|
QuoteMessage quote = new QuoteMessage().setMessageId(1001L);
|
|
// 调用
|
String result = ImMessageUtils.appendQuote("", quote);
|
|
// 断言：内部新建 map，仅含 quote
|
Map<String, Object> map = JsonUtils.parseMap(result);
|
assertEquals(1, map.size());
|
assertTrue(map.containsKey("quote"));
|
}
|
|
@Test
|
public void testAppendQuote_overwriteExistingQuote() {
|
// 准备：content 已经有一个旧 quote，期望被新 quote 覆盖
|
String content = "{\"text\":\"hi\",\"quote\":{\"messageId\":1}}";
|
QuoteMessage quote = new QuoteMessage().setMessageId(9999L).setSenderId(2L).setType(1);
|
|
// 调用
|
String result = ImMessageUtils.appendQuote(content, quote);
|
|
// 断言：quote.messageId 被覆盖为新值
|
Map<String, Object> map = JsonUtils.parseMap(result);
|
assertEquals(9999, ((Map<?, ?>) map.get("quote")).get("messageId"));
|
}
|
|
// ========== removeQuote ==========
|
|
@Test
|
public void testRemoveQuote_blankContent() {
|
// 调用 + 断言：空内容原样返回
|
assertEquals("", ImMessageUtils.removeQuote(""));
|
assertNull(ImMessageUtils.removeQuote(null));
|
}
|
|
@Test
|
public void testRemoveQuote_noQuoteField() {
|
// 准备：content 字符串里没有 quote 关键字，提前返回原值
|
String content = "{\"text\":\"你好\"}";
|
|
// 调用 + 断言：同一引用直接返回
|
assertEquals(content, ImMessageUtils.removeQuote(content));
|
}
|
|
@Test
|
public void testRemoveQuote_withQuote() {
|
// 准备：含 quote 的 content
|
String content = "{\"text\":\"你好\",\"quote\":{\"messageId\":1}}";
|
|
// 调用
|
String result = ImMessageUtils.removeQuote(content);
|
|
// 断言：quote 被移除，保留其它字段
|
Map<String, Object> map = JsonUtils.parseMap(result);
|
assertFalse(map.containsKey("quote"));
|
assertEquals("你好", map.get("text"));
|
}
|
|
@Test
|
public void testRemoveQuote_invalidJsonContainingQuoteToken() {
|
// 准备：字符串包含 quote 字面量但解析失败
|
String content = "not-a-json-but-has-\"quote\"-token";
|
|
// 调用 + 断言：解析失败返回原值
|
assertEquals(content, ImMessageUtils.removeQuote(content));
|
}
|
|
// ========== buildQuote ==========
|
|
@Test
|
public void testBuildQuote_stripsNestedQuote() {
|
// 准备：被引用消息本身也含有 quote 字段；防止嵌套
|
String originalContent = "{\"text\":\"被引用\",\"quote\":{\"messageId\":777}}";
|
|
// 调用
|
QuoteMessage quote = ImMessageUtils.buildQuote(1001L, 2L, 1, originalContent);
|
|
// 断言：基础字段透传，content 已被剥离 quote
|
assertEquals(1001L, quote.getMessageId());
|
assertEquals(2L, quote.getSenderId());
|
assertEquals(1, quote.getType());
|
Map<String, Object> contentMap = JsonUtils.parseMap(quote.getContent());
|
assertFalse(contentMap.containsKey("quote"));
|
assertEquals("被引用", contentMap.get("text"));
|
}
|
|
@Test
|
public void testBuildQuote_originalWithoutQuote() {
|
// 准备：被引用消息原本就没 quote 字段
|
String originalContent = "{\"text\":\"hi\"}";
|
|
// 调用
|
QuoteMessage quote = ImMessageUtils.buildQuote(1001L, 2L, 1, originalContent);
|
|
// 断言：content 原样保留
|
assertEquals(originalContent, quote.getContent());
|
}
|
|
}
|
