Replace numbers in session tests

Author: robbie-c

posthog/clickhouse/test/test_raw_sessions_v3_model.py

@@ -30,6 +30,8 @@ def create_session_id():
 
 @snapshot_clickhouse_queries
 class TestRawSessionsModel(ClickhouseTestMixin, BaseTest):
+    snapshot_replace_all_numbers = True
+
     def select_by_session_id(self, session_id):
         flush_persons_and_events()
         return query_with_columns(

posthog/hogql/database/schema/test/test_sessions_v3.py

@@ -26,6 +26,8 @@
 
 @snapshot_clickhouse_queries
 class TestSessionsV3(ClickhouseTestMixin, APIBaseTest):
+    snapshot_replace_all_numbers = True
+
     def __execute(
         self,
         query,

posthog/test/base.py

@@ -1343,7 +1343,8 @@ def wrapped(self, *args, **kwargs):
 
         for query in queries:
             if "FROM system.columns" not in query:
-                self.assertQueryMatchesSnapshot(query)
+                replace_all_numbers = getattr(self, "snapshot_replace_all_numbers", False)
+                self.assertQueryMatchesSnapshot(query, replace_all_numbers=replace_all_numbers)
 
     return wrapped