scripts-setup-test-db-template.sh
2.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/env bash
# scripts/setup-test-db.sh — 数据库重置脚本：drop + create 空库。
# schema apply 由 Flyway 在 Spring Boot 启动时自动处理（见 docs/04 技术栈 + sql/migrations/V*.sql）。
# seed 数据由测试框架负责（Spring @Sql / Flyway R__seed.sql / data.sql）。
#
# 使用场景：
# - scripts/test.sh 开头：清空库，让 Spring 启动时 Flyway 从 V1 开始重放所有 migration
# - scripts/test.sh 结尾：清空库，避免测试遗留污染下次运行
# - 手动调试时：reset 到零状态
#
# 防护：本脚本只允许在本地 host + 测试库名上执行；非预期目标会被拒绝，
# 避免 .env.local 误指向 staging/prod 时触发不可逆 DROP。
set -euo pipefail
ENV_FILE="$(dirname "$0")/../.env.local"
[ -f "$ENV_FILE" ] || { echo "[setup-test-db] ⚠️ .env.local 不存在（$ENV_FILE）" >&2; exit 1; }
# 用 set -a 加载，让 KEY=VALUE 导出为环境变量；密码中含特殊字符时 .env.local 请用单引号包裹
set -a; . "$ENV_FILE"; set +a
# 防护 1：只允许本地 host（localhost / 127.0.0.1 / ::1）
case "${DB_HOST:-}" in
localhost|127.0.0.1|::1) ;;
*)
echo "[setup-test-db] ⚠️ 拒绝在非本地 host (${DB_HOST}) 上执行 DROP DATABASE" >&2
echo " 如确需对远程 DB 重置（少见，常属误配），请显式声明：TEST_DB_ALLOW_REMOTE=1 $0" >&2
[ "${TEST_DB_ALLOW_REMOTE:-0}" = "1" ] || exit 1
;;
esac
# 防护 2：schema 名需像测试/开发库（含 test / _dev / _local），否则要求显式确认
case "${DB_SCHEMA:-}" in
*test*|*_dev|*_local|*_ci)
;;
*)
echo "[setup-test-db] ⚠️ schema '${DB_SCHEMA}' 不像测试库（期望命名含 test / _dev / _local / _ci）" >&2
echo " 如确为期望行为，请显式声明：TEST_DB_ALLOW_PROD_NAME=1 $0" >&2
[ "${TEST_DB_ALLOW_PROD_NAME:-0}" = "1" ] || exit 1
;;
esac
# 防护 3：显式 banner，让人看见自己在 drop 什么
echo "[setup-test-db] 即将 DROP + CREATE \`${DB_SCHEMA}\` on ${DB_HOST}:${DB_PORT}"
MYSQL_CMD="mysql -h${DB_HOST} -P${DB_PORT} -u${DB_USER} -p${DB_PASSWORD}"
$MYSQL_CMD -e "DROP DATABASE IF EXISTS \`${DB_SCHEMA}\`; CREATE DATABASE \`${DB_SCHEMA}\` CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci;"
echo "[setup-test-db] done — schema will be applied by Flyway when Spring Boot starts"