Skip to content

Commit 3ce7abc

Browse files
committed
feat: Add Python 3.12
1 parent ebf1627 commit 3ce7abc

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

tests/unit/test__datastore_api.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1276,7 +1276,9 @@ def test_w_transaction(stub, datastore_pb2):
12761276
)
12771277

12781278
request = datastore_pb2.CommitRequest.return_value
1279-
assert api.commit.future.assert_called_once_with(request)
1279+
api.commit.future.assert_called_once_with(
1280+
request, metadata=mock.ANY, timeout=mock.ANY
1281+
)
12801282

12811283

12821284
@pytest.mark.usefixtures("in_context")
@@ -1365,7 +1367,9 @@ def test__datastore_allocate_ids(stub, datastore_pb2):
13651367
)
13661368

13671369
request = datastore_pb2.AllocateIdsRequest.return_value
1368-
assert api.allocate_ids.future.assert_called_once_with(request)
1370+
api.allocate_ids.future.assert_called_once_with(
1371+
request, metadata=mock.ANY, timeout=mock.ANY
1372+
)
13691373

13701374

13711375
@pytest.mark.usefixtures("in_context")
@@ -1407,7 +1411,9 @@ def test_read_only(stub, datastore_pb2):
14071411
)
14081412

14091413
request = datastore_pb2.BeginTransactionRequest.return_value
1410-
assert api.begin_transaction.future.assert_called_once_with(request)
1414+
api.begin_transaction.future.assert_called_once_with(
1415+
request, metadata=mock.ANY, timeout=mock.ANY
1416+
)
14111417

14121418
@staticmethod
14131419
@pytest.mark.usefixtures("in_context")
@@ -1432,7 +1438,9 @@ def test_read_write(stub, datastore_pb2):
14321438
)
14331439

14341440
request = datastore_pb2.BeginTransactionRequest.return_value
1435-
assert api.begin_transaction.future.assert_called_once_with(request)
1441+
api.begin_transaction.future.assert_called_once_with(
1442+
request, metadata=mock.ANY, timeout=mock.ANY
1443+
)
14361444

14371445

14381446
@pytest.mark.usefixtures("in_context")
@@ -1463,7 +1471,9 @@ def test__datastore_rollback(stub, datastore_pb2):
14631471
)
14641472

14651473
request = datastore_pb2.RollbackRequest.return_value
1466-
assert api.rollback.future.assert_called_once_with(request)
1474+
api.rollback.future.assert_called_once_with(
1475+
request, metadata=mock.ANY, timeout=mock.ANY
1476+
)
14671477

14681478

14691479
def test__complete():

0 commit comments

Comments
 (0)