Merge pull request #1 from dgorelik/dgorelik-patch-1

Use kwargs in test_subscriber_client
dgorelik · Jun 18, 2019 · 26de36d · 26de36d
2 parents a0492a1 + c44f61e
commit 26de36d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/pubsub/tests/unit/pubsub_v1/subscriber/test_subscriber_client.py b/pubsub/tests/unit/pubsub_v1/subscriber/test_subscriber_client.py
@@ -73,7 +73,9 @@ def test_subscribe(manager_open):
 
     assert future._manager._subscription == "sub_name_a"
     manager_open.assert_called_once_with(
-        mock.ANY, mock.sentinel.callback, future.set_exception
+        mock.ANY,
+        callback=mock.sentinel.callback,
+        on_callback_error=future.set_exception,
     )
 
 
@@ -100,5 +102,7 @@ def test_subscribe_options(manager_open):
     assert future._manager.flow_control == flow_control
     assert future._manager._scheduler == scheduler
     manager_open.assert_called_once_with(
-        mock.ANY, mock.sentinel.callback, future.set_exception
+        mock.ANY,
+        callback=mock.sentinel.callback,
+        on_callback_error=future.set_exception,
     )