package com.ruoyi.account.controller;
|
|
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
|
import com.ruoyi.account.api.model.TAppUserTag;
|
import com.ruoyi.account.service.TAppUserTagService;
|
import com.ruoyi.common.core.domain.R;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.List;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
import static org.mockito.ArgumentMatchers.any;
|
import static org.mockito.ArgumentMatchers.anyCollection;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.times;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.verifyNoInteractions;
|
import static org.mockito.Mockito.when;
|
|
class TAppUserTagControllerTest {
|
|
@Test
|
void listByUserIdsReturnsEmptyWithoutTouchingServiceForNullOrEmptyInput() {
|
TAppUserTagService service = mock(TAppUserTagService.class);
|
TAppUserTagController controller = new TAppUserTagController();
|
ReflectionTestUtils.setField(controller, "appUserTagService", service);
|
|
R<List<TAppUserTag>> nullResult = controller.listByUserIds(null);
|
R<List<TAppUserTag>> emptyResult = controller.listByUserIds(Collections.emptyList());
|
|
assertTrue(nullResult.getData().isEmpty());
|
assertTrue(emptyResult.getData().isEmpty());
|
verifyNoInteractions(service);
|
}
|
|
@Test
|
void listByUserIdsChunksLargeInputByFiveHundredAndMergesResults() {
|
TAppUserTagService service = mock(TAppUserTagService.class);
|
TAppUserTagController controller = new TAppUserTagController();
|
ReflectionTestUtils.setField(controller, "appUserTagService", service);
|
|
LambdaQueryChainWrapper<TAppUserTag> chain = mock(LambdaQueryChainWrapper.class);
|
when(service.lambdaQuery()).thenReturn(chain);
|
when(chain.in(any(), anyCollection())).thenReturn(chain);
|
|
List<Long> userIds = new ArrayList<>();
|
for (int i = 0; i < 1001; i++) {
|
userIds.add((long) i);
|
}
|
TAppUserTag first = new TAppUserTag();
|
first.setAppUserId(0L);
|
TAppUserTag second = new TAppUserTag();
|
second.setAppUserId(500L);
|
TAppUserTag third = new TAppUserTag();
|
third.setAppUserId(1000L);
|
when(chain.list()).thenReturn(
|
Collections.singletonList(first),
|
Collections.singletonList(second),
|
Collections.singletonList(third));
|
|
List<TAppUserTag> result = controller.listByUserIds(userIds).getData();
|
|
assertEquals(3, result.size());
|
verify(service, times(3)).lambdaQuery();
|
}
|
}
|
