From 025aa1b3ba87450bd662fe6a46704651c242aead Mon Sep 17 00:00:00 2001
From: Juan Pablo Ramirez <pabloelcolombiano@gmail.com>
Date: Sat, 13 Mar 2021 00:48:20 +0100
Subject: [PATCH] Alias connections before running migrations

---
 src/Migrator.php                              |  6 +++++-
 .../20200208100000_app_migration.php          | 11 +++++++++-
 .../TestApp/src/Model/Table/ArticlesTable.php | 20 +++++++++++++++++++
 tests/TestCase/MigratorTest.php               | 10 +++++++++-
 tests/bootstrap.php                           | 11 +++++++---
 5 files changed, 52 insertions(+), 6 deletions(-)
 create mode 100644 tests/TestApp/src/Model/Table/ArticlesTable.php

diff --git a/src/Migrator.php b/src/Migrator.php
index 45caf51..db6232b 100644
--- a/src/Migrator.php
+++ b/src/Migrator.php
@@ -46,6 +46,10 @@ public static function migrate(array $config = []): void
     {
         $migrator = new static();
 
+        // Make sure that the connections are aliased, in case
+        // the migrations invoke the table registry.
+        $migrator->getFixtureManager()->aliasConnections();
+
         $migrator
             ->prepareConfig($config)
             ->dropTablesForMissingMigrations()
@@ -129,4 +133,4 @@ protected function getFixtureManager(): FixtureManager
     {
         return $this->fixtureManager;
     }
-}
\ No newline at end of file
+}
\ No newline at end of file
diff --git a/tests/TestApp/config/Migrations/20200208100000_app_migration.php b/tests/TestApp/config/Migrations/20200208100000_app_migration.php
index 60ef3f5..b734344 100644
--- a/tests/TestApp/config/Migrations/20200208100000_app_migration.php
+++ b/tests/TestApp/config/Migrations/20200208100000_app_migration.php
@@ -17,7 +17,16 @@
 class AppMigration extends AbstractMigration
 {
     public function up()
-    {}
+    {
+        $this->table('articles')
+            ->addPrimaryKey(['id'])
+            ->addColumn('title', 'string', [
+                'limit' => 128,
+                'null' => false,
+            ])
+            ->addTimestamps('created', 'modified')
+            ->create();
+    }
 
     public function down()
     {}
diff --git a/tests/TestApp/src/Model/Table/ArticlesTable.php b/tests/TestApp/src/Model/Table/ArticlesTable.php
new file mode 100644
index 0000000..5316d8d
--- /dev/null
+++ b/tests/TestApp/src/Model/Table/ArticlesTable.php
@@ -0,0 +1,20 @@
+<?php
+declare(strict_types=1);
+
+/**
+ * Licensed under The MIT License
+ * For full copyright and license information, please see the LICENSE.txt
+ * Redistributions of files must retain the above copyright notice.
+ *
+ * @copyright     Copyright (c) 2020 Juan Pablo Ramirez and Nicolas Masson
+ * @link          https://webrider.de/
+ * @since         1.0.0
+ * @license       http://www.opensource.org/licenses/mit-license.php MIT License
+ */
+namespace TestApp\Model\Table;
+
+use Cake\ORM\Query;
+use Cake\ORM\Table;
+
+class ArticlesTable extends Table
+{}
diff --git a/tests/TestCase/MigratorTest.php b/tests/TestCase/MigratorTest.php
index 813f6de..0f2020c 100644
--- a/tests/TestCase/MigratorTest.php
+++ b/tests/TestCase/MigratorTest.php
@@ -15,6 +15,7 @@
 
 
 use Cake\Datasource\ConnectionManager;
+use Cake\ORM\TableRegistry;
 use Cake\TestSuite\TestCase;
 use CakephpTestMigrator\Migrator;
 
@@ -59,4 +60,11 @@ public function testMigrate()
         $this->assertSame(['FooMigration'], $fooPluginMigrations);
         $this->assertSame(['BarMigration'], $barPluginMigrations);
     }
-}
\ No newline at end of file
+
+    public function testTableRegistryConnectionName()
+    {
+        $Articles = TableRegistry::getTableLocator()->get('Articles');
+        ConnectionManager::getConfigOrFail('default');
+        $this->assertSame('test', $Articles->getConnection()->configName());
+    }
+}
diff --git a/tests/bootstrap.php b/tests/bootstrap.php
index 2605adb..fc483e6 100644
--- a/tests/bootstrap.php
+++ b/tests/bootstrap.php
@@ -1,6 +1,7 @@
 <?php
 
 use Cake\Core\Configure;
+use Cake\Datasource\ConnectionManager;
 
 if (!defined('DS')) {
     define('DS', DIRECTORY_SEPARATOR);
@@ -67,10 +68,14 @@
     ],
 ];
 
-\Cake\Datasource\ConnectionManager::setConfig('test', $dbConnection);
+ConnectionManager::setConfig('test', $dbConnection);
+
+$dbDefaultConnection = $dbConnection;
+$dbDefaultConnection['database'] = 'migrator';
+ConnectionManager::setConfig('default', $dbConnection);
 
 $dbConnection['migrations'] = ['plugin' => 'BarPlugin'];
-\Cake\Datasource\ConnectionManager::setConfig('test_2', $dbConnection);
+ConnectionManager::setConfig('test_2', $dbConnection);
 
 $dbConnection['migrations'] = true;
-\Cake\Datasource\ConnectionManager::setConfig('test_3', $dbConnection);
+ConnectionManager::setConfig('test_3', $dbConnection);