package com.ruoyi.order.controller;
|
|
import com.ruoyi.chargingPile.api.dto.GetSiteListDTO;
|
import com.ruoyi.chargingPile.api.feignClient.SiteClient;
|
import com.ruoyi.common.core.domain.R;
|
import com.ruoyi.common.security.service.TokenService;
|
import com.ruoyi.order.service.TOrderAppealService;
|
import com.ruoyi.system.api.domain.SysUser;
|
import org.junit.jupiter.api.Test;
|
import org.springframework.test.util.ReflectionTestUtils;
|
|
import java.util.Arrays;
|
import java.util.HashSet;
|
import java.util.Map;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.mockito.Mockito.mock;
|
import static org.mockito.Mockito.verify;
|
import static org.mockito.Mockito.when;
|
|
class TOrderAppealControllerTest {
|
|
@Test
|
void partnerAppealSitesUseAllSitesAssociatedWithCurrentAccount() {
|
TOrderAppealService appealService = mock(TOrderAppealService.class);
|
TokenService tokenService = mock(TokenService.class);
|
SiteClient siteClient = mock(SiteClient.class);
|
TOrderAppealController controller = new TOrderAppealController(appealService, tokenService);
|
ReflectionTestUtils.setField(controller, "siteClient", siteClient);
|
|
SysUser partnerUser = new SysUser();
|
partnerUser.setUserId(7L);
|
partnerUser.setObjectId(9);
|
when(siteClient.getSiteListByUserId(7L))
|
.thenReturn(R.ok(Arrays.asList(site(-1), site(11), null, site(12))));
|
|
Map<Integer, GetSiteListDTO> result = controller.getVisibleSiteMap(partnerUser);
|
|
assertEquals(new HashSet<>(Arrays.asList(11, 12)), result.keySet());
|
verify(siteClient).getSiteListByUserId(7L);
|
}
|
|
private GetSiteListDTO site(Integer id) {
|
GetSiteListDTO site = new GetSiteListDTO();
|
site.setId(id);
|
return site;
|
}
|
}
|
