Re-use spy on queue.close calls

dlt-hub · Feb 28, 2024 · 9352f9d · 9352f9d
1 parent 6431b62
commit 9352f9d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/scraping/test_scraping_source.py b/tests/scraping/test_scraping_source.py
@@ -64,10 +64,9 @@ def test_pipeline_runners_handle_extended_and_simple_use_cases(mocker):
         dataset_name="quotes",
     )
 
-    spy_on_queue_close_err = mocker.spy(sources.scraping.queue.ScrapingQueue, "close")
     with mocker.patch("dlt.Pipeline.run", side_effect=OSError("bla")):
         run_pipeline(err_pipeline, MySpider, dataset_name="quotes")
-        spy_on_queue_close_err.assert_called()
+        spy_on_queue_close.assert_called()
 
 
 @pytest.mark.skip(