Skip to content

Commit 653a992

Browse files
committed
feat: update test cases with code updates!
1 parent 33099f7 commit 653a992

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/unit/test_mediawiki_etl.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def test_load_with_dump_deletion(self, mock_ingestion_pipeline_class):
164164

165165
# Verify that methods were called correctly
166166
mock_ingestion_pipeline_class.assert_called_once_with(
167-
self.community_id, collection_name="mediawiki"
167+
self.community_id, collection_name=self.platform_id
168168
)
169169
mock_pipeline.run_pipeline.assert_called_once_with(documents)
170170
self.assertFalse(os.path.exists(etl.dump_dir))
@@ -192,7 +192,7 @@ def test_load_without_dump_deletion(self, mock_ingestion_pipeline_class):
192192

193193
# Verify that methods were called correctly
194194
mock_ingestion_pipeline_class.assert_called_once_with(
195-
self.community_id, collection_name="mediawiki"
195+
self.community_id, collection_name=self.platform_id
196196
)
197197
mock_pipeline.run_pipeline.assert_called_once_with(documents)
198198
self.assertTrue(os.path.exists(etl.dump_dir))

0 commit comments

Comments
 (0)