From 428d628c7f09c16733e6f65c41a3f03ceeac4db8 Mon Sep 17 00:00:00 2001
From: LeaveMyYard <33721692+LeaveMyYard@users.noreply.github.com>
Date: Mon, 24 Jul 2023 20:47:51 +0300
Subject: Refactor strategy dependancy on metrics

---
 tests/conftest.py | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

(limited to 'tests/conftest.py')

diff --git a/tests/conftest.py b/tests/conftest.py
index 3b89e88..f39a30b 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -5,7 +5,7 @@ from unittest.mock import AsyncMock, patch
 import numpy as np
 import pytest
 
-from robusta_krr.api.models import K8sObjectData, PodData, ResourceAllocations, ResourceHistoryData
+from robusta_krr.api.models import K8sObjectData, PodData, ResourceAllocations, MetricPodData
 
 TEST_OBJECT = K8sObjectData(
     cluster="mock-cluster",
@@ -59,15 +59,7 @@ def mock_prometheus_loader():
     with patch(
         "robusta_krr.core.integrations.prometheus.loader.MetricsLoader.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",
-                },
-            )
+            return_value={pod.name: metric_points_data for pod in TEST_OBJECT.pods},
         ),
     ) as mock_prometheus_loader:
         mock_prometheus_loader
-- 
cgit v1.2.3