EOS/tests/test_eosdashconfig.py

"""Test suite for the EOS Dash configuration module.

This module contains tests for utility functions related to retrieving and processing
configuration data using Pydantic models.
"""

import json
from pathlib import Path
from typing import Union

import pytest
from pydantic.fields import FieldInfo

from akkudoktoreos.core.pydantic import PydanticBaseModel
from akkudoktoreos.prediction.pvforecast import PVForecastPlaneSetting
from akkudoktoreos.server.dash.configuration import (
    configuration,
    get_default_value,
    get_nested_value,
    resolve_nested_types,
)

DIR_TESTDATA = Path(__file__).absolute().parent.joinpath("testdata")

FILE_TESTDATA_EOSSERVER_CONFIG_1 = DIR_TESTDATA.joinpath("eosserver_config_1.json")


class SampleModel(PydanticBaseModel):
    field1: str = "default_value"
    field2: int = 10


class TestEOSdashConfig:
    """Test case for EOS Dash configuration utility functions.

    This class tests functions for retrieving nested values, extracting default values,
    resolving nested types, and generating configuration details from Pydantic models.
    """

    def test_get_nested_value_from_dict(self):
        """Test retrieving a nested value from a dictionary using a sequence of keys."""
        data = {"a": {"b": {"c": 42}}}
        assert get_nested_value(data, ["a", "b", "c"]) == 42
        assert get_nested_value(data, ["a", "x"], default="not found") == "not found"
        with pytest.raises(TypeError):
            get_nested_value("not_a_dict", ["a"])  # type: ignore

    def test_get_nested_value_from_list(self):
        """Test retrieving a nested value from a list using a sequence of keys."""
        data = {"a": {"b": {"c": [42]}}}
        assert get_nested_value(data, ["a", "b", "c", 0]) == 42
        assert get_nested_value(data, ["a", "b", "c", "0"]) == 42

    def test_get_default_value(self):
        """Test retrieving the default value of a field based on FieldInfo metadata."""
        field_info = FieldInfo(default="test_value")
        assert get_default_value(field_info, True) == "test_value"
        field_info_no_default = FieldInfo()
        assert get_default_value(field_info_no_default, True) == ""
        assert get_default_value(field_info, False) == "N/A"

    def test_resolve_nested_types(self):
        """Test resolving nested types within a field, ensuring correct type extraction."""
        nested_types = resolve_nested_types(Union[int, str], [])
        assert (int, []) in nested_types
        assert (str, []) in nested_types

    def test_configuration(self):
        """Test extracting configuration details from a Pydantic model based on provided values."""
        values = {"field1": "custom_value", "field2": 20}
        config = configuration(SampleModel, values)
        assert any(
            item["name"] == "field1" and item["value"] == '"custom_value"' for item in config
        )
        assert any(item["name"] == "field2" and item["value"] == "20" for item in config)

    def test_configuration_eos(self, config_eos):
        """Test extracting EOS configuration details from EOS config based on provided values."""
        with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
            values = json.load(fd)
        config = configuration(config_eos, values)
        assert any(
            item["name"] == "server.eosdash_port" and item["value"] == "8504" for item in config
        )
        assert any(
            item["name"] == "server.eosdash_host" and item["value"] == '"0.0.0.0"'
            for item in config
        )

    def test_configuration_pvforecast_plane_settings(self):
        """Test extracting EOS PV forecast plane configuration details from EOS config based on provided values."""
        with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
            values = json.load(fd)
        config = configuration(
            PVForecastPlaneSetting(), values, values_prefix=["pvforecast", "planes", "0"]
        )
        assert any(
            item["name"] == "pvforecast.planes.0.surface_azimuth" and item["value"] == "-10"
            for item in config
        )
        assert any(
            item["name"] == "pvforecast.planes.0.userhorizon"
            and item["value"] == "[20, 27, 22, 20]"
            for item in config
        )