diff --git a/pubsub/tests/unit/pubsub_v1/subscriber/test_streaming_pull_manager.py b/pubsub/tests/unit/pubsub_v1/subscriber/test_streaming_pull_manager.py index 4acbf74ea8c0..924fde56ea70 100644 --- a/pubsub/tests/unit/pubsub_v1/subscriber/test_streaming_pull_manager.py +++ b/pubsub/tests/unit/pubsub_v1/subscriber/test_streaming_pull_manager.py @@ -272,10 +272,10 @@ def test_heartbeat_inactive(): @mock.patch( - 'google.cloud.pubsub_v1.subscriber._protocol.bidi.ResumableBidiRpc', + 'google.api_core.bidi.ResumableBidiRpc', autospec=True) @mock.patch( - 'google.cloud.pubsub_v1.subscriber._protocol.bidi.BackgroundConsumer', + 'google.api_core.bidi.BackgroundConsumer', autospec=True) @mock.patch( 'google.cloud.pubsub_v1.subscriber._protocol.leaser.Leaser',