Fix path for patch of 'bidi' elements. (#6243)

Closes #6241.
googleapis · Oct 17, 2018 · f70c00b · f70c00b
1 parent 7527a4e
commit f70c00b
Showing 1 changed file with 2 additions and 2 deletions.
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
@@ -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',