Expand DB test case to accept multiple migrations & seeds

Author: MGatner

system/Test/CIDatabaseTestCase.php

@@ -52,7 +52,6 @@
  */
 class CIDatabaseTestCase extends CIUnitTestCase
 {
-
 	/**
 	 * Should the db be refreshed before
 	 * each test?
@@ -62,25 +61,28 @@ class CIDatabaseTestCase extends CIUnitTestCase
 	protected $refresh = true;
 
 	/**
-	 * The name of the fixture used for all tests
-	 * within this test case.
+	 * The seed file(s) used for all tests within this test case.
+	 * Should be fully-namespaced or relative to $basePath
 	 *
-	 * @var string
+	 * @var string|array
 	 */
 	protected $seed = '';
 
 	/**
-	 * The path to where we can find the seeds directory.
+	 * The path to the seeds directory.
 	 * Allows overriding the default application directories.
 	 *
 	 * @var string
 	 */
-	protected $basePath = TESTPATH . '_support/Database';
+	protected $basePath = SUPPORTPATH . 'Database';
 
 	/**
-	 * The namespace to help us find the migration classes.
+	 * The namespace(s) to help us find the migration classes.
+	 * Empty is equivalent to running `spark migrate -all`.
+	 * Note that running "all" runs migrations in date order,
+	 * but specifying namespaces runs them in namespace order (then date)
 	 *
-	 * @var string
+	 * @var string|array|null
 	 */
 	protected $namespace = 'Tests\Support';
 
@@ -163,19 +165,10 @@ protected function setUp(): void
 	{
 		parent::setUp();
 
-		// Add namespaces we need for testing
-		Services::autoloader()->addNamespace('Tests\Support\DatabaseTestMigrations', TESTPATH . '_support/DatabaseTestMigrations');
-
 		$this->loadDependencies();
 
 		if ($this->refresh === true)
 		{
-			if (! empty($this->namespace))
-			{
-				$this->migrations->setNamespace($this->namespace);
-			}
-			$this->migrations->regress(0, 'tests');
-
 			// Delete all of the tables to ensure we're at a clean start.
 			$tables = $this->db->listTables();
 
@@ -187,14 +180,31 @@ protected function setUp(): void
 				{
 					if ($table === $this->db->DBPrefix . 'migrations')
 					{
+						$this->db->table($table)->truncate();
 						continue;
 					}
 
 					$forge->dropTable($table, true);
 				}
 			}
 
-			$this->migrations->latest('tests');
+			// If no namespace was specified then migrate all
+			if (empty($this->namespace))
+			{
+				$this->migrations->setNamespace(null);
+				$this->migrations->latest('tests');
+			}
+
+			// Run migrations for each specified namespace
+			else
+			{
+				$namespaces = is_array($this->namespace) ? $this->namespace : [$this->namespace];
+				foreach ($namespaces as $namespace)
+				{
+					$this->migrations->setNamespace($namespace);
+					$this->migrations->latest('tests');
+				}
+			}
 		}
 
 		if (! empty($this->seed))
@@ -204,7 +214,11 @@ protected function setUp(): void
 				$this->seeder->setPath(rtrim($this->basePath, '/') . '/Seeds');
 			}
 
-			$this->seed($this->seed);
+			$seeds = is_array($this->seed) ? $this->seed : [$this->seed];
+			foreach ($seeds as $seed)
+			{
+				$this->seed($seed);
+			}
 		}
 	}
 
@@ -239,7 +253,6 @@ public function seed(string $name)
 		return $this->seeder->call($name);
 	}
 
-	//--------------------------------------------------------------------
 	//--------------------------------------------------------------------
 	// Database Test Helpers
 	//--------------------------------------------------------------------

tests/_support/Database/Seeds/AnotherSeeder.php

@@ -0,0 +1,15 @@
+<?php namespace Tests\Support\Database\Seeds;
+
+class AnotherSeeder extends \CodeIgniter\Database\Seeder
+{
+	public function run()
+	{
+		$row = [
+			'name'    => 'Jerome Lohan',
+			'email'   => 'jlo@lohanenterprises.com',
+			'country' => 'UK',
+		];
+
+		$this->db->table('user')->insert($row);
+	}
+}

tests/system/Database/DatabaseTestCaseTest.php

@@ -0,0 +1,62 @@
+<?php namespace CodeIgniter\Database;
+
+use Config\Services;
+use CodeIgniter\Test\CIDatabaseTestCase;
+
+class DatabaseTestCaseTest extends CIDatabaseTestCase
+{
+	protected $loaded = false;
+
+	/**
+	 * Should the db be refreshed before
+	 * each test?
+	 *
+	 * @var boolean
+	 */
+	protected $refresh = true;
+
+	/**
+	 * The seed file(s) used for all tests within this test case.
+	 * Should be fully-namespaced or relative to $basePath
+	 *
+	 * @var string|array
+	 */
+	protected $seed = [
+		'Tests\Support\Database\Seeds\CITestSeeder',
+		'Tests\Support\Database\Seeds\AnotherSeeder'
+	];
+
+	/**
+	 * The namespace(s) to help us find the migration classes.
+	 * Empty is equivalent to running `spark migrate -all`.
+	 * Note that running "all" runs migrations in date order,
+	 * but specifying namespaces runs them in namespace order (then date)
+	 *
+	 * @var string|array|null
+	 */
+	protected $namespace = [
+		'Tests\Support',
+		'Tests\Support\MigrationTestMigrations',
+	];
+
+	public function setUp(): void
+	{
+		if (! $this->loaded)
+		{
+			Services::autoloader()->addNamespace('Tests\Support\MigrationTestMigrations', SUPPORTPATH . 'MigrationTestMigrations');
+			$this->loaded = true;
+		}
+
+		parent::setUp();
+	}
+
+	public function testMultipleSeeders()
+	{
+		$this->seeInDatabase('user', ['name' => 'Jerome Lohan']);
+	}
+
+	public function testMultipleMigrationNamespaces()
+	{
+		$this->seeInDatabase('foo', ['key' => 'foobar']);
+	}
+}