Commit 348a0ff02a9efadc6427ab86c3b68f576e53a236
1 parent
b54c5eaa
test: 修复 code view 存在的问题
Showing
2 changed files
with
6 additions
and
2 deletions
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 |