@@ -1253,7 +1253,9 @@ def test_wo_transaction(stub, datastore_pb2):
12531253 )
12541254
12551255 request = datastore_pb2 .CommitRequest .return_value
1256- assert api .commit .future .called_once_with (request )
1256+ api .commit .future .assert_called_once_with (
1257+ request , metadata = mock .ANY , timeout = mock .ANY
1258+ )
12571259
12581260 @staticmethod
12591261 @pytest .mark .usefixtures ("in_context" )
@@ -1276,7 +1278,9 @@ def test_w_transaction(stub, datastore_pb2):
12761278 )
12771279
12781280 request = datastore_pb2 .CommitRequest .return_value
1279- assert api .commit .future .called_once_with (request )
1281+ api .commit .future .assert_called_once_with (
1282+ request , metadata = mock .ANY , timeout = mock .ANY
1283+ )
12801284
12811285
12821286@pytest .mark .usefixtures ("in_context" )
@@ -1365,7 +1369,9 @@ def test__datastore_allocate_ids(stub, datastore_pb2):
13651369 )
13661370
13671371 request = datastore_pb2 .AllocateIdsRequest .return_value
1368- assert api .allocate_ids .future .called_once_with (request )
1372+ api .allocate_ids .future .assert_called_once_with (
1373+ request , metadata = mock .ANY , timeout = mock .ANY
1374+ )
13691375
13701376
13711377@pytest .mark .usefixtures ("in_context" )
@@ -1407,7 +1413,9 @@ def test_read_only(stub, datastore_pb2):
14071413 )
14081414
14091415 request = datastore_pb2 .BeginTransactionRequest .return_value
1410- assert api .begin_transaction .future .called_once_with (request )
1416+ api .begin_transaction .future .assert_called_once_with (
1417+ request , metadata = mock .ANY , timeout = mock .ANY
1418+ )
14111419
14121420 @staticmethod
14131421 @pytest .mark .usefixtures ("in_context" )
@@ -1432,7 +1440,9 @@ def test_read_write(stub, datastore_pb2):
14321440 )
14331441
14341442 request = datastore_pb2 .BeginTransactionRequest .return_value
1435- assert api .begin_transaction .future .called_once_with (request )
1443+ api .begin_transaction .future .assert_called_once_with (
1444+ request , metadata = mock .ANY , timeout = mock .ANY
1445+ )
14361446
14371447
14381448@pytest .mark .usefixtures ("in_context" )
@@ -1463,7 +1473,9 @@ def test__datastore_rollback(stub, datastore_pb2):
14631473 )
14641474
14651475 request = datastore_pb2 .RollbackRequest .return_value
1466- assert api .rollback .future .called_once_with (request )
1476+ api .rollback .future .assert_called_once_with (
1477+ request , metadata = mock .ANY , timeout = mock .ANY
1478+ )
14671479
14681480
14691481def test__complete ():
0 commit comments