fix: updating retry strategy to call _retry_on_aborted_exception for batch api calls

aakashanandg · aakashanandg · commit 8f6be72e8377 · 2025-01-07T00:04:57.000+05:30
diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py
@@ -1931,12 +1931,13 @@ def test_context_mgr_w_commit_stats_error(self):
             return_commit_stats=True,
             request_options=RequestOptions(),
         )
-        api.commit.assert_called_once_with(
+        self.assertEqual(api.commit.call_count, 2)
+        api.commit.assert_any_call(
             request=request,
             metadata=[
                 ("google-cloud-resource-prefix", database.name),
                 ("x-goog-spanner-route-to-leader", "true"),
-            ],
+            ]
         )
 
         database.logger.info.assert_not_called()
