fix: add email source to mixpanel events

Author: aarthy-dk

testgen/__main__.py

@@ -32,7 +32,7 @@
 from testgen.commands.run_test_execution import run_test_execution
 from testgen.commands.run_test_metadata_exporter import run_test_metadata_exporter
 from testgen.commands.run_upgrade_db_config import get_schema_revision, is_db_revision_up_to_date, run_upgrade_db_config
-from testgen.commands.test_generation import run_test_generation
+from testgen.commands.test_generation import run_monitor_generation, run_test_generation
 from testgen.common import (
     configure_logging,
     display_service,
@@ -176,6 +176,20 @@ def run_generation(test_suite_id: str | None = None, table_group_id: str | None
     click.echo("\n" + message)
 
 
+@cli.command("run-monitor-generation", help="Generates or refreshes the monitors for a table group.")
+@click.option(
+    "-t",
+    "--test-suite-id",
+    required=True,
+    type=click.STRING,
+    help="ID of the monitor suite to generate",
+)
+@with_database_session
+def generate_monitors(test_suite_id: str):
+    click.echo(f"run-monitor-generation for suite: {test_suite_id}")
+    run_monitor_generation(test_suite_id, ["Freshness_Trend", "Volume_Trend", "Schema_Drift"])
+
+
 @register_scheduler_job
 @cli.command("run-tests", help="Performs tests defined for a test suite.")
 @click.option(

testgen/common/notifications/monitor_run.py

@@ -230,6 +230,7 @@ def send_monitor_notifications(test_run: TestRun, result_list_ct=20):
                 str(table_group.id),
                 "&table_name_filter=" if table_name else "",
                 table_name if table_name else "",
+                "&source=email",
             )
         )
         try:

testgen/common/notifications/profiling_run.py

@@ -258,7 +258,7 @@ def send_profiling_run_notifications(profiling_run: ProfilingRun, result_list_ct
         return
 
     profiling_run_issues_url = "".join(
-        (PersistedSetting.get("BASE_URL", ""), "/profiling-runs:hygiene?run_id=", str(profiling_run.id))
+        (PersistedSetting.get("BASE_URL", ""), "/profiling-runs:hygiene?run_id=", str(profiling_run.id), "&source=email")
     )
 
     hygiene_issues_summary = []
@@ -304,7 +304,7 @@ def send_profiling_run_notifications(profiling_run: ProfilingRun, result_list_ct
             "id": str(profiling_run.id),
             "issues_url": profiling_run_issues_url,
             "results_url": "".join(
-                (PersistedSetting.get("BASE_URL", ""), "/profiling-runs:results?run_id=", str(profiling_run.id))
+                (PersistedSetting.get("BASE_URL", ""), "/profiling-runs:results?run_id=", str(profiling_run.id), "&source=email")
             ),
             "start_time": profiling_run.profiling_starttime,
             "end_time": profiling_run.profiling_endtime,

testgen/common/notifications/score_drop.py

@@ -180,6 +180,7 @@ def send_score_drop_notifications(notification_data: list[tuple[ScoreDefinition,
                         PersistedSetting.get("BASE_URL", ""),
                         "/quality-dashboard:score-details?definition_id=",
                         str(definition.id),
+                        "&source=email",
                     )
                 ),
                 "diff": context_diff,

testgen/common/notifications/test_run.py

@@ -328,6 +328,7 @@ def send_test_run_notifications(test_run: TestRun, result_list_ct=20, result_sta
             PersistedSetting.get("BASE_URL", ""),
             "/test-runs:results?run_id=",
             str(test_run.id),
+            "&source=email",
         )
     )
 

testgen/ui/components/frontend/js/pages/edit_table_monitors.js

@@ -234,7 +234,7 @@ const EditTableMonitors = (/** @type Properties */ props) => {
                         updated_definitions: Object.values(updatedDefinitions.val),
                         new_metrics: Object.values(newMetrics.val),
                         deleted_metric_ids: deletedMetricIds.val,
-                        close: false,
+                        close: true,
                     };
                     emitEvent('SaveTestDefinition', { payload });
                 },

testgen/ui/components/frontend/js/pages/project_dashboard.js

@@ -261,7 +261,6 @@ const TableGroupLatestProfile = (/** @type TableGroupSummary */ tableGroup) => {
             })
             : '',
         ),
-        div({ style: 'flex: 1 1 120px;' })
     );
 };
 

testgen/ui/navigation/router.py

@@ -27,7 +27,7 @@ def __init__(
         self._routes = {route.path: route(self) for route in routes} if routes else {}
         self._pending_navigation: dict | None = None
 
-    def _init_session(self):
+    def _init_session(self, url: str):
         # Clear cache on initial load or page refresh
         st.cache_data.clear()
 
@@ -37,13 +37,16 @@ def _init_session(self):
         except Exception as e:
             LOG.exception("Error capturing the base URL")
 
+        source = st.query_params.pop("source", None)
+        MixpanelService().send_event(f"nav-{url}", page_load=True, source=source)
+
     def run(self) -> None:
         streamlit_pages = [route.streamlit_page for route in self._routes.values()]
 
         current_page = st.navigation(streamlit_pages, position="hidden")
 
         if not session.initialized:
-            self._init_session()
+            self._init_session(url=current_page.url_path)
             session.initialized = True
 
         # This hack is needed because the auth cookie is not set if navigation happens immediately after login