feat(dev): Show Celery cohort queries in CH debug modal

ee/clickhouse/models/test/__snapshots__/test_cohort.ambr

@@ -1,7 +1,7 @@
 # serializer version: 1
 # name: TestCohort.test_cohortpeople_basic
   '''
-
+  /* cohort_calculation: */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -37,7 +37,7 @@
 # ---
 # name: TestCohort.test_cohortpeople_with_not_in_cohort_operator
   '''
-
+  /* cohort_calculation: */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -71,7 +71,7 @@
 # ---
 # name: TestCohort.test_cohortpeople_with_not_in_cohort_operator.1
   '''
-
+  /* cohort_calculation: */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -184,7 +184,7 @@
 # ---
 # name: TestCohort.test_cohortpeople_with_not_in_cohort_operator_for_behavioural_cohorts
   '''
-
+  /* cohort_calculation: */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -225,7 +225,7 @@
 # ---
 # name: TestCohort.test_cohortpeople_with_not_in_cohort_operator_for_behavioural_cohorts.1
   '''
-
+  /* cohort_calculation: */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,

ee/clickhouse/queries/test/__snapshots__/test_cohort_query.ambr

@@ -506,7 +506,7 @@
 # ---
 # name: TestCohortQuery.test_precalculated_cohort_filter_with_extra_filters.1
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2

ee/clickhouse/queries/test/__snapshots__/test_event_query.ambr

@@ -11,7 +11,7 @@
 # ---
 # name: TestEventQuery.test_account_filters.1
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2

frontend/src/lib/components/CommandPalette/DebugCHQueries.tsx

@@ -91,31 +91,29 @@ function DebugCHQueries(): JSX.Element {
 
     return (
         <>
-            {!!paths?.length && (
-                <div className="flex gap-4 items-end justify-between mb-4">
-                    <div className="flex flex-wrap gap-2">
-                        {paths.map(([path, count]) => (
-                            <LemonButton
-                                key={path}
-                                type={pathFilter === path ? 'primary' : 'tertiary'}
-                                size="small"
-                                onClick={() => (pathFilter === path ? setPathFilter(null) : setPathFilter(path))}
-                            >
-                                {path} <span className="ml-0.5 text-muted ligatures-none">({count})</span>
-                            </LemonButton>
-                        ))}
-                    </div>
-                    <LemonButton
-                        icon={<IconRefresh />}
-                        disabledReason={queriesLoading ? 'Loading…' : null}
-                        onClick={() => loadQueries()}
-                        size="small"
-                        type="secondary"
-                    >
-                        Refresh
-                    </LemonButton>
+            <div className="flex gap-4 items-end justify-between mb-4">
+                <div className="flex flex-wrap gap-2">
+                    {paths?.map(([path, count]) => (
+                        <LemonButton
+                            key={path}
+                            type={pathFilter === path ? 'primary' : 'tertiary'}
+                            size="small"
+                            onClick={() => (pathFilter === path ? setPathFilter(null) : setPathFilter(path))}
+                        >
+                            {path} <span className="ml-0.5 text-muted ligatures-none">({count})</span>
+                        </LemonButton>
+                    ))}
                 </div>
-            )}
+                <LemonButton
+                    icon={<IconRefresh />}
+                    disabledReason={queriesLoading ? 'Loading…' : null}
+                    onClick={() => loadQueries()}
+                    size="small"
+                    type="secondary"
+                >
+                    Refresh
+                </LemonButton>
+            </div>
 
             <LemonTable
                 columns={[

frontend/src/lib/components/CommandPalette/commandPaletteLogic.tsx

@@ -680,7 +680,7 @@ export const commandPaletteLogic = kea<commandPaletteLogicType>([
                     preflightLogic.values.preflight?.instance_preferences?.debug_queries
                         ? {
                               icon: IconDatabase,
-                              display: 'Debug ClickHouse Queries',
+                              display: 'Debug ClickHouse queries',
                               executor: () => openCHQueriesDebugModal(),
                           }
                         : [],

posthog/api/cohort.py

@@ -160,7 +160,7 @@ def create(self, validated_data: Dict, *args: Any, **kwargs: Any) -> Cohort:
         elif cohort.query is not None:
             raise ValidationError("Cannot create a dynamic cohort with a query. Set is_static to true.")
         else:
-            update_cohort(cohort)
+            update_cohort(cohort, initiating_user=request.user)
 
         report_user_action(request.user, "cohort created", cohort.get_analytics_metadata())
         return cohort
@@ -261,7 +261,7 @@ def update(self, cohort: Cohort, validated_data: Dict, *args: Any, **kwargs: Any
                 if request.FILES.get("csv"):
                     self._calculate_static_by_csv(request.FILES["csv"], cohort)
             else:
-                update_cohort(cohort)
+                update_cohort(cohort, initiating_user=request.user)
 
         report_user_action(
             request.user,

posthog/api/test/__snapshots__/test_cohort.ambr

@@ -11,7 +11,7 @@
 # ---
 # name: TestCohort.test_async_deletion_of_cohort.1
   '''
