diff --git a/pigx-common/pigx-common-test/pom.xml b/pigx-common/pigx-common-test/pom.xml
index 005d07d61..659e9a5bb 100755
--- a/pigx-common/pigx-common-test/pom.xml
+++ b/pigx-common/pigx-common-test/pom.xml
@@ -15,6 +15,10 @@
pigx oauth 2.0 单元测试工具类
+
+ com.pig4cloud
+ pigx-common-data
+
com.pig4cloud
pigx-common-security
diff --git a/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/WithMockSecurityContextFactory.java b/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/WithMockSecurityContextFactory.java
index a882b73ce..77c1e5e22 100644
--- a/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/WithMockSecurityContextFactory.java
+++ b/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/WithMockSecurityContextFactory.java
@@ -5,6 +5,7 @@ import java.nio.charset.StandardCharsets;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONUtil;
import com.pig4cloud.pigx.common.core.util.SpringContextHolder;
+import com.pig4cloud.pigx.common.data.tenant.TenantContextHolder;
import com.pig4cloud.pigx.test.annotation.WithMockOAuth2User;
import org.springframework.http.HttpEntity;
@@ -32,6 +33,9 @@ public class WithMockSecurityContextFactory implements WithSecurityContextFactor
@Override
public SecurityContext createSecurityContext(WithMockOAuth2User oAuth2User) {
+ // 0. 初始化环境
+ TenantContextHolder.setTenantId(oAuth2User.tenant());
+
// 1. 请求认证中心获取token
String token = getToken(oAuth2User);
diff --git a/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/annotation/WithMockOAuth2User.java b/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/annotation/WithMockOAuth2User.java
index 23bc599f8..911c6d1ed 100644
--- a/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/annotation/WithMockOAuth2User.java
+++ b/pigx-common/pigx-common-test/src/main/java/com/pig4cloud/pigx/test/annotation/WithMockOAuth2User.java
@@ -1,11 +1,12 @@
package com.pig4cloud.pigx.test.annotation;
-import com.pig4cloud.pigx.test.WithMockSecurityContextFactory;
-import org.springframework.security.test.context.support.WithSecurityContext;
-
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
+import com.pig4cloud.pigx.test.WithMockSecurityContextFactory;
+
+import org.springframework.security.test.context.support.WithSecurityContext;
+
/**
* @author lengleng
* @date 2020/9/22
@@ -26,4 +27,9 @@ public @interface WithMockOAuth2User {
*/
String password() default "123456";
+ /**
+ * 租户编号 默认租户编号1
+ */
+ int tenant() default 1;
+
}
\ No newline at end of file