Fix spatialite migration tests

prisma · Oct 3, 2024 · 41b829c · 41b829c
1 parent 7c834a7
commit 41b829c
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 11 deletions.
diff --git a/schema-engine/connectors/sql-schema-connector/src/sql_renderer/sqlite_renderer.rs b/schema-engine/connectors/sql-schema-connector/src/sql_renderer/sqlite_renderer.rs
@@ -150,7 +150,7 @@ impl SqlRenderer for SqliteFlavour {
         if create_geometries.is_empty() {
             create_table.to_string()
         } else {
-            create_table.to_string() + "\n;" + &create_geometries
+            create_table.to_string() + ";\n" + &create_geometries
         }
     }
 

diff --git a/schema-engine/sql-migration-tests/tests/migrations/diff.rs b/schema-engine/sql-migration-tests/tests/migrations/diff.rs
@@ -822,11 +822,19 @@ fn from_multi_file_schema_datasource_to_url(mut api: TestApi) {
 
     api.diff(input).unwrap();
 
-    let expected_printed_messages = expect![[r#"
-        [
-            "-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"cows\";\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
-        ]
-    "#]];
+    let expected_printed_messages = if api.tags().contains(test_setup::Tags::Spatialite) {
+        expect![[r#"
+            [
+                "-- DropTable\nPRAGMA foreign_keys=off;\nSELECT DropTable(NULL, 'cows');\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
+            ]
+        "#]]
+    } else {
+        expect![[r#"
+            [
+                "-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"cows\";\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
+            ]
+        "#]]
+    };
     expected_printed_messages.assert_debug_eq(&host.printed_messages.lock().unwrap());
 }
 
@@ -888,11 +896,19 @@ fn from_multi_file_schema_datamodel_to_url(mut api: TestApi) {
 
     api.diff(input).unwrap();
 
-    let expected_printed_messages = expect![[r#"
-        [
-            "-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"cows\";\nPRAGMA foreign_keys=on;\n\n-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"dogs\";\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
-        ]
-    "#]];
+    let expected_printed_messages = if api.tags().contains(test_setup::Tags::Spatialite) {
+        expect![[r#"
+            [
+                "-- DropTable\nPRAGMA foreign_keys=off;\nSELECT DropTable(NULL, 'cows');\nPRAGMA foreign_keys=on;\n\n-- DropTable\nPRAGMA foreign_keys=off;\nSELECT DropTable(NULL, 'dogs');\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
+            ]
+        "#]]
+    } else {
+        expect![[r#"
+            [
+                "-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"cows\";\nPRAGMA foreign_keys=on;\n\n-- DropTable\nPRAGMA foreign_keys=off;\nDROP TABLE \"dogs\";\nPRAGMA foreign_keys=on;\n\n-- CreateTable\nCREATE TABLE \"cats\" (\n    \"id\" INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,\n    \"meows\" BOOLEAN DEFAULT true\n);\n",
+            ]
+        "#]]
+    };
     expected_printed_messages.assert_debug_eq(&host.printed_messages.lock().unwrap());
 }