laravel phpunit 使用 RefreshDatabase 多数据库方案

可以尝试用下面的 trait 替换原本的 DatabaseTransactions.

不过需要改造成执行多个数据库连接。

<?php

namespace Tests\Traits;

use Illuminate\Contracts\Console\Kernel;
use Illuminate\Foundation\Testing\RefreshDatabaseState;

trait CleanDatabase
{
    /**
     * Define hooks to migrate the database before and after each test.
     *
     * @return void
     */
    public function cleanDatabase()
    {
        if (RefreshDatabaseState::$migrated) {
            return;
        }

        if ($this->shouldSkip()) {
            return;
        }

        $this->truncateAllTable();

        $this->artisan('migrate');

        if ($this->shouldSeed()) {
            $this->artisan('db:seed');
        }

        $this->app[Kernel::class]->setArtisan(null);

        RefreshDatabaseState::$migrated = true;
    }

    protected function truncateAllTable()
    {
        /**@var \Illuminate\Database\DatabaseManager $database*/
        $database = $this->app->make('db');
        $connections = $database->getConnections();

        /**@var \Brokenice\LaravelMysqlPartition\MysqlConnection $connection*/
        foreach ($connections as $connection) {
            $tableNames = $connection->getDoctrineSchemaManager()->listTableNames();
            $tableNames = $this->excludeTable($tableNames, ['migrations']);
            $connection->select('set foreign_key_checks=0');

            foreach ($tableNames as $tableName) {
                $connection->select("TRUNCATE TABLE {$tableName};");
            }

            $connection->select('set foreign_key_checks=1');
        }
        return $database;
    }

    protected function excludeTable(array $tables, array $excludeTables = [])
    {
        $tables = array_filter($tables, function ($value, $key) use ($excludeTables) {
            if (in_array($value, $excludeTables)) {
                return false;
            }
            return true;
        }, ARRAY_FILTER_USE_BOTH );

        return $tables;
    }

    /**
     * Determine if the seed task should be run when refreshing the database.
     *
     * @return bool
     */
    protected function shouldSeed()
    {
        return property_exists($this, 'seed') ? $this->seed : false;
    }

    /**
     * 是否跳过清理数据，编写测试用例时可以用于加快速度
     * @return bool
     */
    protected function shouldSkip()
    {
        return property_exists($this, 'skipClean') ? $this->skipClean : false;
    }
}

非常感谢大家的回答 我发现了一种解决方案

//添加下面这个配置即可
//db_user  db_report 是你需要回滚的数据库链接
 private $connectionsToTransact = ['db_user','db_report'];