return basic driver in basic client

neo4j-php · Sep 20, 2023 · 87a4d59 · 87a4d59
1 parent d8f8893
commit 87a4d59
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/Basic/Client.php b/src/Basic/Client.php
@@ -14,7 +14,6 @@
 namespace Laudis\Neo4j\Basic;
 
 use Laudis\Neo4j\Contracts\ClientInterface;
-use Laudis\Neo4j\Contracts\DriverInterface;
 use Laudis\Neo4j\Databags\Statement;
 use Laudis\Neo4j\Databags\SummarizedResult;
 use Laudis\Neo4j\Databags\TransactionConfiguration;
@@ -53,9 +52,14 @@ public function beginTransaction(?iterable $statements = null, ?string $alias =
         return new UnmanagedTransaction($this->client->beginTransaction($statements, $alias, $config));
     }
 
-    public function getDriver(?string $alias): DriverInterface
+    public function getDriver(?string $alias): Driver
     {
-        return $this->client->getDriver($alias);
+        $driver = $this->client->getDriver($alias);
+        if ($driver instanceof Driver) {
+            return $driver;
+        }
+
+        return new Driver($driver);
     }
 
     public function hasDriver(string $alias): bool