112 lines
3.6 KiB
Bash
Executable File
112 lines
3.6 KiB
Bash
Executable File
#!/bin/bash
|
|
# Removes E2E test data that leaked into the database.
|
|
# Documents with title LIKE 'E2E%' and persons whose last_name ends with a
|
|
# timestamp suffix (Sender-<ts> / Receiver-<ts> from bilateral fixtures), plus
|
|
# the manually-created E2E Testperson from persons.spec.ts.
|
|
set -euo pipefail
|
|
|
|
CONTAINER="archive-db"
|
|
DB_USER="archive_user"
|
|
DB_NAME="family_archive_db"
|
|
|
|
DRY_RUN=false
|
|
if [[ "${1:-}" == "--dry-run" ]]; then
|
|
DRY_RUN=true
|
|
fi
|
|
|
|
run_sql() {
|
|
docker exec "$CONTAINER" psql -U "$DB_USER" -d "$DB_NAME" -c "$1"
|
|
}
|
|
|
|
# ── Dry-run summary ──────────────────────────────────────────────────────────
|
|
|
|
echo ""
|
|
echo "E2E test data found in the database:"
|
|
echo ""
|
|
|
|
run_sql "
|
|
SELECT count(*) AS e2e_documents
|
|
FROM documents
|
|
WHERE title LIKE 'E2E%';
|
|
"
|
|
|
|
run_sql "
|
|
SELECT count(*) AS e2e_persons
|
|
FROM persons
|
|
WHERE
|
|
-- bilateral fixture: Visual Sender-<ts>, A11y Receiver-<ts>, etc.
|
|
last_name ~ '^(Sender|Receiver)-[0-9]{10,}$'
|
|
-- persons.spec.ts: E2E Testperson
|
|
OR (first_name = 'E2E' AND last_name = 'Testperson');
|
|
"
|
|
|
|
run_sql "
|
|
SELECT count(*) AS e2e_users
|
|
FROM users
|
|
WHERE first_name = 'E2E' AND last_name = 'Testuser';
|
|
"
|
|
|
|
if $DRY_RUN; then
|
|
echo "Dry run — no changes made."
|
|
exit 0
|
|
fi
|
|
|
|
# ── Confirmation ─────────────────────────────────────────────────────────────
|
|
|
|
echo ""
|
|
echo "This will permanently delete all E2E test documents, persons, and users."
|
|
read -rp "Type 'yes' to continue: " CONFIRM
|
|
[ "$CONFIRM" != "yes" ] && echo "Aborted." && exit 0
|
|
|
|
# ── Cleanup ──────────────────────────────────────────────────────────────────
|
|
|
|
echo ""
|
|
echo "Deleting E2E test data..."
|
|
|
|
run_sql "
|
|
-- Notifications have no FK on document_id, must be cleaned manually.
|
|
DELETE FROM notifications
|
|
WHERE document_id IN (SELECT id FROM documents WHERE title LIKE 'E2E%');
|
|
|
|
-- Delete E2E documents. All dependent tables cascade:
|
|
-- document_receivers, document_tags, document_training_labels,
|
|
-- document_versions, document_annotations, document_comments,
|
|
-- comment_mentions, transcription_blocks, transcription_block_versions,
|
|
-- block_mentioned_persons
|
|
DELETE FROM documents WHERE title LIKE 'E2E%';
|
|
|
|
-- Match all test persons:
|
|
-- bilateral fixtures → last_name = 'Sender-<timestamp>' or 'Receiver-<timestamp>'
|
|
-- persons.spec.ts → first_name = 'E2E', last_name = 'Testperson'
|
|
-- Nullify sender FK on any non-E2E documents first (prevents FK violation).
|
|
UPDATE documents
|
|
SET sender_id = NULL
|
|
WHERE sender_id IN (
|
|
SELECT id FROM persons
|
|
WHERE last_name ~ '^(Sender|Receiver)-[0-9]{10,}$'
|
|
OR (first_name = 'E2E' AND last_name = 'Testperson')
|
|
);
|
|
|
|
-- Remove receiver links from non-E2E documents (same edge case).
|
|
DELETE FROM document_receivers
|
|
WHERE person_id IN (
|
|
SELECT id FROM persons
|
|
WHERE last_name ~ '^(Sender|Receiver)-[0-9]{10,}$'
|
|
OR (first_name = 'E2E' AND last_name = 'Testperson')
|
|
);
|
|
|
|
-- Delete test persons. person_name_aliases cascade automatically.
|
|
DELETE FROM persons
|
|
WHERE last_name ~ '^(Sender|Receiver)-[0-9]{10,}$'
|
|
OR (first_name = 'E2E' AND last_name = 'Testperson');
|
|
|
|
-- Delete leaked E2E users (created but not cleaned up in admin.spec.ts).
|
|
DELETE FROM users_groups
|
|
WHERE user_id IN (SELECT id FROM users WHERE first_name = 'E2E' AND last_name = 'Testuser');
|
|
|
|
DELETE FROM users
|
|
WHERE first_name = 'E2E' AND last_name = 'Testuser';
|
|
" --quiet
|
|
|
|
echo "Done."
|