admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-06 16:48:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

!527 优化TenantManager.withoutTenantCondition测试代码

Browse Source 
Merge pull request !527 from tangxin/main
This commit is contained in:
Michael Yang 2025-02-13 02:37:04 +00:00 committed by Gitee
commit 5c851e0b81
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/TenantTester.java
View File

@ -19,7 +19,9 @@ import com.mybatisflex.core.MybatisFlexBootstrap;
import com.mybatisflex.core.audit.AuditManager; import com.mybatisflex.core.audit.AuditManager;
import com.mybatisflex.core.audit.ConsoleMessageCollector; import com.mybatisflex.core.audit.ConsoleMessageCollector;
import com.mybatisflex.core.tenant.TenantManager; import com.mybatisflex.core.tenant.TenantManager;
import com.mybatisflex.core.util.UpdateEntity;
import com.mybatisflex.mapper.TenantAccountMapper; import com.mybatisflex.mapper.TenantAccountMapper;
import org.junit.Test;
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder; import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType; import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
@ -28,7 +30,8 @@ import java.util.Date;
public class TenantTester { public class TenantTester {
public static void main(String[] args) { @Test
public void test(){
DataSource dataSource = new EmbeddedDatabaseBuilder() DataSource dataSource = new EmbeddedDatabaseBuilder()
.setType(EmbeddedDatabaseType.H2) .setType(EmbeddedDatabaseType.H2)
@ -60,17 +63,30 @@ public class TenantTester {
bean.setAge(10); bean.setAge(10);
bean.setBirthday(new Date()); bean.setBirthday(new Date());
bean.setId(102L); bean.setId(102L);
mapper.deleteById(102L);
TenantManager.withoutTenantCondition(() -> mapper.insert(bean , true)); TenantManager.withoutTenantCondition(() -> mapper.insert(bean , true));
bean.setTenantId(4L); bean.setTenantId(4L);
TenantManager.withoutTenantCondition(() -> mapper.update(bean , true)); TenantManager.withoutTenantCondition(() -> mapper.update(bean , true));
bean.setTenantId(4L); bean.setTenantId(4L);
mapper.update(bean , true); mapper.update(bean , true);
TenantManager.setTenantFactory(() -> new Object[]{1, 2,4}); TenantManager.setTenantFactory(() -> new Object[]{1, 2,4,5});
mapper.selectAll().forEach(System.out::println); mapper.selectAll().forEach(System.out::println);
TenantAccount tenantAccount = mapper.selectOneById(102L); TenantAccount tenantAccount = mapper.selectOneById(102L);
mapper.deleteById(102L);
assert tenantAccount.getTenantId() == 4; assert tenantAccount.getTenantId() == 4;
tenantAccount = UpdateEntity.of(TenantAccount.class, 102L);
tenantAccount.setTenantId(5L);
tenantAccount.setAge(8);
mapper.update(tenantAccount , true);
tenantAccount = mapper.selectOneById(102L);
assert tenantAccount.getTenantId() == 4;
tenantAccount = UpdateEntity.of(TenantAccount.class, 102L);
tenantAccount.setTenantId(5L);
tenantAccount.setAge(8);
TenantAccount finalTenantAccount = tenantAccount;
TenantManager.withoutTenantCondition(() -> mapper.update(finalTenantAccount, true));
tenantAccount = mapper.selectOneById(102L);
assert tenantAccount.getTenantId() == 5;
// mapper.selectListByQuery(QueryWrapper.create() // mapper.selectListByQuery(QueryWrapper.create()
// .select(TENANT_ACCOUNT.ALL_COLUMNS) // .select(TENANT_ACCOUNT.ALL_COLUMNS)
// .from(TENANT_ACCOUNT.as("c"), ACCOUNT.as("b")) // .from(TENANT_ACCOUNT.as("c"), ACCOUNT.as("b"))