Add Connection::getColumnBuilderClass() (#421)

Author: Tigrov

CHANGELOG.md

@@ -49,6 +49,7 @@
 - Enh #407, #408, #411: Adapt to conditions refactoring in `yiisoft/db` package (@vjik)
 - Enh #414: Remove `TableSchema` class and refactor `Schema` class (@Tigrov)
 - Enh #415: Support column's collation (@Tigrov)
+- New #421: Add `Connection::getColumnBuilderClass()` method (@Tigrov)
 
 ## 1.2.0 March 21, 2024
 

src/Connection.php

@@ -9,6 +9,7 @@
 use Yiisoft\Db\Connection\ServerInfoInterface;
 use Yiisoft\Db\Driver\Pdo\AbstractPdoConnection;
 use Yiisoft\Db\Driver\Pdo\PdoCommandInterface;
+use Yiisoft\Db\Mysql\Column\ColumnBuilder;
 use Yiisoft\Db\Mysql\Column\ColumnFactory;
 use Yiisoft\Db\QueryBuilder\QueryBuilderInterface;
 use Yiisoft\Db\Schema\Column\ColumnFactoryInterface;
@@ -66,6 +67,11 @@ public function createTransaction(): TransactionInterface
         return new Transaction($this);
     }
 
+    public function getColumnBuilderClass(): string
+    {
+        return ColumnBuilder::class;
+    }
+
     public function getColumnFactory(): ColumnFactoryInterface
     {
         return $this->columnFactory ??= new ColumnFactory();

tests/ColumnBuilderTest.php

@@ -5,7 +5,6 @@
 namespace Yiisoft\Db\Mysql\Tests;
 
 use PHPUnit\Framework\Attributes\DataProviderExternal;
-use Yiisoft\Db\Mysql\Column\ColumnBuilder;
 use Yiisoft\Db\Mysql\Tests\Provider\ColumnBuilderProvider;
 use Yiisoft\Db\Mysql\Tests\Support\TestTrait;
 use Yiisoft\Db\Tests\AbstractColumnBuilderTest;
@@ -17,11 +16,6 @@ class ColumnBuilderTest extends AbstractColumnBuilderTest
 {
     use TestTrait;
 
-    public function getColumnBuilderClass(): string
-    {
-        return ColumnBuilder::class;
-    }
-
     #[DataProviderExternal(ColumnBuilderProvider::class, 'buildingMethods')]
     public function testBuildingMethods(
         string $buildingMethod,

tests/ColumnTest.php

@@ -32,8 +32,6 @@ final class ColumnTest extends CommonColumnTest
 {
     use TestTrait;
 
-    protected const COLUMN_BUILDER = ColumnBuilder::class;
-
     protected function insertTypeValues(ConnectionInterface $db): void
     {
         $db->createCommand()->insert(

tests/ConnectionTest.php

@@ -5,12 +5,9 @@
 namespace Yiisoft\Db\Mysql\Tests;
 
 use PDO;
-use Throwable;
 use Yiisoft\Db\Connection\ConnectionInterface;
-use Yiisoft\Db\Exception\Exception;
 use Yiisoft\Db\Exception\IntegrityException;
-use Yiisoft\Db\Exception\InvalidConfigException;
-use Yiisoft\Db\Exception\NotSupportedException;
+use Yiisoft\Db\Mysql\Column\ColumnBuilder;
 use Yiisoft\Db\Mysql\Column\ColumnFactory;
 use Yiisoft\Db\Mysql\Connection;
 use Yiisoft\Db\Mysql\Tests\Support\TestTrait;
@@ -20,17 +17,11 @@
 
 /**
  * @group mysql
- *
- * @psalm-suppress PropertyNotSetInConstructor
  */
 final class ConnectionTest extends CommonConnectionTest
 {
     use TestTrait;
 
-    /**
-     * @throws Exception
-     * @throws InvalidConfigException
-     */
     public function testInitConnection(): void
     {
         $db = $this->getConnection();
@@ -47,10 +38,6 @@ public function testInitConnection(): void
         $db->close();
     }
 
-    /**
-     * @throws Exception
-     * @throws InvalidConfigException
-     */
     public function testSettingDefaultAttributes(): void
     {
         $db = $this->getConnection();
@@ -72,12 +59,6 @@ public function testSettingDefaultAttributes(): void
         $db->close();
     }
 
-    /**
-     * @throws Exception
-     * @throws InvalidConfigException
-     * @throws NotSupportedException
-     * @throws Throwable
-     */
     public function testTransactionIsolation(): void
     {
         $db = $this->getConnection(true);
@@ -100,11 +81,6 @@ public function testTransactionIsolation(): void
         $db->close();
     }
 
-    /**
-     * @throws Exception
-     * @throws InvalidConfigException
-     * @throws Throwable
-     */
     public function testTransactionShortcutCustom(): void
     {
         $db = $this->getConnection(true);
@@ -162,6 +138,15 @@ public function testNotRestartConnectionOnTimeoutInTransaction(): void
         $db->close();
     }
 
+    public function getColumnBuilderClass(): void
+    {
+        $db = $this->getConnection();
+
+        $this->assertSame(ColumnBuilder::class, $db->getColumnBuilderClass());
+
+        $db->close();
+    }
+
     public function testGetColumnFactory(): void
     {
         $db = $this->getConnection();