使用PostgreSQL时Django测试用例无法读取已有库记录的替代方案

问题根源 #

Django测试框架默认会创建test_前缀的独立数据库，PostgreSQL的测试隔离机制会完全隔离测试库与主库，因此测试用例无法读取主库mydb的数据。而MySQL可能因存储引擎（如MyISAM）或测试事务配置的差异，意外允许测试读取主库数据。

可行方案 #

1. 测试时直接使用主库（应急方案） #

修改Django配置，让测试环境直接连接主库，并开启只读模式避免数据污染：

# settings.py
import sys
if 'test' in sys.argv:
    DATABASES['default'].update({
        'NAME': 'mydb',
        'OPTIONS': {
            'options': '-c default_transaction_read_only=on'
        }
    })

测试类继承TransactionTestCase，确保每个测试方法执行后回滚事务（即使只读，也能避免意外修改）。注意：此方案存在风险，若测试代码有写操作会直接失败，仅适合纯只读测试场景。

2. 利用PostgreSQL模板库快速克隆数据 #

这是最适合大型数据集的方案，通过将主库设为模板，测试时直接克隆模板库创建测试库，无需重新生成数据：

步骤1：将主库设置为模板库

-- 断开所有主库连接
SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = 'mydb';
-- 标记主库为模板库
ALTER DATABASE mydb IS_TEMPLATE = true;

步骤2：修改Django测试配置

# settings.py
import sys
if 'test' in sys.argv:
    DATABASES['default']['TEST'] = {
        'NAME': 'test_mydb',
        'TEMPLATE': 'mydb',  # 指定模板库
    }

后续每次运行测试，Django会基于模板库快速创建测试库，速度远快于重新导入数据。若主库数据更新，需重复步骤1更新模板库。

3. 备份恢复脚本自动化 #

编写shell脚本，在测试前将主库备份恢复到测试库：

#!/bin/bash
# 可选：备份主库（仅当数据更新时执行）
# pg_dump -U your_db_user mydb > mydb_backup.sql

# 创建测试库（若不存在）
createdb -U your_db_user test_mydb
# 恢复备份到测试库
psql -U your_db_user test_mydb < mydb_backup.sql

# 运行Django测试
python manage.py test

将脚本设为可执行，每次运行测试时执行该脚本即可。适合需要定期更新测试数据但不想修改Django配置的场景。

4. 优化Fixture导入（适合需定制测试数据场景） #

若需在主库数据基础上修改部分测试数据，可导出主库数据为CSV，利用PostgreSQL的COPY命令快速导入：

1. 导出主库数据为CSV：

COPY (SELECT * FROM your_table) TO '/path/to/your_table.csv' WITH CSV HEADER;

2. 在测试用例中使用原生COPY命令或第三方工具导入数据，比Django默认的fixture导入快数倍。

内容的提问来源于stack exchange，提问作者Tuhin