Skip to content

Commit 0d5500b

Browse files
committed
Fix tests
1 parent 0597b73 commit 0d5500b

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

sdks/python/apache_beam/runners/worker/sdk_worker_test.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,9 @@ def test_inactive_bundle_processor_returns_empty_progress_response(self):
153153

154154
def test_failed_bundle_processor_returns_failed_progress_response(self):
155155
bundle_processor = mock.MagicMock()
156-
bundle_processor_cache = BundleProcessorCache(None, None, None, {})
156+
data_channel_factory = mock.MagicMock()
157+
bundle_processor_cache = BundleProcessorCache(
158+
None, None, data_channel_factory, {})
157159
bundle_processor_cache.activate('instruction_id')
158160
worker = SdkWorker(bundle_processor_cache)
159161

@@ -262,7 +264,9 @@ def test_harness_monitoring_infos_and_metadata(self):
262264

263265
def test_failed_bundle_processor_returns_failed_split_response(self):
264266
bundle_processor = mock.MagicMock()
265-
bundle_processor_cache = BundleProcessorCache(None, None, None, {})
267+
data_channel_factory = mock.MagicMock()
268+
bundle_processor_cache = BundleProcessorCache(
269+
None, None, data_channel_factory, {})
266270
bundle_processor_cache.activate('instruction_id')
267271
worker = SdkWorker(bundle_processor_cache)
268272

0 commit comments

Comments
 (0)