EOS/tests/test_elecpricesmard.py

# ruff: noqa: S101

import json
from unittest.mock import Mock, patch

import pytest

from akkudoktoreos.core.coreabc import get_ems
from akkudoktoreos.prediction.elecprice import ElecPriceCommonSettings
from akkudoktoreos.prediction.elecpricesmard import ElecPriceSMARD
from akkudoktoreos.utils.datetimeutil import to_datetime


@pytest.fixture
def provider(config_eos):
    """Configure and return the direct SMARD singleton provider."""
    config_eos.elecprice = ElecPriceCommonSettings(provider="ElecPriceSMARD")
    get_ems().set_start_datetime(
        to_datetime("2026-07-27 00:00:00", in_timezone="Europe/Berlin")
    )
    provider = ElecPriceSMARD()
    provider.highest_orig_datetime = None
    assert provider.enabled()
    provider._db_reset_state()
    return provider


def _response(payload):
    response = Mock()
    response.content = json.dumps(payload).encode()
    response.raise_for_status.return_value = None
    return response


@patch("akkudoktoreos.prediction.elecpricesmard.requests.get")
def test_request_forecast_fetches_index_and_overlapping_chunks(mock_get, provider):
    """SMARD index and weekly chunks are combined, sorted, and stripped of null values."""
    chunk_start = 1785103200000
    mock_get.side_effect = [
        _response({"timestamps": [chunk_start]}),
        _response(
            {
                "meta_data": {"version": 1, "created": 1785500527370},
                "series": [
                    [1785103200000, 86.04],
                    [1785106800000, None],
                    [1785110400000, -1.25],
                ],
            }
        ),
    ]

    result = provider._request_forecast(
        start_date="2026-07-27", force_update=True
    )

    assert result.unix_seconds == [1785103200, 1785110400]
    assert result.price == [86.04, -1.25]
    assert result.license_info == "CC BY 4.0 Bundesnetzagentur | SMARD.de"
    assert mock_get.call_count == 2
    assert mock_get.call_args_list[0].args[0].endswith("/4169/DE/index_quarterhour.json")
    assert mock_get.call_args_list[1].args[0].endswith(
        "/4169/DE/4169_DE_quarterhour_1785103200000.json"
    )


def test_chunk_selection_includes_preceding_overlapping_chunk(provider):
    """A range beginning mid-week includes the chunk that started before it."""
    index = provider._validate_index(
        json.dumps({"timestamps": [1000, 2000, 3000]}).encode()
    )
    start = to_datetime(2.5, in_timezone="UTC")
    end = to_datetime(3.5, in_timezone="UTC")

    assert provider._chunk_timestamps(index, start, end) == [2000, 3000]


def test_smard_provider_is_enabled(provider):
    assert provider.enabled()