Skip to content

Commit 8f6be72

Browse files
committed
fix: updating retry strategy to call _retry_on_aborted_exception for batch api calls
1 parent 2893ade commit 8f6be72

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/unit/test_database.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1931,12 +1931,13 @@ def test_context_mgr_w_commit_stats_error(self):
19311931
return_commit_stats=True,
19321932
request_options=RequestOptions(),
19331933
)
1934-
api.commit.assert_called_once_with(
1934+
self.assertEqual(api.commit.call_count, 2)
1935+
api.commit.assert_any_call(
19351936
request=request,
19361937
metadata=[
19371938
("google-cloud-resource-prefix", database.name),
19381939
("x-goog-spanner-route-to-leader", "true"),
1939-
],
1940+
]
19401941
)
19411942

19421943
database.logger.info.assert_not_called()

0 commit comments

Comments
 (0)