add static new() and factory() on ConnectionLocator

src/Connection.php

@@ -20,6 +20,10 @@ class Connection
 
     public static function new(...$args) : Connection
     {
+        if ($args[0] instanceof PDO) {
+            return new static($args[0]);
+        }
+
         return new static(new PDO(...$args));
     }
 

src/ConnectionLocator.php

@@ -36,6 +36,24 @@ class ConnectionLocator
 
     protected $lockToWrite = false;
 
+    public static function new(...$args)
+    {
+        if ($args[0] instanceof Connection) {
+            return new static(function () use ($args) {
+                return $args[0];
+            });
+        }
+
+        return new static(Connection::factory(...$args));
+    }
+
+    public static function factory(...$args)
+    {
+        return function () use ($args) {
+            return static::new(...$args);
+        };
+    }
+
     public function __construct(
         callable $default = null,
         array $read = [],

tests/ConnectionLocatorTest.php

@@ -1,6 +1,8 @@
 <?php
 namespace Atlas\Pdo;
 
+use PDO;
+
 class ConnectionLocatorTest extends \PHPUnit\Framework\TestCase
 {
     /**
@@ -47,6 +49,30 @@ protected function newLocator($read = [], $write = [])
         return new ConnectionLocator($this->default, $read, $write);
     }
 
+    public function testNewWithConnection()
+    {
+        $connection = Connection::new('sqlite::memory:');
+        $actual = ConnectionLocator::new($connection);
+        $this->assertInstanceOf(ConnectionLocator::CLASS, $actual);
+        $this->assertSame($connection, $actual->getDefault());
+    }
+
+    public function testNewWithPdo()
+    {
+        $pdo = new Pdo('sqlite::memory:');
+        $actual = ConnectionLocator::new($pdo);
+        $this->assertInstanceOf(ConnectionLocator::CLASS, $actual);
+        $this->assertSame($pdo, $actual->getDefault()->getPdo());
+    }
+
+    public function testFactory()
+    {
+        $factory = ConnectionLocator::factory('sqlite::memory:');
+        $actual = $factory();
+        $this->assertInstanceOf(ConnectionLocator::CLASS, $actual);
+        $this->assertInstanceOf(PDO::CLASS, $actual->getDefault()->getPdo());
+    }
+
     public function testGetDefault()
     {
         $locator = $this->newLocator();