@@ -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
14691479def test__complete ():
0 commit comments