Skip to content

Commit 7189a01

Browse files
nikitagrover19Nikita Grover
authored andcommitted
Update pubsub_integration_test.py
1 parent 8111140 commit 7189a01

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

sdks/python/apache_beam/io/gcp/pubsub_integration_test.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -326,15 +326,17 @@ def test_batch_write_with_ordering_key(self):
326326
from apache_beam.options.pipeline_options import PipelineOptions
327327
from apache_beam.options.pipeline_options import StandardOptions
328328
from apache_beam.transforms import Create
329+
from google.pubsub_v1.types import Subscription
329330

330331
ordering_topic = self.pub_client.create_topic(
331332
name=self.pub_client.topic_path(
332333
self.project, 'psit_topic_ordering' + self.uuid))
333334
ordering_sub = self.sub_client.create_subscription(
334-
name=self.sub_client.subscription_path(
335-
self.project, 'psit_sub_ordering' + self.uuid),
336-
topic=ordering_topic.name,
337-
enable_message_ordering=True)
335+
request=Subscription(
336+
name=self.sub_client.subscription_path(
337+
self.project, 'psit_sub_ordering' + self.uuid),
338+
topic=ordering_topic.name,
339+
enable_message_ordering=True))
338340
time.sleep(10)
339341

340342
try:

0 commit comments

Comments
 (0)