test that the new sql_down correctly drops everything

devsoc-unsw · May 30, 2024 · 50d963a · 50d963a
1 parent 041e9d0
commit 50d963a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/app/main.py b/app/main.py
@@ -82,6 +82,10 @@ def create_table(metadata: Metadata) -> CreateTableResult:
         # Execute create table
         cur.execute(metadata.sql_up)
 
+        # Test new sql_down
+        cur.execute(metadata.sql_down)
+        cur.execute(metadata.up)
+
         # Store metadata
         cmd = r"INSERT INTO Tables(table_name, up, down) VALUES (%s, %s, %s)"
         cur.execute(cmd, (metadata.table_name, metadata.sql_up, metadata.sql_down))
@@ -94,6 +98,10 @@ def create_table(metadata: Metadata) -> CreateTableResult:
         cur.execute(table_sql[1])  # old sql_down
         cur.execute(metadata.sql_up)
 
+        # Test new sql_down
+        cur.execute(metadata.sql_down)
+        cur.execute(metadata.up)
+
         # Store new metadata
         cmd = r"UPDATE Tables SET up = %s, down = %s WHERE table_name = %s"
         cur.execute(cmd, (metadata.sql_up, metadata.sql_down, metadata.table_name))