configurable query_interval

robusta-dev · Jul 5, 2024 · 0e9d2bd · 0e9d2bd
1 parent d3450e6
commit 0e9d2bd
Show file tree

Hide file tree

Showing 5 changed files with 17 additions and 2 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/prometrix/models/prometheus_config.py b/prometrix/models/prometheus_config.py
@@ -26,6 +26,7 @@ class PrometheusConfig(BaseModel):
         PrometheusApis.FLAGS,
     ]
     query_step: str = "5m"
+    query_interval: str = "1d"
 
 
 class AWSPrometheusConfig(PrometheusConfig):

diff --git a/pyproject.toml b/pyproject.toml
@@ -27,6 +27,7 @@ optional = true
 
 [tool.poetry.group.test.dependencies]
 pyyaml = "^6.0.0"
+pytimeparse = "^1.1.0"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/tests/config_github_kind.yaml b/tests/config_github_kind.yaml
@@ -3,3 +3,4 @@ testConfig:
     params:
       url: http://localhost:9090
       query_step: "5s"
+      query_interval: "15s"
diff --git a/tests/main.py b/tests/main.py
@@ -6,6 +6,7 @@
 import yaml
 from prometheus_api_client import PrometheusApiClientException
 from pydantic import ValidationError
+from pytimeparse import parse as parse_timedelta
 
 from prometrix import (AWSPrometheusConfig, AzurePrometheusConfig,
                        CoralogixPrometheusConfig, CustomPrometheusConnect,
@@ -56,7 +57,7 @@ def run_test(test_type: str, config: PrometheusConfig) -> bool:
             return False
         result = prom_cli.safe_custom_query_range(
             query="container_memory_working_set_bytes",
-            start_time=datetime.now() - timedelta(days=1),
+            start_time=datetime.now() - parse_timedelta(config.query_interval),
             end_time=datetime.now(),
             step=config.query_step,
         )