-  /* user_id:126 celery:posthog.tasks.calculate_cohort.calculate_cohort_ch */
+  /* user_id:126 cohort_calculation:posthog.tasks.calculate_cohort.calculate_cohort_ch */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -84,7 +84,7 @@
 # ---
 # name: TestCohort.test_async_deletion_of_cohort.2
   '''
-  /* user_id:126 celery:posthog.tasks.calculate_cohort.calculate_cohort_ch */
+  /* user_id:126 cohort_calculation:posthog.tasks.calculate_cohort.calculate_cohort_ch */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2
@@ -114,7 +114,7 @@
 # ---
 # name: TestCohort.test_async_deletion_of_cohort.5
   '''
-  /* user_id:126 celery:posthog.tasks.calculate_cohort.calculate_cohort_ch */
+  /* user_id:126 cohort_calculation:posthog.tasks.calculate_cohort.calculate_cohort_ch */
   INSERT INTO cohortpeople
   SELECT id,
          2 as cohort_id,
@@ -148,7 +148,7 @@
 # ---
 # name: TestCohort.test_async_deletion_of_cohort.6
   '''
-  /* user_id:126 celery:posthog.tasks.calculate_cohort.calculate_cohort_ch */
+  /* user_id:126 cohort_calculation:posthog.tasks.calculate_cohort.calculate_cohort_ch */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2

posthog/api/test/__snapshots__/test_feature_flag.ambr

@@ -90,7 +90,7 @@
 # ---
 # name: TestBlastRadius.test_user_blast_radius_with_multiple_precalculated_cohorts.1
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2
@@ -110,7 +110,7 @@
 # ---
 # name: TestBlastRadius.test_user_blast_radius_with_multiple_precalculated_cohorts.3
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2
@@ -213,7 +213,7 @@
 # ---
 # name: TestBlastRadius.test_user_blast_radius_with_multiple_static_cohorts.4
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2
@@ -233,7 +233,7 @@
 # ---
 # name: TestBlastRadius.test_user_blast_radius_with_multiple_static_cohorts.6
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2
@@ -308,7 +308,7 @@
 # ---
 # name: TestBlastRadius.test_user_blast_radius_with_single_cohort.3
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2

posthog/api/test/test_cohort.py

@@ -16,7 +16,7 @@
 from posthog.models.cohort import Cohort
 from posthog.models.team.team import Team
 from posthog.schema import PropertyOperator
-from posthog.tasks.calculate_cohort import calculate_cohort_from_list
+from posthog.tasks.calculate_cohort import calculate_cohort_ch, calculate_cohort_from_list
 from posthog.tasks.tasks import clickhouse_clear_removed_data
 from posthog.test.base import (
     APIBaseTest,
@@ -35,8 +35,9 @@ def capture_select_queries(self):
         return self.capture_queries(("INSERT INTO cohortpeople", "SELECT", "ALTER", "select", "DELETE"))
 
     @patch("posthog.api.cohort.report_user_action")
-    @patch("posthog.tasks.calculate_cohort.calculate_cohort_ch.delay")
-    def test_creating_update_and_calculating(self, patch_calculate_cohort, patch_capture):
+    @patch("posthog.tasks.calculate_cohort.calculate_cohort_ch.delay", side_effect=calculate_cohort_ch)
+    @patch("posthog.models.cohort.util.sync_execute", side_effect=sync_execute)
+    def test_creating_update_and_calculating(self, patch_sync_execute, patch_calculate_cohort, patch_capture):
         self.team.app_urls = ["http://somewebsite.com"]
         self.team.save()
         Person.objects.create(team=self.team, properties={"team_id": 5})
@@ -74,21 +75,24 @@ def test_creating_update_and_calculating(self, patch_calculate_cohort, patch_cap
             },
         )
 
-        response = self.client.patch(
-            f"/api/projects/{self.team.id}/cohorts/{response.json()['id']}",
-            data={
-                "name": "whatever2",
-                "description": "A great cohort!",
-                "groups": [{"properties": {"team_id": 6}}],
-                "created_by": "something something",
-                "last_calculation": "some random date",
-                "errors_calculating": 100,
-                "deleted": False,
-            },
-        )
-        self.assertEqual(response.status_code, 200, response.content)
-        self.assertDictContainsSubset({"name": "whatever2", "description": "A great cohort!"}, response.json())
-        self.assertEqual(patch_calculate_cohort.call_count, 2)
+        with self.capture_queries("INSERT INTO cohortpeople") as insert_statements:
+            response = self.client.patch(
+                f"/api/projects/{self.team.id}/cohorts/{response.json()['id']}",
+                data={
+                    "name": "whatever2",
+                    "description": "A great cohort!",
+                    "groups": [{"properties": {"team_id": 6}}],
+                    "created_by": "something something",
+                    "last_calculation": "some random date",
+                    "errors_calculating": 100,
+                    "deleted": False,
+                },
+            )
+            self.assertEqual(response.status_code, 200, response.content)
+            self.assertDictContainsSubset({"name": "whatever2", "description": "A great cohort!"}, response.json())
+            self.assertEqual(patch_calculate_cohort.call_count, 2)
+
+            self.assertIn(f" user_id:{self.user.id} ", insert_statements[0])
 
         # Assert analytics are sent
         patch_capture.assert_called_with(

posthog/hogql/functions/test/test_cohort.py

@@ -44,7 +44,7 @@ def test_in_cohort_dynamic(self):
             team=self.team,
             groups=[{"properties": [{"key": "$os", "value": "Chrome", "type": "person"}]}],
         )
-        recalculate_cohortpeople(cohort, pending_version=0)
+        recalculate_cohortpeople(cohort, pending_version=0, initiating_user_id=None)
         response = execute_hogql_query(
             f"SELECT event FROM events WHERE person_id IN COHORT {cohort.pk} AND event='{random_uuid}'",
             self.team,

posthog/hogql/test/test_query.py

@@ -411,7 +411,7 @@ def test_prop_cohort_basic(self):
                 ],
                 name="cohort",
             )
-            recalculate_cohortpeople(cohort, pending_version=0)
+            recalculate_cohortpeople(cohort, pending_version=0, initiating_user_id=None)
             with override_settings(PERSON_ON_EVENTS_V2_OVERRIDE=False):
                 response = execute_hogql_query(
                     "SELECT event, count() FROM events WHERE {cohort_filter} GROUP BY event",

posthog/hogql/transforms/test/test_in_cohort.py

@@ -44,7 +44,7 @@ def test_in_cohort_dynamic(self):
             team=self.team,
             groups=[{"properties": [{"key": "$os", "value": "Chrome", "type": "person"}]}],
         )
-        recalculate_cohortpeople(cohort, pending_version=0)
+        recalculate_cohortpeople(cohort, pending_version=0, initiating_user_id=None)
         response = execute_hogql_query(
             f"SELECT event FROM events WHERE person_id IN COHORT {cohort.pk} AND event='{random_uuid}'",
             self.team,
@@ -146,7 +146,7 @@ def test_in_cohort_conjoined_dynamic(self):
             team=self.team,
             groups=[{"properties": [{"key": "$os", "value": "Chrome", "type": "person"}]}],
         )
-        recalculate_cohortpeople(cohort, pending_version=0)
+        recalculate_cohortpeople(cohort, pending_version=0, initiating_user_id=None)
         response = execute_hogql_query(
             f"SELECT event FROM events WHERE person_id IN COHORT {cohort.pk} AND event='{random_uuid}'",
             self.team,

posthog/hogql_queries/insights/test/__snapshots__/test_lifecycle_query_runner.ambr

@@ -11,7 +11,7 @@
 # ---
 # name: TestLifecycleQueryRunner.test_cohort_filter.1
   '''
-
+  /* cohort_calculation: */
   SELECT count(DISTINCT person_id)
   FROM cohortpeople
   WHERE team_id = 2