From 80a42efb6117f025a983afe7801b160d91d0aa2d Mon Sep 17 00:00:00 2001
From: tangxin <617054137@qq.com>
Date: Thu, 13 Feb 2025 10:25:47 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96TenantManager.withoutTenantCo?=
 =?UTF-8?q?ndition=E4=BF=AE=E6=94=B9=E6=B5=8B=E8=AF=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/mybatisflex/test/TenantTester.java    | 24 +++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/TenantTester.java b/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/TenantTester.java
index bfbf11d6..ebd0df91 100644
--- a/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/TenantTester.java
+++ b/mybatis-flex-test/mybatis-flex-native-test/src/main/java/com/mybatisflex/test/TenantTester.java
@@ -19,7 +19,9 @@ import com.mybatisflex.core.MybatisFlexBootstrap;
 import com.mybatisflex.core.audit.AuditManager;
 import com.mybatisflex.core.audit.ConsoleMessageCollector;
 import com.mybatisflex.core.tenant.TenantManager;
+import com.mybatisflex.core.util.UpdateEntity;
 import com.mybatisflex.mapper.TenantAccountMapper;
+import org.junit.Test;
 import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseBuilder;
 import org.springframework.jdbc.datasource.embedded.EmbeddedDatabaseType;
 
@@ -28,7 +30,8 @@ import java.util.Date;
 
 public class TenantTester {
 
-    public static void main(String[] args) {
+    @Test
+    public void test(){
 
         DataSource dataSource = new EmbeddedDatabaseBuilder()
             .setType(EmbeddedDatabaseType.H2)
@@ -60,17 +63,30 @@ public class TenantTester {
         bean.setAge(10);
         bean.setBirthday(new Date());
         bean.setId(102L);
-
+        mapper.deleteById(102L);
         TenantManager.withoutTenantCondition(() -> mapper.insert(bean , true));
         bean.setTenantId(4L);
         TenantManager.withoutTenantCondition(() -> mapper.update(bean , true));
         bean.setTenantId(4L);
         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);
         TenantAccount tenantAccount = mapper.selectOneById(102L);
-        mapper.deleteById(102L);
         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()
 //            .select(TENANT_ACCOUNT.ALL_COLUMNS)
 //            .from(TENANT_ACCOUNT.as("c"), ACCOUNT.as("b"))