From e7d2f0b5baa5f979f76d9a04169a9377cf1d3a4e Mon Sep 17 00:00:00 2001
From: Sergei Tigrov <rrr-r@ya.ru>
Date: Mon, 16 Sep 2024 17:51:46 +0700
Subject: [PATCH] Realize `ColumnBuilder` (#318)

---
 CHANGELOG.md                 |  1 +
 src/Column/ColumnBuilder.php |  9 +++++++++
 src/Column/ColumnFactory.php |  5 +++++
 src/Connection.php           |  7 +++++++
 src/Schema.php               | 11 +++--------
 tests/ColumnBuilderTest.php  | 22 ++++++++++++++++++++++
 tests/ColumnFactoryTest.php  | 20 +++++++++++++++++---
 tests/ConnectionTest.php     |  8 ++++++++
 tests/SchemaTest.php         |  9 ---------
 9 files changed, 72 insertions(+), 20 deletions(-)
 create mode 100644 src/Column/ColumnBuilder.php
 create mode 100644 tests/ColumnBuilderTest.php

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 69152211..cc3a3236 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -11,6 +11,7 @@
 - Enh #315: Raise minimum PHP version to `^8.1` with minor refactoring (@Tigrov)
 - Enh #314: Implement `ColumnFactory` class (@Tigrov)
 - Enh #317: Separate column type constants (@Tigrov)
+- Enh #318: Realize `ColumnBuilder` class (@Tigrov)
 
 ## 1.2.0 March 21, 2024
 
diff --git a/src/Column/ColumnBuilder.php b/src/Column/ColumnBuilder.php
new file mode 100644
index 00000000..b0476207
--- /dev/null
+++ b/src/Column/ColumnBuilder.php
@@ -0,0 +1,9 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite\Column;
+
+final class ColumnBuilder extends \Yiisoft\Db\Schema\Column\ColumnBuilder
+{
+}
diff --git a/src/Column/ColumnFactory.php b/src/Column/ColumnFactory.php
index 96ab9d18..42f40eb1 100644
--- a/src/Column/ColumnFactory.php
+++ b/src/Column/ColumnFactory.php
@@ -62,4 +62,9 @@ protected function getType(string $dbType, array $info = []): string
 
         return $type;
     }
+
+    protected function isDbType(string $dbType): bool
+    {
+        return isset(self::TYPE_MAP[$dbType]);
+    }
 }
diff --git a/src/Connection.php b/src/Connection.php
index 14d209af..c1308d87 100644
--- a/src/Connection.php
+++ b/src/Connection.php
@@ -7,9 +7,11 @@
 use Yiisoft\Db\Driver\Pdo\AbstractPdoConnection;
 use Yiisoft\Db\Driver\Pdo\PdoCommandInterface;
 use Yiisoft\Db\QueryBuilder\QueryBuilderInterface;
+use Yiisoft\Db\Schema\Column\ColumnFactoryInterface;
 use Yiisoft\Db\Schema\Quoter;
 use Yiisoft\Db\Schema\QuoterInterface;
 use Yiisoft\Db\Schema\SchemaInterface;
+use Yiisoft\Db\Sqlite\Column\ColumnFactory;
 use Yiisoft\Db\Transaction\TransactionInterface;
 
 use function str_starts_with;
@@ -70,6 +72,11 @@ public function getQueryBuilder(): QueryBuilderInterface
         return $this->queryBuilder;
     }
 
