package cn.iocoder.yudao.module.im.service.friend;
|
|
import cn.hutool.core.collection.ListUtil;
|
import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum;
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
|
import cn.iocoder.yudao.module.im.controller.admin.friend.vo.ImFriendUpdateReqVO;
|
import cn.iocoder.yudao.module.im.dal.dataobject.friend.ImFriendDO;
|
import cn.iocoder.yudao.module.im.dal.mysql.friend.ImFriendMapper;
|
import cn.iocoder.yudao.module.im.service.message.ImPrivateMessageService;
|
import cn.iocoder.yudao.module.im.service.message.dto.ImPrivateMessageSendDTO;
|
import cn.iocoder.yudao.module.im.service.websocket.ImWebSocketService;
|
import cn.iocoder.yudao.module.system.api.user.AdminUserApi;
|
import org.junit.jupiter.api.Test;
|
import org.mockito.ArgumentCaptor;
|
import org.mockito.InjectMocks;
|
import org.mockito.Mock;
|
import org.springframework.dao.DuplicateKeyException;
|
|
import java.time.LocalDateTime;
|
import java.util.Collections;
|
import java.util.List;
|
|
import static cn.iocoder.yudao.module.im.enums.ErrorCodeConstants.FRIEND_NOT_FRIEND;
|
import static org.junit.jupiter.api.Assertions.*;
|
import static org.mockito.ArgumentMatchers.*;
|
import static org.mockito.Mockito.never;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
/**
|
* {@link ImFriendServiceImpl} 的单元测试
|
*
|
* @author 芋道源码
|
*/
|
public class ImFriendServiceImplTest extends BaseMockitoUnitTest {
|
|
@InjectMocks
|
private ImFriendServiceImpl friendService;
|
|
@Mock
|
private ImFriendMapper imFriendMapper;
|
@Mock
|
private AdminUserApi adminUserApi;
|
@Mock
|
private ImWebSocketService imWebSocketService;
|
@Mock
|
private ImPrivateMessageService privateMessageService;
|
|
// ========== updateFriend ==========
|
|
@Test
|
public void testUpdateFriend_success() {
|
// 准备
|
ImFriendUpdateReqVO reqVO = new ImFriendUpdateReqVO();
|
reqVO.setFriendUserId(2L);
|
reqVO.setSilent(true);
|
ImFriendDO friend = ImFriendDO.builder().id(100L).userId(1L).friendUserId(2L)
|
.silent(false).status(CommonStatusEnum.ENABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(friend);
|
|
// 调用
|
friendService.updateFriend(1L, reqVO);
|
|
// 断言：更新了 silent 字段
|
ArgumentCaptor<ImFriendDO> captor = ArgumentCaptor.forClass(ImFriendDO.class);
|
verify(imFriendMapper).updateById(captor.capture());
|
assertEquals(100L, captor.getValue().getId());
|
assertTrue(captor.getValue().getSilent());
|
// 断言：推送了好友更新通知
|
verify(imWebSocketService).sendNotificationAsync(eq(1L), anyInt(), anyInt(), any());
|
}
|
|
@Test
|
public void testUpdateFriend_notFriend() {
|
// 准备
|
ImFriendUpdateReqVO reqVO = new ImFriendUpdateReqVO();
|
reqVO.setFriendUserId(2L);
|
reqVO.setSilent(true);
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(null);
|
|
// 调用并断言
|
ServiceException exception = assertThrows(ServiceException.class,
|
() -> friendService.updateFriend(1L, reqVO));
|
assertEquals(FRIEND_NOT_FRIEND.getCode(), exception.getCode());
|
}
|
|
@Test
|
public void testUpdateFriend_disabledFriend() {
|
// 准备
|
ImFriendUpdateReqVO reqVO = new ImFriendUpdateReqVO();
|
reqVO.setFriendUserId(2L);
|
reqVO.setSilent(true);
|
ImFriendDO friend = ImFriendDO.builder().id(100L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.DISABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(friend);
|
|
// 调用并断言
|
ServiceException exception = assertThrows(ServiceException.class,
|
() -> friendService.updateFriend(1L, reqVO));
|
assertEquals(FRIEND_NOT_FRIEND.getCode(), exception.getCode());
|
verify(imFriendMapper, never()).updateById(any(ImFriendDO.class));
|
verify(imWebSocketService, never()).sendNotificationAsync(any(Long.class), anyInt(), anyInt(), any());
|
}
|
|
@Test
|
public void testUpdateFriend_displayNameOnly() {
|
// 准备：只传备注、不传 silent —— 走差量更新
|
ImFriendUpdateReqVO reqVO = new ImFriendUpdateReqVO();
|
reqVO.setFriendUserId(2L);
|
reqVO.setDisplayName("老张");
|
ImFriendDO friend = ImFriendDO.builder().id(100L).userId(1L).friendUserId(2L)
|
.silent(false).status(CommonStatusEnum.ENABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(friend);
|
|
// 调用
|
friendService.updateFriend(1L, reqVO);
|
|
// 断言：updateById 收到 displayName 但没有 silent（MyBatis-Plus 靠 NOT_NULL 跳过）
|
ArgumentCaptor<ImFriendDO> captor = ArgumentCaptor.forClass(ImFriendDO.class);
|
verify(imFriendMapper).updateById(captor.capture());
|
assertEquals(100L, captor.getValue().getId());
|
assertEquals("老张", captor.getValue().getDisplayName());
|
assertNull(captor.getValue().getSilent());
|
// 断言：推送好友更新通知
|
verify(imWebSocketService).sendNotificationAsync(eq(1L), anyInt(), anyInt(), any());
|
}
|
|
@Test
|
public void testUpdateFriend_emptyRequest() {
|
// 准备：silent / displayName / pinned 都不传 —— 进入方法立刻返回，不查 mapper 也不发推送
|
ImFriendUpdateReqVO reqVO = new ImFriendUpdateReqVO();
|
reqVO.setFriendUserId(2L);
|
|
// 调用
|
friendService.updateFriend(1L, reqVO);
|
|
// 断言：没查记录、没触发 SQL 更新 / 没发 WebSocket 推送
|
verify(imFriendMapper, never()).selectByUserIdAndFriendUserId(anyLong(), anyLong());
|
verify(imFriendMapper, never()).updateById(any(ImFriendDO.class));
|
verify(imWebSocketService, never()).sendNotificationAsync(any(Long.class), anyInt(), anyInt(), any());
|
}
|
|
// ========== 建立好友 ==========
|
|
@Test
|
public void testAddFriend0_existingEnabledSkip() {
|
// 准备：已存在且启用
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).silent(true).pinned(true).blocked(true).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
// 调用
|
friendService.addFriend0(1L, 2L, null, null);
|
|
// 断言：不插入也不更新
|
verify(imFriendMapper, never()).insert(any(ImFriendDO.class));
|
verify(imFriendMapper, never()).updateById(any(ImFriendDO.class));
|
verify(imFriendMapper, never()).updateReAddFields(anyLong(), anyInt(), any(LocalDateTime.class),
|
any(LocalDateTime.class), anyBoolean(), anyBoolean(), anyBoolean(), any(), any());
|
}
|
|
@Test
|
public void testAddFriend0_existingDisabledRecovers() {
|
// 准备：已存在且 DISABLE，应当恢复状态
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.DISABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
// 调用
|
friendService.addFriend0(1L, 2L, null, null);
|
|
// 断言：恢复 ENABLE，并清空 deleteTime
|
verify(imFriendMapper).updateReAddFields(eq(10L), eq(CommonStatusEnum.ENABLE.getStatus()),
|
any(LocalDateTime.class), any(LocalDateTime.class), eq(false), eq(false), eq(false), isNull(), isNull());
|
verify(imFriendMapper, never()).insert(any(ImFriendDO.class));
|
}
|
|
@Test
|
public void testAddFriend0_duplicateKeyPropagates() {
|
// 准备：mapper 抛并发冲突；极端并发下让异常向外抛，由外层事务回滚
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(null);
|
when(imFriendMapper.insert(any(ImFriendDO.class)))
|
.thenThrow(new DuplicateKeyException("concurrent insert"));
|
|
// 调用 + 断言：异常向外抛
|
assertThrows(DuplicateKeyException.class, () -> friendService.addFriend0(1L, 2L, null, null));
|
verify(imFriendMapper).insert(any(ImFriendDO.class));
|
}
|
|
// ========== deleteFriend ==========
|
|
@Test
|
public void testDeleteFriend0_alreadyDisabled() {
|
// 准备：已经是 DISABLE，不再更新
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.DISABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
boolean result = friendService.deleteFriend0(1L, 2L);
|
|
assertFalse(result);
|
verify(imFriendMapper, never()).updateById(any(ImFriendDO.class));
|
}
|
|
@Test
|
public void testDeleteFriend0_enabledGetsDisabled() {
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
boolean result = friendService.deleteFriend0(1L, 2L);
|
|
assertTrue(result);
|
ArgumentCaptor<ImFriendDO> captor = ArgumentCaptor.forClass(ImFriendDO.class);
|
verify(imFriendMapper).updateById(captor.capture());
|
assertEquals(CommonStatusEnum.DISABLE.getStatus(), captor.getValue().getStatus());
|
assertNotNull(captor.getValue().getDeleteTime());
|
}
|
|
@Test
|
public void testDeleteFriend_alreadyDisabledSkipNotification() {
|
// 准备：已经是 DISABLE，不再推本端删除通知
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.DISABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
friendService.deleteFriend(1L, 2L, true);
|
|
verify(imFriendMapper, never()).updateById(any(ImFriendDO.class));
|
verify(privateMessageService, never()).sendPrivateMessage(anyLong(), any(ImPrivateMessageSendDTO.class));
|
}
|
|
@Test
|
public void testDeleteFriend_enabledSendNotification() {
|
// 准备
|
ImFriendDO exists = ImFriendDO.builder().id(10L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build();
|
when(imFriendMapper.selectByUserIdAndFriendUserId(1L, 2L)).thenReturn(exists);
|
|
friendService.deleteFriend(1L, 2L, true);
|
|
verify(imFriendMapper).updateById(any(ImFriendDO.class));
|
verify(privateMessageService).sendPrivateMessage(eq(1L), any(ImPrivateMessageSendDTO.class));
|
}
|
|
// ========== 其它读方法 ==========
|
|
@Test
|
public void testGetFriendList() {
|
List<ImFriendDO> list = ListUtil.of(
|
ImFriendDO.builder().id(1L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build(),
|
ImFriendDO.builder().id(2L).userId(1L).friendUserId(3L)
|
.status(CommonStatusEnum.DISABLE.getStatus())
|
.deleteTime(LocalDateTime.now()).build()
|
);
|
when(imFriendMapper.selectListByUserId(1L)).thenReturn(list);
|
|
List<ImFriendDO> result = friendService.getFriendList(1L);
|
assertEquals(2, result.size());
|
}
|
|
@Test
|
public void testGetActiveFriendList_emptySkip() {
|
List<ImFriendDO> result = friendService.getActiveFriendList(1L, Collections.emptyList());
|
|
assertTrue(result.isEmpty());
|
verify(imFriendMapper, never()).selectListByUserIdAndFriendUserIdsAndStatus(anyLong(), anyCollection(), anyInt());
|
}
|
|
@Test
|
public void testGetMutualEnableFriendList_filterSingleSideDeleted() {
|
ImFriendDO friend2 = ImFriendDO.builder().id(1L).userId(1L).friendUserId(2L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build();
|
ImFriendDO friend3 = ImFriendDO.builder().id(2L).userId(1L).friendUserId(3L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build();
|
when(imFriendMapper.selectListByUserIdAndStatus(1L, CommonStatusEnum.ENABLE.getStatus()))
|
.thenReturn(ListUtil.of(friend2, friend3));
|
when(imFriendMapper.selectListByUserIdsAndFriendUserIdAndStatus(anyCollection(), eq(1L),
|
eq(CommonStatusEnum.ENABLE.getStatus()))).thenReturn(ListUtil.of(
|
ImFriendDO.builder().userId(2L).friendUserId(1L)
|
.status(CommonStatusEnum.ENABLE.getStatus()).build()));
|
|
List<ImFriendDO> result = friendService.getMutualEnableFriendList(1L);
|
|
assertEquals(1, result.size());
|
assertEquals(2L, result.get(0).getFriendUserId());
|
}
|
|
}
|
