More test plumbing

googleapis · Dec 25, 2024 · 70eeaef · 70eeaef
1 parent 3e47760
commit 70eeaef
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tests/unit/test_batch.py b/tests/unit/test_batch.py
@@ -259,6 +259,10 @@ def test_commit_ok(self):
             [
                 ("google-cloud-resource-prefix", database.name),
                 ("x-goog-spanner-route-to-leader", "true"),
+                (
+                    "x-goog-spanner-request-id",
+                    f"1.{REQ_RAND_PROCESS_ID}.{database._nth_client_id}.1.1.1",
+                ),
             ],
         )
         self.assertEqual(request_options, RequestOptions())
@@ -577,6 +581,10 @@ def _test_batch_write_with_request_options(
             [
                 ("google-cloud-resource-prefix", database.name),
                 ("x-goog-spanner-route-to-leader", "true"),
+                (
+                    "x-goog-spanner-request-id",
+                    f"1.{REQ_RAND_PROCESS_ID}.{database._nth_client_id}.1.1.1",
+                ),
             ],
         )
         if request_options is None:

diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py
@@ -1204,7 +1204,7 @@ def _execute_partitioned_dml_helper(
                 ("x-goog-spanner-route-to-leader", "true"),
                 (
                     "x-goog-spanner-request-id",
-                    f"1.{REQ_RAND_PROCESS_ID}.{_Client.NTH_CLIENT.value}.1.1.1",
+                    f"1.{REQ_RAND_PROCESS_ID}.{database._nth_client_id}.{database._channel_id}.1.1",
                 ),
             ],
         )
@@ -3217,6 +3217,7 @@ def __init__(
         self.directed_read_options = directed_read_options
         self._nth_client_id = _Client.NTH_CLIENT.increment()
         self._nth_request = AtomicCounter()
+        self.credentials = {}
 
     @property
     def _next_nth_request(self):