fix: tests

vehikl · Nov 21, 2024 · d8d44d0 · d8d44d0
1 parent 308f706
commit d8d44d0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/Integration/Services/Databases/DatabaseManagementServiceTest.php b/tests/Integration/Services/Databases/DatabaseManagementServiceTest.php
@@ -10,6 +10,7 @@
 use App\Exceptions\Repository\DuplicateDatabaseNameException;
 use App\Exceptions\Service\Database\TooManyDatabasesException;
 use App\Exceptions\Service\Database\DatabaseClientFeatureNotEnabledException;
+use Database\Factories\NodeFactory;
 
 class DatabaseManagementServiceTest extends IntegrationTestCase
 {
@@ -118,7 +119,7 @@ public function testServerDatabaseCanBeCreated(): void
         $server = $this->createServerModel();
         $name = DatabaseManagementService::generateUniqueDatabaseName('something', $server->id);
 
-        $host = DatabaseHost::factory()->create(['node_id' => $server->node_id]);
+        $host = DatabaseHost::factory()->recycle($server->node)->create();
 
         $username = null;
         $secondUsername = null;
@@ -139,7 +140,7 @@ public function testServerDatabaseCanBeCreated(): void
         $this->assertDatabaseHas('databases', ['server_id' => $server->id, 'id' => $response->id]);
     }
 
-    /**
+/**
      * Test that an exception encountered while creating the database leads to the cleanup code
      * being called and any exceptions encountered while cleaning up go unreported.
      */
@@ -155,7 +156,7 @@ public function testExceptionEncounteredWhileCreatingDatabaseAttemptsToCleanup()
         $server = $this->createServerModel();
         $name = DatabaseManagementService::generateUniqueDatabaseName('something', $server->id);
 
-        $host = DatabaseHost::factory()->create(['node_id' => $server->node_id]);
+        $host = DatabaseHost::factory()->recycle($server->node)->create();
 
         $this->repository->expects('createDatabase')->with($name)->andThrows(new \BadMethodCallException());
         $this->repository->expects('dropDatabase')->with($name);