Merge pull request #156 from TogetherCrew/feat/mediawiki-ingest-and-v…

…ectorize fix: mediaWiki platform name!
TogetherCrew · May 22, 2024 · 19f59f0 · 19f59f0
2 parents 13e7c4e + d21a08a
commit 19f59f0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/dags/hivemind_etl_helpers/src/utils/modules/mediawiki.py b/dags/hivemind_etl_helpers/src/utils/modules/mediawiki.py
@@ -3,7 +3,7 @@
 
 class ModulesMediaWiki(ModulesBase):
     def __init__(self) -> None:
-        self.platform_name = "mediawiki"
+        self.platform_name = "mediaWiki"
         super().__init__()
 
     def get_learning_platforms(

diff --git a/dags/hivemind_etl_helpers/tests/integration/test_mediawiki_modules.py b/dags/hivemind_etl_helpers/tests/integration/test_mediawiki_modules.py
@@ -25,7 +25,7 @@ def test_get_single_data(self):
         self.client["Core"]["platforms"].insert_one(
             {
                 "_id": platform_id,
-                "name": "mediawiki",
+                "name": "mediaWiki",
                 "metadata": {
                     "pageIds": [
                         "Main_Page",
@@ -48,7 +48,7 @@ def test_get_single_data(self):
                     "platforms": [
                         {
                             "platform": platform_id,
-                            "name": "mediawiki",
+                            "name": "mediaWiki",
                             "metadata": {
                                 "api_url": "http://example.com/api",
                             },
@@ -89,14 +89,14 @@ def test_get_mediawiki_communities_data_multiple_platforms(self):
                     "platforms": [
                         {
                             "platform": platform_id1,
-                            "name": "mediawiki",
+                            "name": "mediaWiki",
                             "metadata": {
                                 "api_url": "http://example1.com/api",
                             },
                         },
                         {
                             "platform": platform_id2,
-                            "name": "mediawiki",
+                            "name": "mediaWiki",
                             "metadata": {
                                 "api_url": "http://example2.com/api",
                             },
@@ -109,7 +109,7 @@ def test_get_mediawiki_communities_data_multiple_platforms(self):
         self.client["Core"]["platforms"].insert_one(
             {
                 "_id": platform_id1,
-                "name": "mediawiki",
+                "name": "mediaWiki",
                 "metadata": {
                     "pageIds": [
                         "Main_Page",
@@ -127,7 +127,7 @@ def test_get_mediawiki_communities_data_multiple_platforms(self):
         self.client["Core"]["platforms"].insert_one(
             {
                 "_id": platform_id2,
-                "name": "mediawiki",
+                "name": "mediaWiki",
                 "metadata": {
                     "pageIds": [
                         "Sandbox",