Fix tests

ThiagoAVicente · ThiagoAVicente · commit 05c2dcb955e0 · 2025-12-11T11:48:33.000Z
diff --git a/tests/test_clickhouse_service.py b/tests/test_clickhouse_service.py
@@ -263,6 +263,7 @@ def test_query_processed_latency_success(self, clickhouse_service, mock_clickhou
             start_time=start_time,
             end_time=end_time,
             cell_index=1,
+            window_duration_seconds=10,
             offset=0,
             limit=100
         )
@@ -292,6 +293,7 @@ def test_query_processed_latency_empty_result(self, clickhouse_service, mock_cli
             start_time=start_time,
             end_time=end_time,
             cell_index=1,
+            window_duration_seconds=10,
             offset=0,
             limit=100
         )
@@ -311,6 +313,7 @@ def test_query_processed_latency_with_pagination(self, clickhouse_service, mock_
             start_time=start_time,
             end_time=end_time,
             cell_index=1,
+            window_duration_seconds=10,
             offset=50,
             limit=25
         )
@@ -346,6 +349,7 @@ def test_query_processed_latency_multiple_rows(self, clickhouse_service, mock_cl
             start_time=start_time,
             end_time=end_time,
             cell_index=1,
+            window_duration_seconds=10,
             offset=0,
             limit=100
         )
@@ -380,6 +384,7 @@ def test_query_processed_latency_with_nullable_fields(self, clickhouse_service,
             start_time=start_time,
             end_time=end_time,
             cell_index=1,
+            window_duration_seconds=10,
             offset=0,
             limit=100
         )
diff --git a/tests/test_latency_endpoint.py b/tests/test_latency_endpoint.py
@@ -80,7 +80,8 @@ def test_get_processed_latency_success(self, test_client, mock_clickhouse_servic
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -105,7 +106,8 @@ def test_get_processed_latency_empty_result(self, test_client, mock_clickhouse_s
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -120,6 +122,7 @@ def test_get_processed_latency_missing_required_params(self, test_client, mock_c
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME
+
             }
         )
 
@@ -136,7 +139,8 @@ def test_get_processed_latency_with_pagination(self, test_client, mock_clickhous
                 "end_time": SAMPLE_END_TIME,
                 "cell_index": 1,
                 "offset": 50,
-                "limit": 25
+                "limit": 25,
+                "window_duration_seconds":300
             }
         )
 
@@ -155,7 +159,8 @@ def test_get_processed_latency_invalid_limit(self, test_client, mock_clickhouse_
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
                 "cell_index": 1,
-                "limit": 5000  # Exceeds max of 1000
+                "limit": 5000,  # Exceeds max of 1000
+                "window_duration_seconds":300
             }
         )
 
@@ -169,7 +174,8 @@ def test_get_processed_latency_invalid_offset(self, test_client, mock_clickhouse
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
                 "cell_index": 1,
-                "offset": -10
+                "offset": -10,
+                "window_duration_seconds":300
             }
         )
 
