Skip to content

Commit d076bc6

Browse files
committed
fix broken presubmit
1 parent dbc204b commit d076bc6

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

sdks/python/apache_beam/runners/interactive/utils_test.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -244,9 +244,9 @@ def test_child_module_logger_can_override_logging_level(self, mock_emit):
244244
reason='[interactive] dependency is not installed.')
245245
class ProgressIndicatorTest(unittest.TestCase):
246246
def setUp(self):
247-
self.patcher = patch(
248-
'apache_beam.runners.interactive.cache_manager.CacheManager.cleanup')
249-
self.patcher.start()
247+
self.gcs_patcher = patch(
248+
'apache_beam.io.gcp.gcsfilesystem.GCSFileSystem.delete')
249+
self.gcs_patcher.start()
250250
ie.new_env()
251251

252252
@patch('IPython.get_ipython', new_callable=mock_get_ipython)
@@ -283,7 +283,7 @@ def test_progress_in_HTML_JS_when_in_notebook(
283283
mocked_js.assert_called()
284284

285285
def tearDown(self):
286-
self.patcher.stop()
286+
self.gcs_patcher.stop()
287287

288288

289289
@unittest.skipIf(
@@ -293,9 +293,9 @@ class MessagingUtilTest(unittest.TestCase):
293293
SAMPLE_DATA = {'a': [1, 2, 3], 'b': 4, 'c': '5', 'd': {'e': 'f'}}
294294

295295
def setUp(self):
296-
self.patcher = patch(
297-
'apache_beam.runners.interactive.cache_manager.CacheManager.cleanup')
298-
self.patcher.start()
296+
self.gcs_patcher = patch(
297+
'apache_beam.io.gcp.gcsfilesystem.GCSFileSystem.delete')
298+
self.gcs_patcher.start()
299299
ie.new_env()
300300

301301
def test_as_json_decorator(self):
@@ -308,7 +308,7 @@ def dummy():
308308
self.assertEqual(json.loads(dummy()), MessagingUtilTest.SAMPLE_DATA)
309309

310310
def tearDown(self):
311-
self.patcher.stop()
311+
self.gcs_patcher.stop()
312312

313313

314314
class GeneralUtilTest(unittest.TestCase):

0 commit comments

Comments
 (0)