test(query-engine): Fix PlanetScale test, #4485

prisma · Jun 6, 2024 · e7ad242 · e7ad242
1 parent 8a7b4f7
commit e7ad242
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/.../connector-test-kit-rs/query-engine-tests/tests/writes/top_level_mutations/update_many.rs b/.../connector-test-kit-rs/query-engine-tests/tests/writes/top_level_mutations/update_many.rs
@@ -123,7 +123,7 @@ mod update_many {
     }
 
     // "An updateMany mutation" should "correctly apply all number operations for Int"
-    #[connector_test(exclude(Vitess("planetscale.js", "planetscale.js.wasm"), CockroachDb))]
+    #[connector_test(exclude(CockroachDb))]
     async fn apply_number_ops_for_int(runner: Runner) -> TestResult<()> {
         create_row(&runner, r#"{ id: 1, optStr: "str1" }"#).await?;
         create_row(&runner, r#"{ id: 2, optStr: "str2", optInt: 2 }"#).await?;
@@ -240,7 +240,7 @@ mod update_many {
     }
 
     // "An updateMany mutation" should "correctly apply all number operations for Float"
-    #[connector_test(exclude(Vitess("planetscale.js", "planetscale.js.wasm")))]
+    #[connector_test]
     async fn apply_number_ops_for_float(runner: Runner) -> TestResult<()> {
         create_row(&runner, r#"{ id: 1, optStr: "str1" }"#).await?;
         create_row(&runner, r#"{ id: 2, optStr: "str2", optFloat: 2 }"#).await?;
@@ -297,10 +297,10 @@ mod update_many {
         let count = &res["data"]["updateManyTestModel"]["count"];
 
         // MySql does not count incrementing a null so the count is different
-        if !matches!(runner.connector_version(), ConnectorVersion::MySql(_)) {
-            // On PlanetScale, this fails with:
-            //   left: Number(2)
-            //   right: 3
+        if !matches!(
+            runner.connector_version(),
+            ConnectorVersion::MySql(_) | ConnectorVersion::Vitess(_)
+        ) {
             assert_eq!(count, 3);
         }