update test for cockroach

prisma · Nov 28, 2024 · 1081887 · 1081887
1 parent 9c55967
commit 1081887
Showing 1 changed file with 24 additions and 1 deletion.
diff --git a/schema-engine/sql-migration-tests/tests/migrations/diff.rs b/schema-engine/sql-migration-tests/tests/migrations/diff.rs
@@ -191,7 +191,7 @@ fn from_unique_index_to_pk(mut api: TestApi) {
                 ],
             ]
         "#]]
-    } else if api.is_postgres() || api.is_cockroach() {
+    } else if api.is_postgres() && !api.is_cockroach() {
         expect![[r#"
             [
                 [
@@ -214,6 +214,29 @@ fn from_unique_index_to_pk(mut api: TestApi) {
                 ],
             ]
         "#]]
+    } else if api.is_cockroach() {
+        expect![[r#"
+            [
+                [
+                    "-- DropIndex",
+                    "DROP INDEX \"C_secondary_key\";",
+                    "",
+                    "-- AlterTable",
+                    "ALTER TABLE \"A\" DROP COLUMN \"name\";",
+                    "ALTER TABLE \"A\" ADD CONSTRAINT \"A_pkey\" PRIMARY KEY (\"id\");",
+                    "",
+                    "-- DropIndex",
+                    "DROP INDEX \"A_id_key\";",
+                    "",
+                    "-- AlterTable",
+                    "ALTER TABLE \"B\" ADD CONSTRAINT \"B_pkey\" PRIMARY KEY (\"x\", \"y\");",
+                    "",
+                    "-- DropIndex",
+                    "DROP INDEX \"B_x_y_key\";",
+                    "",
+                ],
+            ]
+        "#]]
     } else if api.is_mssql() {
         expect![[r#"
             [