Merge pull request #176 from TogetherCrew/feat/175-update-google-coll…

…ection-name feat: updated gdrive collection name to google!
TogetherCrew · Jun 3, 2024 · 3def03b · 3def03b
2 parents 38f7b59 + 5fda3a6
commit 3def03b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 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")

diff --git a/dags/hivemind_google_drive_etl.py b/dags/hivemind_google_drive_etl.py
@@ -38,7 +38,7 @@ def process_gdrive_data(
         )
 
         ingest_data = CustomIngestionPipeline(
-            community_id=community_id, collection_name="gdrive"
+            community_id=community_id, collection_name="google"
         )
         ingest_data.run_pipeline(load_file_data)