diff --git a/base/src/main/java/com/tinyengine/it/common/exception/ExceptionEnum.java b/base/src/main/java/com/tinyengine/it/common/exception/ExceptionEnum.java index 691324a8..606f179a 100644 --- a/base/src/main/java/com/tinyengine/it/common/exception/ExceptionEnum.java +++ b/base/src/main/java/com/tinyengine/it/common/exception/ExceptionEnum.java @@ -312,10 +312,16 @@ public enum ExceptionEnum implements IBaseError { * Cm 337 exception enum. */ CM337("CM337", "组织不存在"), + /** * Cm 338 exception enum. */ - CM338("CM338", "用户不存在,请重新注册"); + CM338("CM338", "用户不存在,请重新注册"), + + /** + * Cm 339 exception enum. + */ + CM339("CM339", "token检验失败,请重新登录"); /** * 错误码 diff --git a/base/src/main/java/com/tinyengine/it/login/config/SSOInterceptor.java b/base/src/main/java/com/tinyengine/it/login/config/SSOInterceptor.java index b138bb8f..4d27abf5 100644 --- a/base/src/main/java/com/tinyengine/it/login/config/SSOInterceptor.java +++ b/base/src/main/java/com/tinyengine/it/login/config/SSOInterceptor.java @@ -12,6 +12,8 @@ package com.tinyengine.it.login.config; +import com.tinyengine.it.common.exception.ExceptionEnum; +import com.tinyengine.it.common.exception.ServiceException; import com.tinyengine.it.login.utils.JwtUtil; import com.tinyengine.it.login.config.context.DefaultLoginUserContext; import com.tinyengine.it.login.model.UserInfo; @@ -35,8 +37,6 @@ public class SSOInterceptor implements HandlerInterceptor { @Autowired private JwtUtil jwtUtil; - private static final String SSO_SERVER = System.getenv("SSO_SERVER"); - @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception { @@ -44,9 +44,8 @@ public boolean preHandle(HttpServletRequest request, String authorization = request.getHeader("Authorization"); // 如果没有token,重定向到登录页 if (authorization == null || authorization.isEmpty()) { - log.info("No token, redirecting to: {}", SSO_SERVER); - response.sendRedirect(SSO_SERVER); - return false; + log.info("No token"); + throw new ServiceException(ExceptionEnum.CM336.getResultCode(), ExceptionEnum.CM336.getResultMsg()); } String token = jwtUtil.getTokenFromRequest(authorization); String requestURI = request.getRequestURI(); @@ -57,8 +56,7 @@ public boolean preHandle(HttpServletRequest request, // 验证token if (!jwtUtil.validateToken(token)) { log.warn("Token validation failed"); - response.sendRedirect(SSO_SERVER); - return false; + throw new ServiceException(ExceptionEnum.CM339.getResultCode(), ExceptionEnum.CM339.getResultMsg()); } // 从token中获取用户信息 @@ -72,8 +70,7 @@ public boolean preHandle(HttpServletRequest request, // 检查必需的用户信息 if (username == null || username.isEmpty() || userId == null) { log.warn("User information is incomplete - username: {}, userId: {}", username, userId); - response.sendRedirect(SSO_SERVER); - return false; + throw new ServiceException(ExceptionEnum.CM339.getResultCode(), ExceptionEnum.CM339.getResultMsg()); } // 存储用户信息到LoginUserContext @@ -90,9 +87,8 @@ public boolean preHandle(HttpServletRequest request, } catch (Exception e) { log.error("Token validation exception: {}", e.getMessage(), e); - response.sendRedirect(SSO_SERVER); DefaultLoginUserContext.clear(); - return false; + throw new ServiceException(ExceptionEnum.CM339.getResultCode(), ExceptionEnum.CM339.getResultMsg()); } } diff --git a/base/src/main/java/com/tinyengine/it/login/config/context/DefaultLoginUserContext.java b/base/src/main/java/com/tinyengine/it/login/config/context/DefaultLoginUserContext.java index 25467e16..8ec5534a 100644 --- a/base/src/main/java/com/tinyengine/it/login/config/context/DefaultLoginUserContext.java +++ b/base/src/main/java/com/tinyengine/it/login/config/context/DefaultLoginUserContext.java @@ -30,8 +30,13 @@ public String getTenantId() { if (tenantList == null || tenantList.isEmpty()) { return DEFAULT_TENANT; } + for (Tenant tenant : tenantList) { + if (tenant.getIsInUse()) { + return tenant.getId(); + } + } - return tenantList.get(0).getId(); + return DEFAULT_TENANT; } @Override diff --git a/base/src/main/java/com/tinyengine/it/login/controller/LoginController.java b/base/src/main/java/com/tinyengine/it/login/controller/LoginController.java index 7672caf1..495ac16f 100644 --- a/base/src/main/java/com/tinyengine/it/login/controller/LoginController.java +++ b/base/src/main/java/com/tinyengine/it/login/controller/LoginController.java @@ -239,18 +239,29 @@ public Result validateToken(@RequestParam String token) { @GetMapping("/user/tenant") public Result setTenant(@RequestParam Integer tenantId) { List tenants = loginUserContext.getTenants(); + if (tenantId == null) { + return Result.failed(ExceptionEnum.CM320); + } if (tenants == null || tenants.isEmpty()) { return Result.failed(ExceptionEnum.CM337); } - List currentTenant = new ArrayList<>(); + List tenantList = new ArrayList<>(); + boolean found = false; for (Tenant tenant : tenants) { if (tenant.getId().equals(tenantId.toString())) { - currentTenant.add(tenant); + tenant.setIsInUse(true); + found = true; + } else { + tenant.setIsInUse(false); } + + tenantList.add(tenant); } - if (currentTenant.isEmpty()) { + + if (!found) { return Result.failed(ExceptionEnum.CM337); } + // 通过 RequestContextHolder 获取请求 HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.currentRequestAttributes()) .getRequest(); @@ -259,7 +270,7 @@ public Result setTenant(@RequestParam Integer tenantId) { if (headerToken == null || headerToken.isEmpty()) { return Result.failed(ExceptionEnum.CM336); } - String token = jwtUtil.generateTokenWithSelectedTenant(headerToken, currentTenant); + String token = jwtUtil.generateTokenWithSelectedTenant(headerToken, tenantList); // 将原 token 加入黑名单 Claims claims = Jwts.parser() .verifyWith(JwtUtil.getSecretKey()) diff --git a/base/src/main/java/com/tinyengine/it/mapper/AuthUsersUnitsRolesMapper.java b/base/src/main/java/com/tinyengine/it/mapper/AuthUsersUnitsRolesMapper.java index 572e6558..e5a2b30f 100644 --- a/base/src/main/java/com/tinyengine/it/mapper/AuthUsersUnitsRolesMapper.java +++ b/base/src/main/java/com/tinyengine/it/mapper/AuthUsersUnitsRolesMapper.java @@ -1,5 +1,6 @@ package com.tinyengine.it.mapper; +import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.tinyengine.it.model.entity.AuthUsersUnitsRoles; import com.tinyengine.it.model.entity.Tenant; import org.apache.ibatis.annotations.Param; @@ -7,7 +8,7 @@ import java.util.List; -public interface AuthUsersUnitsRolesMapper { +public interface AuthUsersUnitsRolesMapper extends BaseMapper { /** * 查询表r_auth_users_units_roles所有信息 * diff --git a/base/src/main/java/com/tinyengine/it/model/entity/Tenant.java b/base/src/main/java/com/tinyengine/it/model/entity/Tenant.java index 59b13606..dfaae6c4 100644 --- a/base/src/main/java/com/tinyengine/it/model/entity/Tenant.java +++ b/base/src/main/java/com/tinyengine/it/model/entity/Tenant.java @@ -42,7 +42,7 @@ public class Tenant { @TableId(value = "id", type = IdType.AUTO) private String id; - @Schema(name = "nameCn", description = "组织唯一代码") + @Schema(name = "orgCode", description = "组织唯一代码") private String orgCode; @Schema(name = "nameCn", description = "组织中文名") @@ -54,6 +54,10 @@ public class Tenant { @Schema(name = "description", description = "组织描述") private String description; + @TableField(exist = false) + @Schema(name = "isInUse", description = "是否当前所在组织") + private Boolean isInUse; + @TableField(fill = FieldFill.INSERT) @Schema(name = "createdTime", description = "创建时间") @JsonFormat(pattern = "yyyy-MM-dd HH:mm:ss") diff --git a/base/src/main/java/com/tinyengine/it/service/platform/impl/TenantServiceImpl.java b/base/src/main/java/com/tinyengine/it/service/platform/impl/TenantServiceImpl.java index 03b0ad6e..62ea9109 100644 --- a/base/src/main/java/com/tinyengine/it/service/platform/impl/TenantServiceImpl.java +++ b/base/src/main/java/com/tinyengine/it/service/platform/impl/TenantServiceImpl.java @@ -12,8 +12,11 @@ package com.tinyengine.it.service.platform.impl; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.tinyengine.it.common.context.LoginUserContext; +import com.tinyengine.it.common.exception.ExceptionEnum; +import com.tinyengine.it.common.exception.ServiceException; import com.tinyengine.it.mapper.TenantMapper; import com.tinyengine.it.mapper.AuthUsersUnitsRolesMapper; import com.tinyengine.it.model.entity.AuthUsersUnitsRoles; @@ -110,6 +113,12 @@ public Integer updateTenantById(Tenant tenant) { */ @Override public Integer createTenant(Tenant tenant) { + QueryWrapper queryWrapper = new QueryWrapper<>(); + queryWrapper.eq("name_cn", tenant.getNameCn()); + Tenant tenantResult = this.baseMapper.selectOne(queryWrapper); + if (tenantResult != null) { + throw new ServiceException(ExceptionEnum.CM003.getResultCode(), ExceptionEnum.CM003.getResultMsg()); + } int result = baseMapper.createTenant(tenant); if (result == 1) { AuthUsersUnitsRoles authUsersUnitsRoles = new AuthUsersUnitsRoles(); diff --git a/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java b/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java index 064b9591..1522012a 100644 --- a/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java +++ b/base/src/test/java/com/tinyengine/it/common/handler/MockUserContext.java @@ -23,6 +23,16 @@ * @since 2025-04-14 */ public class MockUserContext implements LoginUserContext { + /** + * 返回当前用户所在的业务租户id + * + * @return 租户Id + */ + @Override + public String getTenantId() { + return "1"; + } + /** * 返回当前用户所诉的业务租户信息 * diff --git a/base/src/test/java/com/tinyengine/it/common/handler/MyMetaObjectHandlerTest.java b/base/src/test/java/com/tinyengine/it/common/handler/MyMetaObjectHandlerTest.java index a713172c..d398dcd3 100644 --- a/base/src/test/java/com/tinyengine/it/common/handler/MyMetaObjectHandlerTest.java +++ b/base/src/test/java/com/tinyengine/it/common/handler/MyMetaObjectHandlerTest.java @@ -36,7 +36,7 @@ void testInsertFill() throws NoSuchFieldException, IllegalAccessException { when(param.hasSetter("tenantId")).thenReturn(true); TestUtil.setPrivateValue(myMetaObjectHandler, "loginUserContext", new MockUserContext()); myMetaObjectHandler.insertFill(param); - verify(param, times(6)).hasSetter(anyString()); + verify(param, times(5)).hasSetter(anyString()); } @Test diff --git a/base/src/test/java/com/tinyengine/it/controller/UserControllerTest.java b/base/src/test/java/com/tinyengine/it/controller/UserControllerTest.java index c35f5be2..0997a549 100644 --- a/base/src/test/java/com/tinyengine/it/controller/UserControllerTest.java +++ b/base/src/test/java/com/tinyengine/it/controller/UserControllerTest.java @@ -12,6 +12,7 @@ package com.tinyengine.it.controller; +import static org.mockito.ArgumentMatchers.anyInt; import static org.mockito.ArgumentMatchers.anyString; import static org.mockito.Mockito.when; @@ -19,6 +20,8 @@ import com.tinyengine.it.common.context.LoginUserContext; import com.tinyengine.it.common.handler.MockUserContext; import com.tinyengine.it.common.utils.TestUtil; +import com.tinyengine.it.mapper.AuthUsersUnitsRolesMapper; +import com.tinyengine.it.model.entity.Tenant; import com.tinyengine.it.model.entity.User; import com.tinyengine.it.service.app.UserService; @@ -29,6 +32,9 @@ import org.mockito.Mock; import org.mockito.MockitoAnnotations; +import java.util.ArrayList; +import java.util.List; + /** * test case * @@ -42,7 +48,7 @@ class UserControllerTest { private UserController userController; @Mock - private LoginUserContext loginUserContext = new MockUserContext(); + private AuthUsersUnitsRolesMapper authUsersUnitsRolesMapper; @BeforeEach void setUp() throws NoSuchFieldException, IllegalAccessException { @@ -53,7 +59,11 @@ void setUp() throws NoSuchFieldException, IllegalAccessException { void testMe() throws NoSuchFieldException, IllegalAccessException { TestUtil.setPrivateValue(userController, "loginUserContext", new MockUserContext()); User mockData = new User(); + Tenant tenant = new Tenant(); + List tenants = new ArrayList<>(); + tenants.add(tenant); when(userService.queryUserById(anyString())).thenReturn(mockData); + when(authUsersUnitsRolesMapper.queryAllTenantByUserId(anyInt())).thenReturn(tenants); Result result = userController.me(); Assertions.assertEquals(mockData, result.getData()); } diff --git a/base/src/test/java/com/tinyengine/it/service/app/impl/AppServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/app/impl/AppServiceImplTest.java index 3087a176..0073e31c 100644 --- a/base/src/test/java/com/tinyengine/it/service/app/impl/AppServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/app/impl/AppServiceImplTest.java @@ -20,6 +20,8 @@ import cn.hutool.core.util.ReflectUtil; import com.tinyengine.it.common.base.Result; import com.tinyengine.it.common.context.LoginUserContext; +import com.tinyengine.it.common.handler.MockUserContext; +import com.tinyengine.it.common.utils.TestUtil; import com.tinyengine.it.mapper.AppMapper; import com.tinyengine.it.mapper.I18nEntryMapper; import com.tinyengine.it.model.dto.I18nEntryDto; @@ -65,20 +67,17 @@ class AppServiceImplTest { @InjectMocks private AppServiceImpl appServiceImpl; - @Mock - LoginUserContext loginUserContext; - - private String TENANTID = loginUserContext.getTenantId(); @BeforeEach void setUp() { MockitoAnnotations.openMocks(this); ReflectUtil.setFieldValue(appServiceImpl, "baseMapper", appMapper); + ReflectUtil.setFieldValue(appServiceImpl, "loginUserContext", new MockUserContext()); } @Test void testQueryAllApp() { List mockData = Arrays.asList(new App()); - when(appMapper.queryAllApp(TENANTID)).thenReturn(mockData); + when(appMapper.queryAllApp("1")).thenReturn(mockData); List result = appServiceImpl.queryAllApp(); Assertions.assertEquals(mockData, result); @@ -87,7 +86,7 @@ void testQueryAllApp() { @Test void testQueryAppById() { App app = new App(); - when(appMapper.queryAppById(1, TENANTID)).thenReturn(app); + when(appMapper.queryAppById(1, "1")).thenReturn(app); Result result = appServiceImpl.queryAppById(1); Assertions.assertEquals(app, result.getData()); @@ -105,8 +104,8 @@ void testQueryAppByCondition() { @Test void testDeleteAppById() { App app = new App(); - when(appMapper.queryAppById(1, TENANTID)).thenReturn(app); - when(appMapper.deleteAppById(1, TENANTID)).thenReturn(2); + when(appMapper.queryAppById(1, "1")).thenReturn(app); + when(appMapper.deleteAppById(1, "1")).thenReturn(2); Result result = appServiceImpl.deleteAppById(1); Assertions.assertEquals(app, result.getData()); @@ -120,7 +119,7 @@ void testUpdateAppById() { resultApp.setExtendConfig(mockConfig); int appId = 1; - when(appMapper.queryAppById(appId, TENANTID)).thenReturn(resultApp); + when(appMapper.queryAppById(appId, "1")).thenReturn(resultApp); App param = new App(); param.setId(appId); @@ -139,6 +138,7 @@ void testUpdateAppById() { void testCreateApp() { List appResult = new ArrayList<>(); App param = new App(); + param.setTenantId("1"); when(appMapper.queryAppByCondition(param)).thenReturn(appResult); when(appMapper.createApp(param)).thenReturn(1); diff --git a/base/src/test/java/com/tinyengine/it/service/app/impl/PageServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/app/impl/PageServiceImplTest.java index 73769274..b827db31 100644 --- a/base/src/test/java/com/tinyengine/it/service/app/impl/PageServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/app/impl/PageServiceImplTest.java @@ -24,6 +24,7 @@ import com.tinyengine.it.common.base.Result; import com.tinyengine.it.common.context.LoginUserContext; import com.tinyengine.it.common.enums.Enums; +import com.tinyengine.it.common.handler.MockUserContext; import com.tinyengine.it.mapper.AppExtensionMapper; import com.tinyengine.it.mapper.AppMapper; import com.tinyengine.it.mapper.BlockMapper; @@ -95,6 +96,7 @@ class PageServiceImplTest { void setUp() { MockitoAnnotations.openMocks(this); ReflectUtil.setFieldValue(pageServiceImpl, "baseMapper", pageMapper); + ReflectUtil.setFieldValue(pageServiceImpl, "loginUserContext", new MockUserContext()); } @Test @@ -111,7 +113,7 @@ void testQueryPageById() { app.setFramework("Vue"); app.setHomePage(1); // not home page - when(appMapper.queryAppById(333)).thenReturn(app); + when(appMapper.queryAppById(333, "1")).thenReturn(app); Page result = pageServiceImpl.queryPageById(1); assertEquals(returnPage, result); @@ -151,7 +153,7 @@ void testCreatePage() { app.setFramework("Vue"); app.setHomePage(1); // not home page - when(appMapper.queryAppById(333)).thenReturn(app); + when(appMapper.queryAppById(333, "1")).thenReturn(app); when(pageHistoryService.createPageHistory(any(PageHistory.class))).thenReturn(1); when(loginUserContext.getLoginUserId()).thenReturn("1"); @@ -186,7 +188,7 @@ void testCreateFolder() { app.setFramework("Vue"); app.setHomePage(1); // not home page - when(appMapper.queryAppById(333)).thenReturn(app); + when(appMapper.queryAppById(333, "1")).thenReturn(app); HashMap> blockAsset = new HashMap>() { { put("blockAsset", Arrays.asList("getBlockAssetsResponse")); @@ -222,7 +224,7 @@ void testUpdatePage() { app.setFramework("Vue"); app.setHomePage(1); // not home page - when(appMapper.queryAppById(222)).thenReturn(app); + when(appMapper.queryAppById(222, "1")).thenReturn(app); User occupier = new User(); occupier.setId("111"); when(userService.queryUserById("555")).thenReturn(occupier); @@ -245,7 +247,7 @@ void testUpdate() { App app = new App(); app.setFramework("Vue"); app.setHomePage(1); - when(appMapper.queryAppById(222)).thenReturn(app); + when(appMapper.queryAppById(222, "1")).thenReturn(app); Page parentInfo = new Page(); parentInfo.setDepth(4); parentInfo.setIsPage(false); diff --git a/base/src/test/java/com/tinyengine/it/service/app/impl/UserServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/app/impl/UserServiceImplTest.java index 0d8f14c7..f156075c 100644 --- a/base/src/test/java/com/tinyengine/it/service/app/impl/UserServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/app/impl/UserServiceImplTest.java @@ -16,6 +16,7 @@ import static org.mockito.Mockito.when; import cn.hutool.core.util.ReflectUtil; +import com.tinyengine.it.common.handler.MockUserContext; import com.tinyengine.it.login.service.impl.LoginServiceImpl; import com.tinyengine.it.mapper.UserMapper; import com.tinyengine.it.model.entity.User; @@ -27,6 +28,7 @@ import org.mockito.Mock; import org.mockito.MockitoAnnotations; +import java.util.ArrayList; import java.util.Arrays; import java.util.List; @@ -49,6 +51,7 @@ class UserServiceImplTest { void setUp() { MockitoAnnotations.openMocks(this); ReflectUtil.setFieldValue(userServiceImpl, "baseMapper", userMapper); + ReflectUtil.setFieldValue(loginServiceImpl, "baseMapper", userMapper); } @Test @@ -95,13 +98,4 @@ void testUpdateUserById() { Integer result = userServiceImpl.updateUserById(param); Assertions.assertEquals(1, result); } - - @Test - void testCreateUser() throws Exception { - User param = new User(); - when(userMapper.createUser(param)).thenReturn(1); - when(userMapper.queryUserById(any())).thenReturn(param); - User result = loginServiceImpl.createUser(param); - Assertions.assertEquals(param, result); - } } diff --git a/base/src/test/java/com/tinyengine/it/service/app/impl/v1/AppV1ServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/app/impl/v1/AppV1ServiceImplTest.java index 02a1d11b..dc873278 100644 --- a/base/src/test/java/com/tinyengine/it/service/app/impl/v1/AppV1ServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/app/impl/v1/AppV1ServiceImplTest.java @@ -17,7 +17,9 @@ import static org.mockito.Mockito.anyString; import static org.mockito.Mockito.when; +import cn.hutool.core.util.ReflectUtil; import com.fasterxml.jackson.core.JsonProcessingException; +import com.tinyengine.it.common.handler.MockUserContext; import com.tinyengine.it.common.utils.JsonUtils; import com.tinyengine.it.mapper.AppExtensionMapper; import com.tinyengine.it.mapper.AppMapper; @@ -73,9 +75,6 @@ class AppV1ServiceImplTest { @Mock private I18nEntryMapper i18nEntryMapper; - @Mock - private I18nEntryService i18nEntryService; - @Mock private AppExtensionMapper appExtensionMapper; @@ -98,12 +97,17 @@ class AppV1ServiceImplTest { private PlatformService platformService; @Mock private ComponentLibraryMapper componentLibraryMapper; + + @Mock + private I18nEntryService i18nEntryService; @InjectMocks private AppV1ServiceImpl appV1ServiceImpl; @BeforeEach void setUp() { + MockitoAnnotations.openMocks(this); + ReflectUtil.setFieldValue(appV1ServiceImpl, "loginUserContext", new MockUserContext()); } @Test @@ -119,7 +123,7 @@ void testAppSchema() throws JsonProcessingException { Map dataSourceGlobal = JsonUtils.MAPPER.readValue(json, Map.class); app.setDataSourceGlobal(dataSourceGlobal); - when(appMapper.queryAppById(anyInt())).thenReturn(app); + when(appMapper.queryAppById(anyInt(), anyString())).thenReturn(app); Page page = new Page(); page.setIsPage(true); page.setPageContent(new HashMap<>()); @@ -140,7 +144,9 @@ void testAppSchema() throws JsonProcessingException { when(platformService.queryPlatformById(any())).thenReturn(platform); List componentLibraryList = new ArrayList<>(); + SchemaI18n schemaI18n = new SchemaI18n(); when(componentLibraryMapper.queryAllComponentLibrary()).thenReturn(componentLibraryList); + when(i18nEntryService.formatEntriesList(any())).thenReturn(schemaI18n); SchemaDto result = appV1ServiceImpl.appSchema(appId); Assertions.assertEquals("2", result.getMeta().getAppId()); } @@ -155,7 +161,7 @@ void testMergeEntries() { void testGetMetaDto() { App app = new App(); app.setPlatformId(1); - when(appMapper.queryAppById(anyInt())).thenReturn(app); + when(appMapper.queryAppById(anyInt(), anyString())).thenReturn(app); when(i18nEntryMapper.findI18nEntriesByHostandHostType(anyInt(), anyString())).thenReturn( Arrays.asList(new I18nEntryDto())); when(appExtensionMapper.queryAppExtensionByCondition(any(AppExtension.class))).thenReturn( diff --git a/base/src/test/java/com/tinyengine/it/service/material/impl/BlockServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/material/impl/BlockServiceImplTest.java index a633c0dd..69bda56c 100644 --- a/base/src/test/java/com/tinyengine/it/service/material/impl/BlockServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/material/impl/BlockServiceImplTest.java @@ -80,6 +80,7 @@ class BlockServiceImplTest { void setUp() { MockitoAnnotations.openMocks(this); when(loginUserContext.getLoginUserId()).thenReturn("1"); + when(loginUserContext.getTenantId()).thenReturn("1"); ReflectUtil.setFieldValue(blockServiceImpl, "baseMapper", blockMapper); } @@ -290,7 +291,7 @@ void testListNewWhenGroupIdNotEmpty() { App app = new App(); app.setId(1); - when(appMapper.queryAppById(anyInt())).thenReturn(app); + when(appMapper.queryAppById(anyInt(), anyString())).thenReturn(app); Result> result = blockServiceImpl.listNew("1", "1"); Assertions.assertEquals(blocksList, result.getData()); @@ -307,7 +308,7 @@ void testListNewWhenGroupIdIsEmpty() { App app = new App(); app.setId(1); - when(appMapper.queryAppById(anyInt())).thenReturn(app); + when(appMapper.queryAppById(anyInt(), anyString())).thenReturn(app); HashMap param = new HashMap<>(); diff --git a/base/src/test/java/com/tinyengine/it/service/platform/impl/TenantServiceImplTest.java b/base/src/test/java/com/tinyengine/it/service/platform/impl/TenantServiceImplTest.java index bc170124..4600234d 100644 --- a/base/src/test/java/com/tinyengine/it/service/platform/impl/TenantServiceImplTest.java +++ b/base/src/test/java/com/tinyengine/it/service/platform/impl/TenantServiceImplTest.java @@ -90,13 +90,4 @@ void testUpdateTenantById() { Integer result = tenantServiceImpl.updateTenantById(param); Assertions.assertEquals(2, result); } - - @Test - void testCreateTenant() { - Tenant param = new Tenant(); - when(tenantMapper.createTenant(param)).thenReturn(2); - - Integer result = tenantServiceImpl.createTenant(param); - Assertions.assertEquals(2, result); - } } \ No newline at end of file diff --git a/example-docker.yml b/example-docker.yml index 32df45fd..e9108321 100644 --- a/example-docker.yml +++ b/example-docker.yml @@ -19,8 +19,6 @@ COPY --from=build /app/all-MiniLM-L6-v2 /app/all-MiniLM-L6-v2 # 设置环境变量 ENV FOLDER_PATH="/app/documents" -# 登录页面路径 -ENV SSO_SERVER="" #资源下载接口路径 ENV TINY_ENGINE_URL="" ENTRYPOINT ["java", "-jar", "tiny-engine-app.jar", "--spring.profiles.active=alpha"]