CustomDumpTrait. Позволяет задать собственный дамп БД для тестов.

src/Base/BitrixableTestCase.php

@@ -2,8 +2,10 @@
 
 namespace Prokl\BitrixTestingTools\Base;
 
+use Prokl\BitrixTestingTools\Migrator;
 use Prokl\BitrixTestingTools\Helpers\ClassUtils;
 use Prokl\BitrixTestingTools\Helpers\Database;
+use Prokl\BitrixTestingTools\Traits\CustomDumpTrait;
 use Prokl\BitrixTestingTools\Traits\ResetDatabaseTrait;
 use Prokl\TestingTools\Base\BaseTestCase;
 use Sheerockoff\BitrixCi\Bootstrap;
@@ -33,12 +35,12 @@ protected function setUp(): void
         if ($this->dropBase) {
             $dbManager->dropBase();
             $dbManager->createDatabaseIfNotExist();
-            Bootstrap::migrate();
+            $this->migrateDatabase();
         } else {
             $dbManager->createDatabaseIfNotExist();
 
             if ($dbManager->hasEmptyBase()) {
-                Bootstrap::migrate();
+                $this->migrateDatabase();
             }
         }
 
@@ -52,14 +54,7 @@ protected function tearDown(): void
     {
         parent::tearDown();
 
-        // Битриксовые штучки-дрючки с буфером.
-        while (ob_get_level()) {
-            ob_end_clean();
-        }
-
-        if ($GLOBALS['APPLICATION']) {
-            $GLOBALS['APPLICATION']->RestartBuffer();
-        }
+        $this->clearBitrixBuffer();
 
         if ($this->dropBase) {
             $dbManager = $this->getDbManager();
@@ -80,6 +75,32 @@ protected function setupDatabaseData() : void
         putenv('MYSQL_PASSWORD=');
     }
 
+    /**
+     * Битриксовые штучки-дрючки с буфером.
+     *
+     * @return void
+     */
+    private function clearBitrixBuffer() : void
+    {
+        while (ob_get_level()) {
+            ob_end_clean();
+        }
+
+        if ($GLOBALS['APPLICATION']) {
+            $GLOBALS['APPLICATION']->RestartBuffer();
+        }
+    }
+
+    /**
+     * Загрузить дамп - кастомный или нативный.
+     *
+     * @return void
+     */
+    private function migrateDatabase() : void
+    {
+        $this->useCustomDump() ? Migrator::migrate($this->getDumpPath()) : Bootstrap::migrate();
+    }
+
     /**
      * Экземпляр менеджера БД.
      *
@@ -106,4 +127,16 @@ private function needDropBase() : bool
 
         return in_array(ResetDatabaseTrait::class, $traits, true);
     }
+
+    /**
+     * Использовать ли кастомный дамп. Признак - трэйт CustomDumpTrait.
+     *
+     * @return boolean
+     */
+    private function useCustomDump() : bool
+    {
+        $traits = ClassUtils::class_uses_recursive($this);
+
+        return in_array(CustomDumpTrait::class, $traits, true);
+    }
 }

src/Migrator.php

@@ -0,0 +1,42 @@
+<?php
+
+namespace Prokl\BitrixTestingTools;
+
+use InvalidArgumentException;
+use Sheerockoff\BitrixCi\SqlDump;
+
+/**
+ * Class Migrator
+ * Декоратор к мигратору дампов (позволяет загружать свои).
+ * @package Prokl\BitrixTestingTools
+ *
+ * @since 24.04.2021
+ */
+class Migrator
+{
+    /**
+     * @param string $pathDump Путь к дампу базы.
+     *
+     * @return void
+     */
+    public static function migrate(string $pathDump) : void
+    {
+        $db = mysqli_connect(
+            getenv('MYSQL_HOST', true) ?: getenv('MYSQL_HOST'),
+            getenv('MYSQL_USER', true) ?: getenv('MYSQL_USER'),
+            getenv('MYSQL_PASSWORD', true) ?: getenv('MYSQL_PASSWORD'),
+            getenv('MYSQL_DATABASE', true) ?: getenv('MYSQL_DATABASE')
+        );
+
+        if (!$db) {
+            throw new InvalidArgumentException('Mysql connection error.');
+        }
+
+        $sqlDump = new SqlDump($pathDump);
+        foreach ($sqlDump->parse() as $query) {
+            mysqli_query($db, $query);
+        }
+
+        mysqli_close($db);
+    }
+}

src/Traits/CustomDumpTrait.php

@@ -0,0 +1,22 @@
+<?php
+
+namespace Prokl\BitrixTestingTools\Traits;
+
+/**
+ * Trait CustomDumpTrait
+ * Использовать свой дамп базы для тестов. Применяется в сочетании с трэйтом ResetDatabaseTrait.
+ * @package Prokl\BitrixTestingTools\Traits
+ *
+ * @since 24.04.2021
+ */
+trait CustomDumpTrait
+{
+    /**
+     * Путь к кастомному дампу БД.
+     * @return string
+     */
+    protected function getDumpPath() : string
+    {
+        return '';
+    }
+}