fix: update collection name based on updates for ingestion pipeline!

TogetherCrew · Jun 3, 2024 · 5fda3a6 · 5fda3a6
1 parent e176c67
commit 5fda3a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dags/hivemind_etl_helpers/tests/integration/test_ingestion_pipeline_etl.py b/dags/hivemind_etl_helpers/tests/integration/test_ingestion_pipeline_etl.py
@@ -10,7 +10,7 @@ class TestIngestionPipeline(unittest.TestCase):
     def test_run_pipeline(self):
         ingest_pipeline = Mock(IngestionPipeline)
         community = "1234"
-        collection_name = "gdrive"
+        collection_name = "google"
         ingestion_pipeline = CustomIngestionPipeline(
             community_id=community, collection_name=collection_name, testing=True
         )
@@ -31,7 +31,7 @@ def test_run_pipeline(self):
 
     def test_load_pipeline_run_exception(self):
         ingestion_pipeline = CustomIngestionPipeline(
-            "1234", collection_name="gdrive", testing=True
+            "1234", collection_name="google", testing=True
         )
         ingestion_pipeline.run_pipeline = Mock()
         ingestion_pipeline.run_pipeline.side_effect = Exception("Test Exception")