add public connection() method to the Tenant DB manager interface

src/Database/Contracts/StatefulTenantDatabaseManager.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Stancl\Tenancy\Database\Contracts;
+
+use Illuminate\Database\Connection;
+use Stancl\Tenancy\Database\Exceptions\NoConnectionSetException;
+
+/**
+ * Tenant database manager with a persistent connection.
+ */
+interface StatefulTenantDatabaseManager extends TenantDatabaseManager
+{
+    /** Get the DB connection used by the tenant database manager. */
+    public function database(): Connection; // todo rename to connection()
+
+    /**
+     * Set the DB connection that should be used by the tenant database manager.
+     *
+     * @throws NoConnectionSetException
+     */
+    public function setConnection(string $connection): void;
+}

src/Database/Contracts/TenantDatabaseManager.php

@@ -4,8 +4,6 @@ declare(strict_types=1);
 
 namespace Stancl\Tenancy\Database\Contracts;
 
-use Stancl\Tenancy\Database\Exceptions\NoConnectionSetException;
-
 interface TenantDatabaseManager
 {
     /** Create a database. */
@@ -19,11 +17,4 @@ interface TenantDatabaseManager
 
     /** Construct a DB connection config array. */
     public function makeConnectionConfig(array $baseConfig, string $databaseName): array;
-
-    /**
-     * Set the DB connection that should be used by the tenant database manager.
-     *
-     * @throws NoConnectionSetException
-     */
-    public function setConnection(string $connection): void;
 }

src/Database/DatabaseConfig.php

@@ -154,7 +154,9 @@ class DatabaseConfig
         /** @var Contracts\TenantDatabaseManager $databaseManager */
         $databaseManager = app($databaseManagers[$driver]);
 
-        $databaseManager->setConnection($this->getTemplateConnectionName());
+        if ($databaseManager instanceof Contracts\StatefulTenantDatabaseManager) {
+            $databaseManager->setConnection($this->getTemplateConnectionName());
+        }
 
         return $databaseManager;
     }

src/Database/TenantDatabaseManagers/TenantDatabaseManager.php

@@ -6,15 +6,15 @@ namespace Stancl\Tenancy\Database\TenantDatabaseManagers;
 
 use Illuminate\Database\Connection;
 use Illuminate\Support\Facades\DB;
-use Stancl\Tenancy\Database\Contracts\TenantDatabaseManager as Contract;
+use Stancl\Tenancy\Database\Contracts\StatefulTenantDatabaseManager;
 use Stancl\Tenancy\Database\Exceptions\NoConnectionSetException;
 
-abstract class TenantDatabaseManager implements Contract // todo better naming?
+abstract class TenantDatabaseManager implements StatefulTenantDatabaseManager
 {
     /** The database connection to the server. */
     protected string $connection;
 
-    protected function database(): Connection
+    public function database(): Connection
     {
         if (! isset($this->connection)) {
             throw new NoConnectionSetException(static::class);

tests/DatabasePreparationTest.php

@@ -22,9 +22,7 @@ test('database can be created after tenant creation', function () {
     })->toListener());
 
     $tenant = Tenant::create();
-
-    $manager = app(MySQLDatabaseManager::class);
-    $manager->setConnection('mysql');
+    $manager = $tenant->database()->manager();
 
     expect($manager->databaseExists($tenant->database()->getName()))->toBeTrue();
 });

tests/TenantDatabaseManagerTest.php

@@ -8,6 +8,7 @@ use Illuminate\Support\Facades\Schema;
 use Illuminate\Support\Str;
 use Stancl\JobPipeline\JobPipeline;
 use Stancl\Tenancy\Bootstrappers\DatabaseTenancyBootstrapper;
+use Stancl\Tenancy\Database\Contracts\StatefulTenantDatabaseManager;
 use Stancl\Tenancy\Database\DatabaseManager;
 use Stancl\Tenancy\Events\TenancyEnded;
 use Stancl\Tenancy\Events\TenancyInitialized;
@@ -36,7 +37,10 @@ test('databases can be created and deleted', function ($driver, $databaseManager
     $name = 'db' . pest()->randomString();
 
     $manager = app($databaseManager);
-    $manager->setConnection($driver);
+
+    if ($manager instanceof StatefulTenantDatabaseManager) {
+        $manager->setConnection($driver);
+    }
 
     expect($manager->databaseExists($name))->toBeFalse();