Commit 348a0ff02a9efadc6427ab86c3b68f576e53a236

Authored by gaibu
1 parent b54c5eaa

test: 修复 code view 存在的问题

yudao-framework/yudao-spring-boot-starter-biz-tenant/src/test/java/cn/iocoder/yudao/framework/tenant/core/job/TenantJobTest.java
... ... @@ -4,8 +4,8 @@ import cn.iocoder.yudao.framework.tenant.core.service.TenantFrameworkService;
4 4 import cn.iocoder.yudao.framework.test.core.ut.BaseMockitoUnitTest;
5 5 import com.google.common.collect.Lists;
6 6 import org.junit.jupiter.api.Test;
  7 +import org.mockito.Mock;
7 8 import org.mockito.Mockito;
8   -import org.mockito.Spy;
9 9  
10 10 import java.util.List;
11 11  
... ... @@ -18,7 +18,8 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
18 18 * @author gaibu
19 19 */
20 20 public class TenantJobTest extends BaseMockitoUnitTest {
21   - @Spy
  21 +
  22 + @Mock
22 23 TenantFrameworkService tenantFrameworkService;
23 24  
24 25 @Test
... ... @@ -31,8 +32,10 @@ public class TenantJobTest extends BaseMockitoUnitTest {
31 32 TestJob testJob = new TestJob();
32 33 // 创建任务装饰器
33 34 TenantJobHandlerDecorator tenantJobHandlerDecorator = new TenantJobHandlerDecorator(tenantFrameworkService, testJob);
  35 +
34 36 // 执行任务
35 37 tenantJobHandlerDecorator.execute(null);
  38 +
36 39 // 断言返回值
37 40 assertEquals(testJob.getTenantIds(), tenantIds);
38 41 }
... ...
yudao-framework/yudao-spring-boot-starter-biz-tenant/src/test/java/cn/iocoder/yudao/framework/tenant/core/job/TestJob.java
... ... @@ -11,6 +11,7 @@ import java.util.concurrent.CopyOnWriteArrayList;
11 11 @Component
12 12 @TenantJob // 标记多租户
13 13 public class TestJob implements JobHandler {
  14 +
14 15 private final List<Long> tenantIds = new CopyOnWriteArrayList<>();
15 16  
16 17 @Override
... ...