src/Biz/UnitTests/DatabaseDataClearer.php

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

<?php

namespace Biz\UnitTests;

class DatabaseDataClearer
{
    public function __construct($db)
    {
        $this->db = $db;
    }

    public function clear()
    {
        $tableNames = $this->db->getSchemaManager()->listTableNames();
        $this->clearWithTablenames($tableNames);
    }

    public function clearQuickly()
    {
        $tableNames = $this->db->getInsertedTables();
        $tableNames = array_unique($tableNames);
        $this->clearWithTablenames($tableNames);
    }

    private function clearWithTablenames(array $tableNames)
    {
        $sql = '';

        foreach ($tableNames as $tableName) {
            if ('migrations' == $tableName) {
                continue;
            }

            $sql .= "TRUNCATE {$tableName};";
        }

        if (!empty($sql)) {
            $this->db->exec($sql);
            $this->db->resetInsertedTables();
        }
    }
}