From 0332190e778eec6548c66ff6a6bcee23f644af55 Mon Sep 17 00:00:00 2001
From: Robbie <robbie.coomber@gmail.com>
Date: Wed, 24 Apr 2024 13:48:31 +0100
Subject: [PATCH] Add tests for session properties and session property values
 api

---
 posthog/api/test/test_session.py | 137 +++++++++++++++++++++++++++++++
 1 file changed, 137 insertions(+)
 create mode 100644 posthog/api/test/test_session.py

diff --git a/posthog/api/test/test_session.py b/posthog/api/test/test_session.py
new file mode 100644
index 0000000000000..46fcafabd7c13
--- /dev/null
+++ b/posthog/api/test/test_session.py
@@ -0,0 +1,137 @@
+import uuid
+
+from rest_framework import status
+
+from posthog.models.event.util import create_event
+from posthog.test.base import APIBaseTest
+
+
+class TestSessionsAPI(APIBaseTest):
+    def setUp(self) -> None:
+        super().setUp()
+
+        create_event(
+            team=self.team,
+            event="$pageview",
+            distinct_id="d1",
+            properties={"$session_id": "s1", "utm_source": "google"},
+            event_uuid=(uuid.uuid4()),
+        )
+        create_event(
+            team=self.team,
+            event="$pageview",
+            distinct_id="d1",
+            properties={"$session_id": "s1", "utm_source": "youtube"},
+            event_uuid=(uuid.uuid4()),
+        )
+
+    def test_expected_session_properties(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/property_definitions/")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_properties = {entry["name"] for entry in response.json()["results"]}
+        expected_properties = {
+            "$autocapture_count",
+            "$channel_type",
+            "$end_timestamp",
+            "$entry_url",
+            "$exit_url",
+            "$initial_gad_source",
+            "$initial_gclid",
+            "$initial_referring_domain",
+            "$initial_utm_campaign",
+            "$initial_utm_content",
+            "$initial_utm_medium",
+            "$initial_utm_source",
+            "$initial_utm_term",
+            "$pageview_count",
+            "$session_duration",
+            "$start_timestamp",
+        }
+        assert actual_properties == expected_properties
+
+    def test_search_session_properties(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/property_definitions/?search=utm")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_properties = {entry["name"] for entry in response.json()["results"]}
+        expected_properties = {
+            "$initial_utm_campaign",
+            "$initial_utm_content",
+            "$initial_utm_medium",
+            "$initial_utm_source",
+            "$initial_utm_term",
+        }
+        assert actual_properties == expected_properties
+
+    def test_empty_search_session_properties(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/property_definitions/?search=doesnotexist")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        assert len(response.json()["results"]) == 0
+
+    def test_list_channel_type_values(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/values/?key=$channel_type")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_values = {entry["name"] for entry in response.json()}
+        expected_values = {
+            "Affiliate",
+            "Audio",
+            "Cross Network",
+            "Direct",
+            "Email",
+            "Organic Search",
+            "Organic Shopping",
+            "Organic Video",
+            "Other",
+            "Paid Other",
+            "Paid Search",
+            "Paid Shopping",
+            "Paid Video",
+            "Push",
+            "Referral",
+            "SMS",
+        }
+        assert actual_values == expected_values
+
+    def test_search_channel_type_values(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/values/?key=$channel_type&value=paid")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_values = {entry["name"] for entry in response.json()}
+        expected_values = {
+            "Paid Other",
+            "Paid Search",
+            "Paid Shopping",
+            "Paid Video",
+        }
+        assert actual_values == expected_values
+
+    def test_list_session_property_values(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/values/?key=$initial_utm_source")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_values = {entry["name"] for entry in response.json()}
+        expected_values = {
+            "google",
+            "youtube",
+        }
+        assert actual_values == expected_values
+
+    def test_search_session_property_values(self):
+        response = self.client.get(f"/api/projects/{self.team.pk}/sessions/values/?key=$initial_utm_source&value=tub")
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        actual_values = {entry["name"] for entry in response.json()}
+        expected_values = {
+            "youtube",
+        }
+        assert actual_values == expected_values
+
+    def test_search_session_property_no_matching_values(self):
+        response = self.client.get(
+            f"/api/projects/{self.team.pk}/sessions/values/?key=$initial_utm_source&value=doesnotexist"
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        assert len(response.json()) == 0
+
+    def test_search_missing_session_property_values(self):
+        response = self.client.get(
+            f"/api/projects/{self.team.pk}/sessions/values/?key=$initial_utm_source&value=doesnotexist"
+        )
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        assert len(response.json()) == 0