| | |
| | | import com.ruoyi.chargingPile.api.feignClient.SiteClient; |
| | | import com.ruoyi.common.core.domain.R; |
| | | import com.ruoyi.common.security.service.TokenService; |
| | | import com.ruoyi.other.api.feignClient.GroupCompanyClient; |
| | | import com.ruoyi.system.api.feignClient.SysUserClient; |
| | | import org.apache.poi.xssf.usermodel.XSSFWorkbook; |
| | | import org.junit.jupiter.api.BeforeEach; |
| | |
| | | import org.junit.jupiter.api.extension.ExtendWith; |
| | | import org.mockito.Mock; |
| | | import org.mockito.junit.jupiter.MockitoExtension; |
| | | import org.redisson.api.RLock; |
| | | import org.redisson.api.RedissonClient; |
| | | import org.springframework.mock.web.MockMultipartFile; |
| | | import org.springframework.test.util.ReflectionTestUtils; |
| | | import org.springframework.transaction.PlatformTransactionManager; |
| | |
| | | |
| | | import java.io.ByteArrayOutputStream; |
| | | import java.util.Collections; |
| | | import java.util.concurrent.TimeUnit; |
| | | import java.util.concurrent.atomic.AtomicLong; |
| | | |
| | | import static org.junit.jupiter.api.Assertions.assertEquals; |
| | | import static org.mockito.ArgumentMatchers.any; |
| | | import static org.mockito.ArgumentMatchers.anyLong; |
| | | import static org.mockito.Mockito.doAnswer; |
| | | import static org.mockito.Mockito.mock; |
| | | import static org.mockito.Mockito.verify; |
| | |
| | | @Mock private TokenService tokenService; |
| | | @Mock private SysUserClient sysUserClient; |
| | | @Mock private PlatformTransactionManager transactionManager; |
| | | @Mock private GroupCompanyClient groupCompanyClient; |
| | | @Mock private RedissonClient redissonClient; |
| | | |
| | | private TAppUserWhitelistServiceImpl service; |
| | | |
| | |
| | | ReflectionTestUtils.setField(service, "tokenService", tokenService); |
| | | ReflectionTestUtils.setField(service, "sysUserClient", sysUserClient); |
| | | ReflectionTestUtils.setField(service, "transactionManager", transactionManager); |
| | | ReflectionTestUtils.setField(service, "groupCompanyClient", groupCompanyClient); |
| | | ReflectionTestUtils.setField(service, "redissonClient", redissonClient); |
| | | RLock lock = mock(RLock.class); |
| | | when(redissonClient.getLock(any())).thenReturn(lock); |
| | | try { |
| | | when(lock.tryLock(anyLong(), anyLong(), any(TimeUnit.class))).thenReturn(true); |
| | | } catch (InterruptedException exception) { |
| | | throw new IllegalStateException(exception); |
| | | } |
| | | } |
| | | |
| | | @Test |