test: update unique constraint error expectations

Signed-off-by: Alexey Orlenko's AI Agent <robot@aqrln.net>

Author: tensordreams

query-engine/connector-test-kit-rs/query-engine-tests/tests/writes/ids/byoid.rs

@@ -44,6 +44,20 @@ mod byoid {
         schema.to_owned()
     }
 
+    fn id_unique_constraint_target(runner: &Runner, model: &str) -> String {
+        match runner.connector_version() {
+            ConnectorVersion::MySql(_)
+            | ConnectorVersion::Vitess(Some(query_tests_setup::VitessVersion::PlanetscaleJsWasm)) => {
+                "constraint: `PRIMARY`".to_owned()
+            }
+            ConnectorVersion::Postgres(_) | ConnectorVersion::CockroachDb(_) => {
+                std::format!("constraint: `{model}_pkey`")
+            }
+            ConnectorVersion::Vitess(_) => "(not available)".to_owned(),
+            _ => "fields: (`id`)".to_owned(),
+        }
+    }
+
     // "A Create Mutation" should "create and return item with own Id"
     #[connector_test(schema(schema_1), only(MySql, Postgres, Sqlite, Vitess))]
     async fn create_and_return_item_woi_1(runner: Runner) -> TestResult<()> {
@@ -54,14 +68,7 @@ mod byoid {
           @r###"{"data":{"createOneParent":{"p":"Parent","id":"Own Id"}}}"###
         );
 
-        let error_target = match runner.connector_version() {
-            query_engine_tests::ConnectorVersion::MySql(_)
-            | query_engine_tests::ConnectorVersion::Vitess(Some(query_tests_setup::VitessVersion::PlanetscaleJsWasm)) => {
-                "constraint: `PRIMARY`"
-            }
-            query_engine_tests::ConnectorVersion::Vitess(_) => "(not available)",
-            _ => "fields: (`id`)",
-        };
+        let error_target = id_unique_constraint_target(&runner, "Parent");
 
         assert_error!(
             &runner,
@@ -85,14 +92,7 @@ mod byoid {
           @r###"{"data":{"createOneParent":{"p":"Parent","id":"Own Id"}}}"###
         );
 
-        let error_target = match runner.connector_version() {
-            query_engine_tests::ConnectorVersion::MySql(_)
-            | query_engine_tests::ConnectorVersion::Vitess(Some(query_tests_setup::VitessVersion::PlanetscaleJsWasm)) => {
-                "constraint: `PRIMARY`"
-            }
-            ConnectorVersion::Vitess(_) => "(not available)",
-            _ => "fields: (`id`)",
-        };
+        let error_target = id_unique_constraint_target(&runner, "Parent");
 
         assert_error!(
             &runner,
@@ -146,14 +146,7 @@ mod byoid {
           @r###"{"data":{"createOneParent":{"p":"Parent","id":"Own Id","childOpt":{"c":"Child","id":"Own Child Id"}}}}"###
         );
 
-        let error_target = match runner.connector_version() {
-            query_engine_tests::ConnectorVersion::MySql(_)
-            | query_engine_tests::ConnectorVersion::Vitess(Some(query_tests_setup::VitessVersion::PlanetscaleJsWasm)) => {
-                "constraint: `PRIMARY`"
-            }
-            ConnectorVersion::Vitess(_) => "(not available)",
-            _ => "fields: (`id`)",
-        };
+        let error_target = id_unique_constraint_target(&runner, "Child");
 
         assert_error!(
             &runner,
@@ -177,14 +170,7 @@ mod byoid {
           @r###"{"data":{"createOneParent":{"p":"Parent","id":"Own Id","childOpt":{"c":"Child","id":"Own Child Id"}}}}"###
         );
 
-        let error_target = match runner.connector_version() {
-            query_engine_tests::ConnectorVersion::MySql(_)
-            | query_engine_tests::ConnectorVersion::Vitess(Some(query_tests_setup::VitessVersion::PlanetscaleJsWasm)) => {
-                "constraint: `PRIMARY`"
-            }
-            ConnectorVersion::Vitess(_) => "(not available)",
-            _ => "fields: (`id`)",
-        };
+        let error_target = id_unique_constraint_target(&runner, "Child");
 
         assert_error!(
             &runner,