45 lines
1.6 KiB
Python
45 lines
1.6 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, "_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,
|
||
|
|
)
|
||
|
|
|
||
|
|
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()
|