Kevin Veen-BirkenbachandClaude Opus 5.5
b30344d494
Dumps stream through gzip into <db>.backup.sql.gz; no uncompressed copy touches the disk. Restore reads gzip and plain dumps by their magic bytes, verifies a gzip dump to its end before anything is dropped, and still finds the uncompressed names of older generations. The manifest names the compression. 🎯 Fix: 92 % 💥 Break: 20 % 💸 Cost: – ⏱️ Time: – 🛡️ Security: ~100 % 🏆 Score: 74 🚦 Gate: ⛔ break 20 % > 5 Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
47 lines
1.7 KiB
Python
47 lines
1.7 KiB
Python
import tempfile
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from baudolo.restore.db import mariadb as mariadb_mod
|
|
|
|
|
|
class TestMariadbEmptyDrop(unittest.TestCase):
|
|
def test_drops_run_in_one_session_with_fk_checks_off(self) -> None:
|
|
calls = []
|
|
|
|
def _capture(container, argv, **kwargs):
|
|
calls.append(argv)
|
|
result = MagicMock()
|
|
result.stdout = b"users\nfetches\n"
|
|
return result
|
|
|
|
with tempfile.NamedTemporaryFile(suffix=".sql") as sql:
|
|
sql.write(b"CREATE TABLE t (id int);\n")
|
|
sql.flush()
|
|
with (
|
|
patch.object(mariadb_mod, "docker_exec", side_effect=_capture),
|
|
patch.object(mariadb_mod, "docker_exec_stream"),
|
|
patch.object(mariadb_mod, "_pick_client", return_value="mariadb"),
|
|
):
|
|
mariadb_mod.restore_mariadb_sql(
|
|
container="db",
|
|
db_name="mailu",
|
|
user="mailu",
|
|
password="pw",
|
|
sql_path=sql.name,
|
|
empty=True,
|
|
check_version=False,
|
|
)
|
|
|
|
drop_calls = [argv for argv in calls if any("DROP TABLE" in a for a in argv)]
|
|
self.assertEqual(len(drop_calls), 1, f"expected ONE drop session: {calls}")
|
|
drop_sql = drop_calls[0][-1]
|
|
self.assertTrue(drop_sql.startswith("SET FOREIGN_KEY_CHECKS=0; "))
|
|
self.assertIn("DROP TABLE IF EXISTS `mailu`.`users`;", drop_sql)
|
|
self.assertIn("DROP TABLE IF EXISTS `mailu`.`fetches`;", drop_sql)
|
|
self.assertTrue(drop_sql.endswith("SET FOREIGN_KEY_CHECKS=1;"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|