Add unit test for dropping and truncating tables during schema load

src/TestSuite/Fixture/SchemaLoader.php

@@ -79,7 +79,7 @@ class SchemaLoader
         $files = (array)$files;
 
         // Don't create schema if we are in a phpunit separate process test method.
-        if (empty($files) || isset($GLOBALS['__PHPUNIT_BOOTSTRAP'])) {
+        if (isset($GLOBALS['__PHPUNIT_BOOTSTRAP'])) {
             return;
         }
 
@@ -95,10 +95,6 @@ class SchemaLoader
             }
 
             $sql = file_get_contents($file);
-            if ($sql === false) {
-                throw new InvalidArgumentException("Unable to load schema file `$file`.");
-            }
-
             $connection->execute($sql)->closeCursor();
         }
 

tests/TestCase/TestSuite/Fixture/SchemaLoaderTest.php

@@ -15,6 +15,11 @@ declare(strict_types=1);
  */
 namespace Cake\Test\TestCase\TestSuite\Fixture;
 
+use Cake\Console\ConsoleIo;
+use Cake\Database\Connection;
+use Cake\Database\DriverInterface;
+use Cake\Database\Schema\Collection;
+use Cake\Database\Schema\SchemaDialect;
 use Cake\Database\Schema\TableSchema;
 use Cake\Datasource\ConnectionManager;
 use Cake\TestSuite\Fixture\SchemaCleaner;
@@ -29,20 +34,32 @@ class SchemaLoaderTest extends TestCase
      */
     protected $restore;
 
+    /**
+     * @var \Cake\TestSuite\Fixture\SchemaLoader
+     */
+    protected $loader;
+
     public function setUp(): void
     {
         parent::setUp();
         $this->restore = $GLOBALS['__PHPUNIT_BOOTSTRAP'];
         unset($GLOBALS['__PHPUNIT_BOOTSTRAP']);
+
+        $this->loader = new SchemaLoader(['outputLevel' => ConsoleIo::QUIET]);
     }
 
     public function tearDown(): void
     {
         parent::tearDown();
         $GLOBALS['__PHPUNIT_BOOTSTRAP'] = $this->restore;
+
         (new SchemaCleaner())->dropTables('test', ['schema_loader_test_one', 'schema_loader_test_two']);
+        ConnectionManager::drop('schema_test');
     }
 
+    /**
+     * Tests loading schema files.
+     */
     public function testLoadingFiles(): void
     {
         $connection = ConnectionManager::get('test');
@@ -50,7 +67,7 @@ class SchemaLoaderTest extends TestCase
         $schemaFiles[] = $this->createSchemaFile('schema_loader_test_one');
         $schemaFiles[] = $this->createSchemaFile('schema_loader_test_two');
 
-        (new SchemaLoader())->loadFiles($schemaFiles, 'test', false, false);
+        $this->loader->loadFiles($schemaFiles, 'test', false, false);
 
         $connection = ConnectionManager::get('test');
         $tables = $connection->getSchemaCollection()->listTables();
@@ -58,10 +75,46 @@ class SchemaLoaderTest extends TestCase
         $this->assertContains('schema_loader_test_two', $tables);
     }
 
+    /**
+     * Tests loading missing files.
+     */
     public function testLoadMissingFile(): void
     {
         $this->expectException(InvalidArgumentException::class);
-        (new SchemaLoader())->loadFiles('missing_schema_file.sql', 'test', false, false);
+        $this->loader->loadFiles('missing_schema_file.sql', 'test', false, false);
+    }
+
+    /**
+     * Tests dropping and truncating tables during schema load.
+     */
+    public function testDropTruncateTables(): void
+    {
+        $connection = $this->createMock(Connection::class);
+
+        $tableSchema = $this->createMock(TableSchema::class);
+        $schemaDialect = $this->createMock(SchemaDialect::class);
+        $schemaDialect
+            ->expects($this->atLeastOnce())->method('dropConstraintSql')->with($tableSchema)->willReturn(['']);
+        $schemaDialect
+            ->expects($this->atLeastOnce())->method('dropTableSql')->with($tableSchema)->willReturn(['']);
+        $schemaDialect
+            ->expects($this->atLeastOnce())->method('truncateTableSql')->with($tableSchema)->willReturn(['']);
+        $driver = $this->createMock(DriverInterface::class);
+        $driver
+            ->expects($this->atLeastOnce())->method('schemaDialect')->willReturn($schemaDialect);
+        $connection
+            ->expects($this->atLeastOnce())->method('getDriver')->willReturn($driver);
+
+        $schemaCollection = $this->createMock(Collection::class);
+        $schemaCollection
+            ->expects($this->atLeastOnce())->method('listTables')->willReturn(['schema_test']);
+        $schemaCollection
+            ->expects($this->atLeastOnce())->method('describe')->willReturn($tableSchema);
+        $connection
+            ->expects($this->atLeastOnce())->method('getSchemaCollection')->willReturn($schemaCollection);
+
+        ConnectionManager::setConfig('schema_test', $connection);
+        $this->loader->loadFiles([], 'schema_test', true, true);
     }
 
     protected function createSchemaFile(string $tableName): string