From 850546390b2e33a84a2c5b68af20cb94997b8065 Mon Sep 17 00:00:00 2001
From: Mohammad Amin <dadgaramin96@gmail.com>
Date: Tue, 16 Jul 2024 09:45:57 +0330
Subject: [PATCH] fix: missing values in raw data! more realistic test cases.

---
 tests/integration/test_heatmaps_hourly_mentions.py | 2 ++
 tests/integration/test_heatmaps_reactions.py       | 4 ++++
 tests/integration/test_heatmaps_replier.py         | 1 +
 3 files changed, 7 insertions(+)

diff --git a/tests/integration/test_heatmaps_hourly_mentions.py b/tests/integration/test_heatmaps_hourly_mentions.py
index 4a56143..325cdc8 100644
--- a/tests/integration/test_heatmaps_hourly_mentions.py
+++ b/tests/integration/test_heatmaps_hourly_mentions.py
@@ -60,6 +60,7 @@ def test_mentioned_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     }
                     prepared_rawmemberactivities.append(prepared_rawdata)
@@ -80,6 +81,7 @@ def test_mentioned_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": chId + "AAA",  # could be thr_message
+                            "bot_activity": False,
                         },
                     }
                     prepared_rawmemberactivities.append(rawdata_self_interaction)
diff --git a/tests/integration/test_heatmaps_reactions.py b/tests/integration/test_heatmaps_reactions.py
index 09cd53b..bbc421b 100644
--- a/tests/integration/test_heatmaps_reactions.py
+++ b/tests/integration/test_heatmaps_reactions.py
@@ -65,6 +65,7 @@ def test_reacted_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     },
                     {
@@ -82,6 +83,7 @@ def test_reacted_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     },
                     {
@@ -99,6 +101,7 @@ def test_reacted_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     },
                     {
@@ -116,6 +119,7 @@ def test_reacted_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     },
                 ]
diff --git a/tests/integration/test_heatmaps_replier.py b/tests/integration/test_heatmaps_replier.py
index a716060..cecebd8 100644
--- a/tests/integration/test_heatmaps_replier.py
+++ b/tests/integration/test_heatmaps_replier.py
@@ -78,6 +78,7 @@ def test_reply_messages():
                         "metadata": {
                             "channel_id": chId,
                             "thread_id": None,
+                            "bot_activity": False,
                         },
                     },
                 ]