From ad8d83355d030e7d3af1b717bb84b58a719f96ce Mon Sep 17 00:00:00 2001
From: Alexander Timmermann <alexander.timmermann@holi.social>
Date: Tue, 25 Mar 2025 15:32:16 +0100
Subject: [PATCH] fix: add new consent property to unit tests

---
 openbook_common/tests/test_tracking.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/openbook_common/tests/test_tracking.py b/openbook_common/tests/test_tracking.py
index 5e8fe285f..f75925dd1 100644
--- a/openbook_common/tests/test_tracking.py
+++ b/openbook_common/tests/test_tracking.py
@@ -14,6 +14,7 @@ class UserForTest(TrackingUser):
     username: str
     tracking_consent_analytics: bool
     tracking_consent_personalization: bool
+    tracking_consent_ad_partners: bool
 
 
 class TestTracking(unittest.TestCase):
@@ -22,7 +23,7 @@ class TestTracking(unittest.TestCase):
 
     @mock.patch("posthog.Posthog.capture")
     def test_no_tracking_without_analytics_consent(self, mock_posthog_capture):
-        user = UserForTest("donald_trump", False, faker.boolean())
+        user = UserForTest("donald_trump", False, faker.boolean(), faker.boolean())
         track(user, TrackingEvent("testEvent"))
         mock_posthog_capture.assert_not_called()
 
@@ -31,7 +32,8 @@ class TestTracking(unittest.TestCase):
         username = "donald_trump"
         consent_analytics = True
         consent_personalization = faker.boolean()
-        user = UserForTest(username, consent_analytics, consent_personalization)
+        consent_ad_partners = faker.boolean()
+        user = UserForTest(username, consent_analytics, consent_personalization, consent_ad_partners)
         track(user, TrackingEvent("testEvent", {}, 1, 2, 3))
         mock_posthog_capture.assert_called_once_with(
             distinct_id=username,
@@ -42,6 +44,7 @@ class TestTracking(unittest.TestCase):
                 "event_version__patch": 3,
                 "trackingConsentAnalytics": consent_analytics,
                 "trackingConsentPersonalization": consent_personalization,
+                "trackingConsentAdPartners": consent_ad_partners,
             },
         )
 
@@ -50,7 +53,8 @@ class TestTracking(unittest.TestCase):
         username = "donald_trump"
         consent_analytics = True
         consent_personalization = faker.boolean()
-        user = UserForTest(username, consent_analytics, consent_personalization)
+        consent_ad_partners = faker.boolean()
+        user = UserForTest(username, consent_analytics, consent_personalization, consent_ad_partners)
         track(user, TrackingEvent("testEvent", {"random_name": "random_value"}, 1, 2, 3))
         mock_posthog_capture.assert_called_once_with(
             distinct_id=username,
@@ -61,13 +65,14 @@ class TestTracking(unittest.TestCase):
                 "event_version__patch": 3,
                 "trackingConsentAnalytics": consent_analytics,
                 "trackingConsentPersonalization": consent_personalization,
+                "trackingConsentAdPartners": consent_ad_partners,
                 "random_name": "random_value",
             },
         )
 
     @mock.patch("posthog.Posthog.capture")
     def test_adds_default_properties_and_version(self, mock_posthog_capture):
-        user = UserForTest("donald_trump", True, True)
+        user = UserForTest("donald_trump", True, True, True)
         track(user, TrackingEvent("testEvent"))
         mock_posthog_capture.assert_called_once_with(
             distinct_id=user.username,
@@ -78,13 +83,14 @@ class TestTracking(unittest.TestCase):
                 "event_version__patch": 0,
                 "trackingConsentAnalytics": True,
                 "trackingConsentPersonalization": True,
+                "trackingConsentAdPartners": True,
             },
         )
 
     @mock.patch("posthog.Posthog.capture")
     def test_does_not_raise(self, mock_posthog_capture):
         mock_posthog_capture.side_effect = Exception("Boom!")
-        user = UserForTest("donald_trump", True, True)
+        user = UserForTest("donald_trump", True, True, True)
         # noinspection PyBroadException
         try:
             track(user, TrackingEvent("testEvent"))
-- 
GitLab