@@ -184,7 +190,8 @@ def test_get_processed_latency_service_error(self, test_client, mock_clickhouse_
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -202,7 +209,8 @@ def test_get_processed_latency_multiple_results(self, test_client, mock_clickhou
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -219,7 +227,8 @@ def test_get_processed_latency_default_pagination(self, test_client, mock_clickh
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -252,7 +261,8 @@ def test_get_processed_latency_with_null_fields(self, test_client, mock_clickhou
             params={
                 "start_time": SAMPLE_START_TIME,
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 
@@ -268,7 +278,8 @@ def test_get_processed_latency_invalid_datetime_format(self, test_client, mock_c
             params={
                 "start_time": "invalid-timestamp",
                 "end_time": SAMPLE_END_TIME,
-                "cell_index": 1
+                "cell_index": 1,
+                "window_duration_seconds":300
             }
         )
 

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,8 @@ def test_get_processed_latency_success(self, test_client, mock_clickhouse_servic`
`80`	`80`	`params={`
`81`	`81`	`"start_time": SAMPLE_START_TIME,`
`82`	`82`	`"end_time": SAMPLE_END_TIME,`
`83`		`- "cell_index": 1`
	`83`	`+ "cell_index": 1,`
	`84`	`+ "window_duration_seconds":300`
`84`	`85`	`}`
`85`	`86`	`)`
`86`	`87`
`@@ -105,7 +106,8 @@ def test_get_processed_latency_empty_result(self, test_client, mock_clickhouse_s`
`105`	`106`	`params={`
`106`	`107`	`"start_time": SAMPLE_START_TIME,`
`107`	`108`	`"end_time": SAMPLE_END_TIME,`
`108`		`- "cell_index": 1`
	`109`	`+ "cell_index": 1,`
	`110`	`+ "window_duration_seconds":300`
`109`	`111`	`}`
`110`	`112`	`)`
`111`	`113`
`@@ -120,6 +122,7 @@ def test_get_processed_latency_missing_required_params(self, test_client, mock_c`
`120`	`122`	`params={`
`121`	`123`	`"start_time": SAMPLE_START_TIME,`
`122`	`124`	`"end_time": SAMPLE_END_TIME`
	`125`	`+`
`123`	`126`	`}`
`124`	`127`	`)`
`125`	`128`
`@@ -136,7 +139,8 @@ def test_get_processed_latency_with_pagination(self, test_client, mock_clickhous`
`136`	`139`	`"end_time": SAMPLE_END_TIME,`
`137`	`140`	`"cell_index": 1,`
`138`	`141`	`"offset": 50,`
`139`		`- "limit": 25`
	`142`	`+ "limit": 25,`
	`143`	`+ "window_duration_seconds":300`
`140`	`144`	`}`
`141`	`145`	`)`
`142`	`146`
`@@ -155,7 +159,8 @@ def test_get_processed_latency_invalid_limit(self, test_client, mock_clickhouse_`
`155`	`159`	`"start_time": SAMPLE_START_TIME,`
`156`	`160`	`"end_time": SAMPLE_END_TIME,`
`157`	`161`	`"cell_index": 1,`
`158`		`- "limit": 5000 # Exceeds max of 1000`
	`162`	`+ "limit": 5000, # Exceeds max of 1000`
	`163`	`+ "window_duration_seconds":300`
`159`	`164`	`}`
`160`	`165`	`)`
`161`	`166`
`@@ -169,7 +174,8 @@ def test_get_processed_latency_invalid_offset(self, test_client, mock_clickhouse`
`169`	`174`	`"start_time": SAMPLE_START_TIME,`
`170`	`175`	`"end_time": SAMPLE_END_TIME,`
`171`	`176`	`"cell_index": 1,`
`172`		`- "offset": -10`
	`177`	`+ "offset": -10,`
	`178`	`+ "window_duration_seconds":300`
`173`	`179`	`}`
`174`	`180`	`)`
`175`	`181`
`@@ -184,7 +190,8 @@ def test_get_processed_latency_service_error(self, test_client, mock_clickhouse_`
`184`	`190`	`params={`
`185`	`191`	`"start_time": SAMPLE_START_TIME,`
`186`	`192`	`"end_time": SAMPLE_END_TIME,`
`187`		`- "cell_index": 1`
	`193`	`+ "cell_index": 1,`
	`194`	`+ "window_duration_seconds":300`
`188`	`195`	`}`
`189`	`196`	`)`
`190`	`197`
`@@ -202,7 +209,8 @@ def test_get_processed_latency_multiple_results(self, test_client, mock_clickhou`
`202`	`209`	`params={`
`203`	`210`	`"start_time": SAMPLE_START_TIME,`
`204`	`211`	`"end_time": SAMPLE_END_TIME,`
`205`		`- "cell_index": 1`
	`212`	`+ "cell_index": 1,`
	`213`	`+ "window_duration_seconds":300`
`206`	`214`	`}`
`207`	`215`	`)`
`208`	`216`
`@@ -219,7 +227,8 @@ def test_get_processed_latency_default_pagination(self, test_client, mock_clickh`
`219`	`227`	`params={`
`220`	`228`	`"start_time": SAMPLE_START_TIME,`
`221`	`229`	`"end_time": SAMPLE_END_TIME,`
`222`		`- "cell_index": 1`
	`230`	`+ "cell_index": 1,`
	`231`	`+ "window_duration_seconds":300`
`223`	`232`	`}`
`224`	`233`	`)`
`225`	`234`
`@@ -252,7 +261,8 @@ def test_get_processed_latency_with_null_fields(self, test_client, mock_clickhou`
`252`	`261`	`params={`
`253`	`262`	`"start_time": SAMPLE_START_TIME,`
`254`	`263`	`"end_time": SAMPLE_END_TIME,`
`255`		`- "cell_index": 1`
	`264`	`+ "cell_index": 1,`
	`265`	`+ "window_duration_seconds":300`
`256`	`266`	`}`
`257`	`267`	`)`
`258`	`268`
`@@ -268,7 +278,8 @@ def test_get_processed_latency_invalid_datetime_format(self, test_client, mock_c`
`268`	`278`	`params={`
`269`	`279`	`"start_time": "invalid-timestamp",`
`270`	`280`	`"end_time": SAMPLE_END_TIME,`
`271`		`- "cell_index": 1`
	`281`	`+ "cell_index": 1,`
	`282`	`+ "window_duration_seconds":300`
`272`	`283`	`}`
`273`	`284`	`)`
`274`	`285`