tests added.

Author: N3XT0R

src/Service/Parser/SchemaParserInterface.php

@@ -3,6 +3,8 @@
 
 namespace N3XT0R\MigrationGenerator\Service\Parser;
 
+use Illuminate\Database\ConnectionInterface;
+
 
 interface SchemaParserInterface
 {
@@ -12,4 +14,9 @@ public function getSortedTablesFromSchema(string $schema): array;
 
     public function setConnectionByName(string $connectionName = ''): void;
 
+    public function setConnection(?ConnectionInterface $connection): void;
+
+    public function getConnection(): ConnectionInterface;
+    
+
 }

tests/Integration/Service/Parser/SchemaParserTest.php

@@ -5,6 +5,7 @@
 
 
 use Illuminate\Foundation\Application;
+use Illuminate\Support\Facades\DB;
 use N3XT0R\MigrationGenerator\Service\Parser\SchemaParserInterface;
 use Tests\DbTestCase;
 
@@ -48,7 +49,7 @@ public function testGetTablesFromSchemaWorks(): void
                 'users',
             ],
         };
-        
+
 
         self::assertSame($expectedTables, $tables);
     }
@@ -85,4 +86,26 @@ public function testGetSortedTablesFromSchema(): void
         self::assertCount(count($expectedTables), $tables);
         self::assertSame($expectedTables, $tables);
     }
+
+
+    public function testSetConnectionByNameWorksWithDefaultConnection(): void
+    {
+        $defaultConnection = DB::connection();
+
+        $this->parser->setConnectionByName();
+        $connection = $this->parser->getConnection();
+
+        self::assertSame($defaultConnection->getDatabaseName(), $connection->getDatabaseName());
+    }
+
+    public function testSetConnectionByNameWorksWithCustomConnection(): void
+    {
+        $connectionName = $this->getDatabaseFromEnv();
+        $customConnection = DB::connection($connectionName);
+
+        $this->parser->setConnectionByName($connectionName);
+        $connection = $this->parser->getConnection();
+
+        self::assertSame($customConnection->getDatabaseName(), $connection->getDatabaseName());
+    }
 }