truncateDatabase 支持自定义连接，分表分库写测试用例需要

hunzhiwange · Mar 18, 2023 · e6448a4 · e6448a4
1 parent 8db97a1
commit e6448a4
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/Leevel/Kernel/Testing/Database.php b/src/Leevel/Kernel/Testing/Database.php
@@ -14,7 +14,7 @@ trait Database
     /**
      * 清理数据表.
      */
-    protected function truncateDatabase(array $tables): void
+    protected function truncateDatabase(array $tables, ?string $connect = null): void
     {
         if (!$tables) {
             return;
@@ -28,16 +28,22 @@ protected function truncateDatabase(array $tables): void
                     false
                 ]
                 eot;
-            Db::table($table)->truncate();
+            Db::connect($connect)
+                ->table($table)
+                ->truncate()
+            ;
 
             $this->assertSame(
                 sprintf($sql, $table),
                 $this->varJson(
-                    Db::getRealLastSql()
+                    Db::connect($connect)->getRealLastSql()
                 )
             );
 
-            Db::table($table)->truncate();
+            Db::connect($connect)
+                ->table($table)
+                ->truncate()
+            ;
         }
     }
 }