diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/conftest.py | 80 | ||||
| -rw-r--r-- | tests/test_krr.py | 22 |
2 files changed, 91 insertions, 11 deletions
diff --git a/tests/conftest.py b/tests/conftest.py new file mode 100644 index 0000000..ab9ca01 --- /dev/null +++ b/tests/conftest.py @@ -0,0 +1,80 @@ +from unittest.mock import AsyncMock, MagicMock, PropertyMock, patch +import pytest +import numpy as np +from datetime import datetime, timedelta +import random +from robusta_krr.api.models import K8sObjectData, PodData, ResourceAllocations, ResourceHistoryData + + +TEST_OBJECT = K8sObjectData( + cluster="mock-cluster", + name="mock-object-1", + container="mock-container-1", + pods=[ + PodData(name="mock-pod-1", deleted=False), + PodData(name="mock-pod-2", deleted=False), + PodData(name="mock-pod-3", deleted=True), + ], + namespace="default", + kind="Deployment", + allocations=ResourceAllocations( + requests={"cpu": 1, "memory": 1}, # type: ignore + limits={"cpu": 2, "memory": 2}, # type: ignore + ), +) + + +@pytest.fixture(autouse=True, scope="session") +def mock_list_clusters(): + with patch( + "robusta_krr.core.integrations.kubernetes.KubernetesLoader.list_clusters", + new=AsyncMock(return_value=[TEST_OBJECT.cluster]), + ): + yield + + +@pytest.fixture(autouse=True, scope="session") +def mock_list_scannable_objects(): + with patch( + "robusta_krr.core.integrations.kubernetes.KubernetesLoader.list_scannable_objects", + new=AsyncMock(return_value=[TEST_OBJECT]), + ): + yield + + +@pytest.fixture(autouse=True, scope="session") +def mock_config_loaded(): + with patch("robusta_krr.core.models.config.Config.config_loaded", new_callable=PropertyMock) as mock_config: + mock_config.return_value = True + yield + + +@pytest.fixture(autouse=True, scope="session") +def mock_prometheus_loader(): + now = datetime.now() + start = now - timedelta(hours=1) + now_ts, start_ts = now.timestamp(), start.timestamp() + metric_points_data = np.array([(t, random.randrange(0, 100)) for t in np.linspace(start_ts, now_ts, 3600)]) + + with patch( + "robusta_krr.core.integrations.prometheus.loader.PrometheusLoader.gather_data", + new=AsyncMock( + return_value=ResourceHistoryData( + data={pod.name: metric_points_data for pod in TEST_OBJECT.pods}, + metric={ # type: ignore + "query": f"example_promql_metric{{pod_name=~\"{'|'.join(pod.name for pod in TEST_OBJECT.pods)}\"}}", + "start_time": start, + "end_time": now, + "step": "30s", + }, + ) + ), + ) as mock_prometheus_loader: + mock_prometheus_loader + yield + + +@pytest.fixture(autouse=True, scope="session") +def mock_prometheus_init(): + with patch("robusta_krr.core.integrations.prometheus.loader.PrometheusLoader.__init__", return_value=None): + yield diff --git a/tests/test_krr.py b/tests/test_krr.py index 23dab01..f013ee4 100644 --- a/tests/test_krr.py +++ b/tests/test_krr.py @@ -1,8 +1,3 @@ -""" - Test the krr command line interface and a general execution. - Requires a running kubernetes cluster with the kubectl command configured. -""" - import json import pytest @@ -36,14 +31,19 @@ def test_run(log_flag: str): @pytest.mark.parametrize("format", ["json", "yaml", "table", "pprint"]) def test_output_formats(format: str): - result = runner.invoke(app, [STRATEGY_NAME, "-q", "-f", format, "--namespace", "default"]) + result = runner.invoke(app, [STRATEGY_NAME, "-q", "-f", format]) try: assert result.exit_code == 0, result.exc_info except AssertionError as e: raise e from result.exception - if format == "json": - assert json.loads(result.stdout), result.stdout - - if format == "yaml": - assert yaml.safe_load(result.stdout), result.stdout + try: + if format == "json": + json_output = json.loads(result.stdout) + assert json_output, result.stdout + assert len(json_output["scans"]) > 0, result.stdout + + if format == "yaml": + assert yaml.safe_load(result.stdout), result.stdout + except Exception as e: + raise Exception(result.stdout) from e |