+    public function getColumnFactory(): ColumnFactoryInterface
+    {
+        return new ColumnFactory();
+    }
+
     public function getQuoter(): QuoterInterface
     {
         if ($this->quoter === null) {
diff --git a/src/Schema.php b/src/Schema.php
index 29a321c1..67752a2a 100644
--- a/src/Schema.php
+++ b/src/Schema.php
@@ -18,10 +18,8 @@
 use Yiisoft\Db\Expression\Expression;
 use Yiisoft\Db\Helper\DbArrayHelper;
 use Yiisoft\Db\Schema\Builder\ColumnInterface;
-use Yiisoft\Db\Schema\Column\ColumnFactoryInterface;
 use Yiisoft\Db\Schema\Column\ColumnSchemaInterface;
 use Yiisoft\Db\Schema\TableSchemaInterface;
-use Yiisoft\Db\Sqlite\Column\ColumnFactory;
 
 use function array_change_key_case;
 use function array_column;
@@ -81,11 +79,6 @@ public function createColumn(string $type, array|int|string $length = null): Col
         return new Column($type, $length);
     }
 
-    public function getColumnFactory(): ColumnFactoryInterface
-    {
-        return new ColumnFactory();
-    }
-
     /**
      * Returns all table names in the database.
      *
@@ -445,8 +438,10 @@ public function getSchemaDefaultValues(string $schema = '', bool $refresh = fals
      */
     private function loadColumnSchema(array $info): ColumnSchemaInterface
     {
+        $columnFactory = $this->db->getColumnFactory();
+
         $dbType = strtolower($info['type']);
-        $column = $this->getColumnFactory()->fromDefinition($dbType, ['name' => $info['name']]);
+        $column = $columnFactory->fromDefinition($dbType, ['name' => $info['name']]);
         $column->dbType($dbType);
         $column->allowNull(!$info['notnull']);
         $column->primaryKey((bool) $info['pk']);
diff --git a/tests/ColumnBuilderTest.php b/tests/ColumnBuilderTest.php
new file mode 100644
index 00000000..61f45283
--- /dev/null
+++ b/tests/ColumnBuilderTest.php
@@ -0,0 +1,22 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Yiisoft\Db\Sqlite\Tests;
+
+use Yiisoft\Db\Sqlite\Column\ColumnBuilder;
+use Yiisoft\Db\Sqlite\Tests\Support\TestTrait;
+use Yiisoft\Db\Tests\AbstractColumnBuilderTest;
+
+/**
+ * @group sqlite
+ */
+class ColumnBuilderTest extends AbstractColumnBuilderTest
+{
+    use TestTrait;
+
+    public function getColumnBuilderClass(): string
+    {
+        return ColumnBuilder::class;
+    }
+}
diff --git a/tests/ColumnFactoryTest.php b/tests/ColumnFactoryTest.php
index a9b02608..6c6dfd83 100644
--- a/tests/ColumnFactoryTest.php
+++ b/tests/ColumnFactoryTest.php
@@ -21,9 +21,23 @@ public function testFromDbType(string $dbType, string $expectedType, string $exp
     }
 
     /** @dataProvider \Yiisoft\Db\Sqlite\Tests\Provider\ColumnFactoryProvider::definitions */
-    public function testFromDefinition(string $definition, string $expectedType, string $expectedInstanceOf, array $expectedInfo = []): void
-    {
-        parent::testFromDefinition($definition, $expectedType, $expectedInstanceOf, $expectedInfo);
+    public function testFromDefinition(
+        string $definition,
+        string $expectedType,
+        string $expectedInstanceOf,
+        array $expectedMethodResults = []
+    ): void {
+        parent::testFromDefinition($definition, $expectedType, $expectedInstanceOf, $expectedMethodResults);
+    }
+
+    /** @dataProvider \Yiisoft\Db\Sqlite\Tests\Provider\ColumnFactoryProvider::pseudoTypes */
+    public function testFromPseudoType(
+        string $pseudoType,
+        string $expectedType,
+        string $expectedInstanceOf,
+        array $expectedMethodResults = []
+    ): void {
+        parent::testFromPseudoType($pseudoType, $expectedType, $expectedInstanceOf, $expectedMethodResults);
     }
 
     /** @dataProvider \Yiisoft\Db\Sqlite\Tests\Provider\ColumnFactoryProvider::types */
diff --git a/tests/ConnectionTest.php b/tests/ConnectionTest.php
index c6472935..83371fc2 100644
--- a/tests/ConnectionTest.php
+++ b/tests/ConnectionTest.php
@@ -14,6 +14,7 @@
 use Yiisoft\Db\Exception\InvalidConfigException;
 use Yiisoft\Db\Exception\NotSupportedException;
 use Yiisoft\Db\Profiler\ProfilerInterface;
+use Yiisoft\Db\Sqlite\Column\ColumnFactory;
 use Yiisoft\Db\Sqlite\Tests\Support\TestTrait;
 use Yiisoft\Db\Tests\Common\CommonConnectionTest;
 use Yiisoft\Db\Transaction\TransactionInterface;
@@ -181,6 +182,13 @@ private function runExceptionTest(ConnectionInterface $db): void
         $this->assertTrue($thrown, 'An exception should have been thrown by the command.');
     }
 
+    public function testGetColumnFactory(): void
+    {
+        $db = $this->getConnection();
+
+        $this->assertInstanceOf(ColumnFactory::class, $db->getColumnFactory());
+    }
+
     private function createProfiler(): ProfilerInterface
     {
         return $this->createMock(ProfilerInterface::class);
diff --git a/tests/SchemaTest.php b/tests/SchemaTest.php
index 758383d2..2a171127 100644
--- a/tests/SchemaTest.php
+++ b/tests/SchemaTest.php
@@ -11,7 +11,6 @@
 use Yiisoft\Db\Exception\Exception;
 use Yiisoft\Db\Exception\InvalidConfigException;
 use Yiisoft\Db\Exception\NotSupportedException;
-use Yiisoft\Db\Sqlite\Column\ColumnFactory;
 use Yiisoft\Db\Sqlite\Schema;
 use Yiisoft\Db\Sqlite\Tests\Support\TestTrait;
 use Yiisoft\Db\Tests\Common\CommonSchemaTest;
@@ -360,12 +359,4 @@ public function testNotConnectionPDO(): void
 
         $schema->refresh();
     }
-
-    public function testGetColumnFactory(): void
-    {
-        $db = $this->getConnection();
-        $factory = $db->getSchema()->getColumnFactory();
-
-        $this->assertInstanceOf(ColumnFactory::class, $factory);
-    }
 }