fix: missng test cases to update!

+ fixed black linter isssues too.
TogetherCrew · Aug 6, 2024 · 7b87f2a · 7b87f2a
1 parent ee9897e
commit 7b87f2a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/dags/analyzer_helper/discourse/transform_raw_data.py b/dags/analyzer_helper/discourse/transform_raw_data.py
@@ -24,9 +24,9 @@ def create_data_entry(
             metadata = {
                 **metadata,  # previous ones
                 "link": (
-                    f"https://{self.forum_endpoint}/t/" +
-                    f"{int(topic_id)}/{int(post_number)}"
-                )
+                    f"https://{self.forum_endpoint}/t/"
+                    + f"{int(topic_id)}/{int(post_number)}"
+                ),
             }
 
         result = {

diff --git a/dags/analyzer_helper/tests/unit/test_discourse_transform_raw_data.py b/dags/analyzer_helper/tests/unit/test_discourse_transform_raw_data.py
@@ -127,6 +127,7 @@ def test_transform_data_with_replied_user(self):
             },
             {
                 "author_id": "4444",
+                "text": "some content #2",
                 "date": datetime.datetime(
                     2023, 9, 11, 21, 41, 43, 553000, tzinfo=datetime.timezone.utc
                 ),
@@ -135,6 +136,7 @@ def test_transform_data_with_replied_user(self):
                     "category_id": None,
                     "topic_id": 6134,
                     "bot_activity": False,
+                    "link": f"https://{self.forum_endpoint}/t/6134/1",
                 },
                 "actions": [],
                 "interactions": [
@@ -207,19 +209,20 @@ def test_transform_data_with_reactions(self):
                         "name": "reaction",
                         "type": "emitter",
                         "users_engaged_id": ["6168"],
-                        "link": f"https://{self.forum_endpoint}/t/6134/1",
                     }
                 ],
                 "metadata": {
                     "category_id": None,
                     "topic_id": 6134,
                     "bot_activity": False,
+                    "link": f"https://{self.forum_endpoint}/t/6134/1",
                 },
                 "source_id": "6261",
             },
             {
                 "actions": [],
                 "author_id": "2",
+                "text": "some content #2",
                 "date": datetime.datetime(
                     2023, 9, 11, 21, 42, 43, 553000, tzinfo=datetime.timezone.utc
                 ),
@@ -234,6 +237,7 @@ def test_transform_data_with_reactions(self):
                     "category_id": None,
                     "topic_id": 6134,
                     "bot_activity": False,
+                    "link": f"https://{self.forum_endpoint}/t/6134/1",
                 },
                 "source_id": "6261",
             },