"""Tests for dashboard.openstack_utils.audits."""
from unittest.mock import MagicMock, patch

from django.test import TestCase

from dashboard.openstack_utils.audits import convert_cpu_data, get_current_cluster_cpu


class ConvertCpuDataTest(TestCase):
    """Tests for convert_cpu_data."""

    def test_aggregates_cpu_usage_per_host(self):
        data = [
            {
                "metric": {"host": "compute-0", "instanceName": "inst1"},
                "values": [[1000, "10.0"], [1001, "20.0"]],
            },
            {
                "metric": {"host": "compute-0", "instanceName": "inst2"},
                "values": [[1000, "5.0"]],
            },
            {
                "metric": {"host": "compute-1", "instanceName": "inst3"},
                "values": [[1000, "30.0"]],
            },
        ]
        result = convert_cpu_data(data)
        self.assertIn("host", result.columns)
        self.assertIn("cpu_usage", result.columns)
        hosts = result["host"].tolist()
        self.assertEqual(len(hosts), 2)
        self.assertIn("compute-0", hosts)
        self.assertIn("compute-1", hosts)
        # compute-0: (10+20)/2 for ts 1000 and 5 for ts 1000 -> groupby host,timestamp sum -> then groupby host mean
        # For compute-0: two timestamps 1000 (10+5=15) and 1001 (20). Mean over timestamps = (15+20)/2 = 17.5
        # For compute-1: one value 30
        by_host = result.set_index("host")["cpu_usage"]
        self.assertAlmostEqual(by_host["compute-0"], 17.5)
        self.assertAlmostEqual(by_host["compute-1"], 30.0)

    def test_empty_data_returns_empty_dataframe_with_columns(self):
        result = convert_cpu_data([])
        self.assertIn("host", result.columns)
        self.assertIn("cpu_usage", result.columns)
        self.assertEqual(len(result), 0)


class GetCurrentClusterCpuTest(TestCase):
    """Tests for get_current_cluster_cpu."""

    @patch("dashboard.openstack_utils.audits.query_prometheus")
    def test_returns_empty_lists_when_no_data(self, mock_query):
        mock_query.return_value = []
        conn = MagicMock()
        result = get_current_cluster_cpu(conn)
        self.assertEqual(result["host_labels"], [])
        self.assertEqual(result["cpu_current"], [])

    @patch("dashboard.openstack_utils.audits.convert_cpu_data")
    @patch("dashboard.openstack_utils.audits.query_prometheus")
    def test_returns_host_labels_and_cpu_current(self, mock_query, mock_convert):
        import pandas as pd
        mock_query.return_value = [{"metric": {"host": "h0"}, "values": [[0, "1.0"]]}]
        mock_convert.return_value = pd.DataFrame({
            "host": ["compute-0", "compute-1"],
            "cpu_usage": [25.0, 35.0],
        })
        conn = MagicMock()
        result = get_current_cluster_cpu(conn)
        self.assertEqual(result["host_labels"], ["compute-0", "compute-1"])
        self.assertEqual(result["cpu_current"], [25.0, 35.0])