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