fix(testing): WithTestObject will correctly remove table after test.

LastDragon-ru · Apr 29, 2024 · 5f3b1ab · 5f3b1ab
1 parent 746c4c5
commit 5f3b1ab
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 12 deletions.
diff --git a/packages/eloquent/src/Testing/Package/Models/WithTestObject.php b/packages/eloquent/src/Testing/Package/Models/WithTestObject.php
@@ -11,7 +11,7 @@
 /**
  * @internal
  *
- * @mixin TestCase
+ * @phpstan-require-extends TestCase
  */
 trait WithTestObject {
     /**
@@ -43,11 +43,7 @@ protected function withTestObjectAfter(): void {
             $schema = Container::getInstance()->make(Builder::class);
             $table  = (new TestObject())->getTable();
 
-            if ($schema->hasTable($table)) {
-                return;
-            }
-
-            $schema->drop($table);
+            $schema->dropIfExists($table);
         });
     }
 }
diff --git a/packages/graphql/src/Testing/Package/Data/Models/WithTestObject.php b/packages/graphql/src/Testing/Package/Data/Models/WithTestObject.php
@@ -12,7 +12,7 @@
 /**
  * @internal
  *
- * @mixin TestCase
+ * @phpstan-require-extends TestCase
  */
 trait WithTestObject {
     /**
@@ -44,11 +44,7 @@ protected function withTestObjectAfter(): void {
             $schema = Container::getInstance()->make(Builder::class);
             $table  = (new TestObject())->getTable();
 
-            if ($schema->hasTable($table)) {
-                return;
-            }
-
-            $schema->drop($table);
+            $schema->dropIfExists($table);
         });
     }
 }