PHP 怎么重置数据库测试

wen PHP项目 2026-08-11 151

本文目录导读：

PHP 怎么重置数据库测试

使用迁移和种子数据（推荐）
使用 SQL 脚本重置
使用 PHPUnit 测试框架
使用 ORM 提供的重置功能
使用备份恢复方案
最佳实践建议

在 PHP 中重置数据库测试数据有多种方法,以下是几种常用的方案：

使用迁移和种子数据（推荐）

// 使用 Laravel 的 Artisan 命令
// 刷新迁移并重新填充数据
php artisan migrate:fresh --seed
// 或者只重置特定表
php artisan db:seed --class=YourTableSeeder
// 回滚所有迁移并重新执行
php artisan migrate:refresh --seed

使用 SQL 脚本重置

// 创建重置脚本文件 reset_database.php
<?php
class DatabaseResetter {
    private $pdo;
    public function __construct($host, $dbname, $user, $pass) {
        try {
            $this->pdo = new PDO("mysql:host=$host;dbname=$dbname", $user, $pass);
            $this->pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
        } catch (PDOException $e) {
            die("连接失败: " . $e->getMessage());
        }
    }
    // 方法1: 执行 SQL 文件
    public function resetWithSqlFile($filePath) {
        try {
            // 读取并执行 SQL 文件
            $sql = file_get_contents($filePath);
            $this->pdo->exec($sql);
            echo "数据库重置成功！\n";
        } catch (PDOException $e) {
            echo "重置失败: " . $e->getMessage() . "\n";
        }
    }
    // 方法2: 清空所有表
    public function cleanAllTables() {
        try {
            $this->pdo->exec("SET FOREIGN_KEY_CHECKS = 0;");
            $stmt = $this->pdo->query("SHOW TABLES");
            while ($row = $stmt->fetch(PDO::FETCH_NUM)) {
                $table = $row[0];
                // 清空表数据并重置自增ID
                $this->pdo->exec("TRUNCATE TABLE `$table`");
                echo "已清空表: $table\n";
            }
            $this->pdo->exec("SET FOREIGN_KEY_CHECKS = 1;");
            echo "所有表已清空\n";
        } catch (PDOException $e) {
            echo "清空失败: " . $e->getMessage() . "\n";
        }
    }
    // 方法3: 使用事务回滚测试数据
    public function resetWithTransaction() {
        $this->pdo->beginTransaction();
        try {
            // 在这里执行你的测试操作
            // ... 测试代码
            // 回滚事务，撤销所有更改
            $this->pdo->rollBack();
            echo "测试修改已回滚\n";
        } catch (Exception $e) {
            $this->pdo->rollBack();
            echo "错误: " . $e->getMessage() . "\n";
        }
    }
    // 方法4: 导出并重新导入数据库
    public function restoreFromBackup($backupFile) {
        try {
            // 使用系统命令导入备份
            $command = "mysql -u username -p password database_name < " . $backupFile;
            exec($command, $output, $return_var);
            if ($return_var === 0) {
                echo "数据库从备份恢复成功\n";
            } else {
                echo "数据库恢复失败\n";
            }
        } catch (Exception $e) {
            echo "恢复出错: " . $e->getMessage() . "\n";
        }
    }
}
// 使用示例
$resetter = new DatabaseResetter('localhost', 'test_db', 'root', 'password');
// 方式1: 执行SQL文件
$resetter->resetWithSqlFile('path/to/reset.sql');
// 方式2: 清空所有表
$resetter->cleanAllTables();
// 方式3: 使用事务回滚
$resetter->resetWithTransaction();
// 方式4: 从备份文件恢复
$resetter->restoreFromBackup('backup.sql');
?>

使用 PHPUnit 测试框架

<?php
use PHPUnit\Framework\TestCase;
class DatabaseTest extends TestCase {
    private static $pdo;
    private $transaction;
    protected function setUp(): void {
        // 开始事务
        $this->transaction = self::$pdo->beginTransaction();
    }
    protected function tearDown(): void {
        // 回滚事务，确保测试数据隔离
        if ($this->transaction) {
            self::$pdo->rollBack();
        }
    }
    public function testDatabaseOperations() {
        // 测试代码会自动在安全环境中运行
        self::$pdo->exec("INSERT INTO users (name) VALUES ('测试用户')");
        // 无论测试结果如何，都会回滚
    }
}
?>

使用 ORM 提供的重置功能

Doctrine ORM

<?php
// 使用 Doctrine SchemaTool
$schemaTool = $entityManager->getConnection()->getSchemaManager();
$metadata = $entityManager->getMetadataFactory()->getAllMetadata();
$schemaTool->dropDatabase();
$schemaTool->createDatabase();
$schemaTool->createSchema($metadata);
?>

Laravel

<?php
// 在测试中
use Illuminate\Foundation\Testing\DatabaseMigrations;
use Illuminate\Foundation\Testing\RefreshDatabase;
class ExampleTest extends TestCase {
    use RefreshDatabase; // 每次测试后会重置数据库
    public function test_example() {
        // 测试代码
    }
}
?>

使用备份恢复方案

<?php
function createBackup($db, $host, $user, $pass) {
    // 创建数据库备份
    $command = "mysqldump -h $host -u $user -p$pass $db > backup.sql";
    exec($command);
}
function restoreBackup($db, $host, $user, $pass) {
    // 从备份恢复
    $command = "mysql -h $host -u $user -p$pass $db < backup.sql";
    exec($command);
}
// 使用示例
createBackup('test_db', 'localhost', 'root', 'password');
// 测试后恢复
restoreBackup('test_db', 'localhost', 'root', 'password');
?>