fix(chalice): fixed Math-operators validation

refactor(chalice): search for sessions that have events for heatmaps
tahayk · Jul 4, 2024 · 30cd56c · 30cd56c
1 parent 42aa035
commit 30cd56c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/api/chalicelib/core/heatmaps.py b/api/chalicelib/core/heatmaps.py
@@ -171,6 +171,10 @@ def search_short_session(data: schemas.HeatMapSessionsSearch, project_id, user_i
                                                           value=[0],
                                                           operator=schemas.MathOperator._greater))
 
+    data.filters.append(schemas.SessionSearchFilterSchema(type=schemas.FilterType.events_count,
+                                                          value=[0],
+                                                          operator=schemas.MathOperator._greater))
+
     full_args, query_part = sessions.search_query_parts(data=data, error_status=None, errors_only=False,
                                                         favorite_only=data.bookmarked, issue=None,
                                                         project_id=project_id, user_id=user_id)

diff --git a/ee/api/chalicelib/core/heatmaps.py b/ee/api/chalicelib/core/heatmaps.py
@@ -187,6 +187,10 @@ def search_short_session(data: schemas.HeatMapSessionsSearch, project_id, user_i
                                                               value=[0],
                                                               operator=schemas.MathOperator._greater))
 
+        data.filters.append(schemas.SessionSearchFilterSchema(type=schemas.FilterType.events_count,
+                                                              value=[0],
+                                                              operator=schemas.MathOperator._greater))
+
         full_args, query_part = sessions.search_query_parts(data=data, error_status=None, errors_only=False,
                                                             favorite_only=data.bookmarked, issue=None,
                                                             project_id=project_id, user_id=user_id)
@@ -318,6 +322,10 @@ def search_short_session(data: schemas.HeatMapSessionsSearch, project_id, user_i
                                                               value=[0],
                                                               operator=schemas.MathOperator._greater))
 
+        data.filters.append(schemas.SessionSearchFilterSchema(type=schemas.FilterType.events_count,
+                                                              value=[0],
+                                                              operator=schemas.MathOperator._greater))
+
         full_args, query_part = sessions.search_query_parts_ch(data=data, error_status=None, errors_only=False,
                                                                favorite_only=data.bookmarked, issue=None,
                                                                project_id=project_id, user_id=user_id)