mirror of
https://github.com/Akkudoktor-EOS/EOS.git
synced 2026-08-05 16:36:12 +00:00
Bobby NoelteandGitHub
1905682113
Change PDF visualization to be created on demand and per optimization algorithm. The PDF for the GENETIC0 optimization is provided by the /visualization_results.pdf endpoint. There is no change in the interface. By this the optimization algorithm is offloaded from the PDF generation which spares some time. To cope with several users may call the /visualization_results.pdf endpoint at the same time the PDF is generated on the fly without any intermediate file taking the stored GENETIC0 solution as an input. SVG picture generation is removed as this would again create intermediate files. Chart pictures can easily be taken from the PDF. To allow on demand creation of the optimization results visualization the optimisation solution stored is extended by several new attributes. To keep the deprecated /optimize endpoint compatible the optimization solution is stripped to the legacy content before returned. Due to the extension of the solution the optimization tests were adapted to cover the extended content. The optimization tests are adapted to test the generated visualization report by the pypdf reader. Pypdf is added to the development dependencies. Besides the adaptation several fixes and improvements are added: * feat: extend /v1/prediction/series endpoint by resampling and filling Add parameters for resampling and filling. Add the processing parameter to control wether raw data or resampled data shall be returned. * feat: extend /v1/measurement/series endpoint by resampling and filling Add parameters for resampling and filling: Add the processing parameter to control wether raw data or resampled data shall be returned. * feat: standardize and improve API error response Use FASTApi exception handlers to provide a standardized API exception handling. All exceptions are logged. Exception traces are only returned if the new logging configuration parameter logging.api_logging_level is set to "DEBUG" or "TRACE". Avoids unwanted leackage of server internals on exceptions. * fix: align to intervall when resampling Ensure resampling is aligned to interval also when the buckets are shifted due to the align_to_intervall parameter is set. * chore: make dropna mandatory and default to True * chore: refactor key_to_xxx data management methods Make key_to_series the central method for data resampling and fill. Add a new key_to_raw_series to retrieve the data as it is stored (without resampling and filling). Users of key_to_series were mostly moved to key_to_raw_series as this resembles the former interface. Especially in predictions and tests this was done. * chore: create test data sub-directory for each optimization algorithm To prevent cluttering the test data directory and ease test data management for optimization algorithms each algorithm got it's own sub-directory. The current test data was moved to these sub-directories. * chore: update version Signed-off-by: Bobby Noelte <b0661n0e17e@gmail.com>
899 lines
35 KiB
Python
899 lines
35 KiB
Python
import json
|
|
import os
|
|
import random
|
|
import signal
|
|
import time
|
|
from http import HTTPStatus
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
import requests
|
|
|
|
from akkudoktoreos.utils.datetimeutil import to_datetime
|
|
|
|
DIR_TESTDATA = Path(__file__).absolute().parent.joinpath("testdata")
|
|
|
|
FILE_TESTDATA_EOSSERVER_CONFIG_1 = DIR_TESTDATA.joinpath("eosserver_config_1.json")
|
|
|
|
|
|
class TestSystem:
|
|
|
|
def test_prediction_brightsky(self, server_setup_for_class, is_system_test):
|
|
"""Test weather prediction by BrightSky."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
result = requests.get(f"{server}/v1/config", timeout=2)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Get testing config
|
|
config_json = result.json()
|
|
config_folder_path = Path(config_json["general"]["config_folder_path"])
|
|
# Assure we are working in test environment
|
|
assert str(config_folder_path).startswith(eos_dir)
|
|
|
|
result = requests.put(f"{server}/v1/config/weather/provider", json="BrightSky")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "BrightSky" in providers
|
|
|
|
if is_system_test:
|
|
result = requests.post(f"{server}/v1/prediction/update/BrightSky")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/series",
|
|
params={
|
|
"key": "weather_temp_air",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()
|
|
assert len(data["data"]) > 24
|
|
|
|
else:
|
|
pass
|
|
|
|
def test_prediction_clearoutside(self, server_setup_for_class, is_system_test):
|
|
"""Test weather prediction by ClearOutside."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
result = requests.put(f"{server}/v1/config/weather/provider", json="ClearOutside")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "ClearOutside" in providers
|
|
|
|
if is_system_test:
|
|
result = requests.post(f"{server}/v1/prediction/update/ClearOutside")
|
|
assert result.status_code == HTTPStatus.OK, f"Failed: {result.headers} {result.text}"
|
|
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/series",
|
|
params={
|
|
"key": "weather_temp_air",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()
|
|
assert len(data["data"]) > 24
|
|
|
|
else:
|
|
pass
|
|
|
|
def test_prediction_pvforecastakkudoktor(self, server_setup_for_class, is_system_test):
|
|
"""Test PV prediction by PVForecastAkkudoktor."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
# Reset config
|
|
with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
|
|
config = json.load(fd)
|
|
config["pvforecast"]["provider"] = "PVForecastAkkudoktor"
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "PVForecastAkkudoktor" in providers
|
|
|
|
if is_system_test:
|
|
result = requests.post(f"{server}/v1/prediction/update/PVForecastAkkudoktor")
|
|
assert result.status_code == HTTPStatus.OK, f"Failed: {result.headers} {result.text}"
|
|
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/series",
|
|
params={
|
|
"key": "pvforecast_ac_power",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()
|
|
assert len(data["data"]) > 24
|
|
|
|
else:
|
|
pass
|
|
|
|
def test_prediction_elecpriceakkudoktor(self, server_setup_for_class, is_system_test):
|
|
"""Test electricity price prediction by ElecPriceImport."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
# Reset config
|
|
with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
|
|
config = json.load(fd)
|
|
config["elecprice"]["provider"] = "ElecPriceAkkudoktor"
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "ElecPriceAkkudoktor" in providers
|
|
|
|
if is_system_test:
|
|
result = requests.post(f"{server}/v1/prediction/update/ElecPriceAkkudoktor")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/series",
|
|
params={
|
|
"key": "elecprice_marketprice_wh",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()
|
|
assert len(data["data"]) > 24
|
|
|
|
else:
|
|
pass
|
|
|
|
def test_prediction_loadakkudoktor(self, server_setup_for_class, is_system_test):
|
|
"""Test load prediction by LoadAkkudoktor."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
result = requests.put(f"{server}/v1/config/load/provider", json="LoadAkkudoktor")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "LoadAkkudoktor" in providers
|
|
|
|
if is_system_test:
|
|
result = requests.post(f"{server}/v1/prediction/update/LoadAkkudoktor")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/series",
|
|
params={
|
|
"key": "loadforecast_power_w",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()
|
|
assert len(data["data"]) > 24
|
|
|
|
else:
|
|
pass
|
|
|
|
def test_prediction_import(self, server_setup_for_class, is_system_test):
|
|
"""Test eprediction import."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
# Reset config
|
|
with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
|
|
config = json.load(fd)
|
|
config["elecprice"]["provider"] = "ElecPriceImport"
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Assure prediction is enabled
|
|
result = requests.get(f"{server}/v1/prediction/providers?enabled=true")
|
|
assert result.status_code == HTTPStatus.OK
|
|
providers = result.json()
|
|
assert "ElecPriceImport" in providers
|
|
|
|
# Elex price payload
|
|
payload = {
|
|
"start_datetime": "2026-05-27 00:00:00",
|
|
"interval": "1 hour",
|
|
"elecprice_marketprice_wh": [0.000234, 0.000228, 0.000231],
|
|
}
|
|
|
|
result = requests.put(
|
|
f"{server}/v1/prediction/import/ElecPriceImport",
|
|
params={"force_enable": True},
|
|
json=payload,
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Import should work with any provider
|
|
config["feedintariff"]["provider"] = "FeedInTariffImport"
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# Feed In Tariff payload
|
|
payload = {
|
|
"start_datetime": "2026-05-27 00:00:00",
|
|
"interval": "1 hour",
|
|
"feed_in_tariff_wh": [0.000071, 0.000072, 0.000073],
|
|
}
|
|
|
|
result = requests.put(
|
|
f"{server}/v1/prediction/import/FeedInTariffImport",
|
|
params={"force_enable": True},
|
|
json=payload,
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
def test_measurement(self, server_setup_for_class, is_system_test):
|
|
"""Test measurement endpoints comprehensively."""
|
|
server = server_setup_for_class["server"]
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 1. Setup: Reset config with test measurement keys
|
|
# ----------------------------------------------------------------------
|
|
with FILE_TESTDATA_EOSSERVER_CONFIG_1.open("r", encoding="utf-8", newline=None) as fd:
|
|
config = json.load(fd)
|
|
|
|
config.setdefault("measurement", {})
|
|
config["measurement"]["pv_production_emr_keys"] = ["pv1_emr", "pv2_emr"]
|
|
config["measurement"]["load_emr_keys"] = ["load1_emr"]
|
|
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK, f"Config update failed: {result.text}"
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 2. GET /v1/measurement/keys
|
|
# ----------------------------------------------------------------------
|
|
result = requests.get(f"{server}/v1/measurement/keys")
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to get measurement keys: {result.text}"
|
|
|
|
keys = result.json()
|
|
assert isinstance(keys, list)
|
|
assert "pv1_emr" in keys
|
|
assert "pv2_emr" in keys
|
|
assert "load1_emr" in keys
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 3. PUT /v1/measurement/value
|
|
# ----------------------------------------------------------------------
|
|
# Float value
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": "2026-03-08T18:00:00Z",
|
|
"key": "pv1_emr",
|
|
"value": "1000.0",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to PUT float value: {result.text}"
|
|
series_response = result.json()
|
|
# PydanticDateTimeSeries has shape: {"data": {datetime_str: value}, "dtype": str, "tz": str|None}
|
|
assert "data" in series_response
|
|
assert isinstance(series_response["data"], dict)
|
|
assert len(series_response["data"]) >= 1
|
|
|
|
# String value that converts to float
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": "2026-03-08T19:00:00Z",
|
|
"key": "pv1_emr",
|
|
"value": "2000.0",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to PUT string float value: {result.text}"
|
|
|
|
# Non-numeric string value must be rejected
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": "2026-03-08T20:00:00Z",
|
|
"key": "pv1_emr",
|
|
"value": "not_a_number",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.BAD_REQUEST, (
|
|
f"Expected 400 for non-numeric string, got {result.status_code}"
|
|
)
|
|
|
|
# Non-existent key must be rejected
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": "2026-03-08T18:00:00Z",
|
|
"key": "non_existent_key",
|
|
"value": "1000.0",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.NOT_FOUND, (
|
|
f"Expected 404 for unknown key, got {result.status_code}"
|
|
)
|
|
|
|
# Missing required parameter (datetime)
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={"key": "pv1_emr", "value": "1000.0"},
|
|
)
|
|
assert result.status_code == HTTPStatus.UNPROCESSABLE_ENTITY, (
|
|
f"Expected 422 for missing datetime, got {result.status_code}"
|
|
)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 4. GET /v1/measurement/series
|
|
# ----------------------------------------------------------------------
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": "pv1_emr",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to GET series: {result.text}"
|
|
|
|
series_response = result.json()
|
|
# PydanticDateTimeSeries: {"data": {datetime_str: value, ...}, "dtype": "float64", "tz": ...}
|
|
assert "data" in series_response
|
|
assert isinstance(series_response["data"], dict)
|
|
assert "dtype" in series_response
|
|
assert len(series_response["data"]) >= 2 # at least the two values inserted above
|
|
|
|
# Non-existent key must be rejected
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": "non_existent_key",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.NOT_FOUND, (
|
|
f"Expected 404 for unknown series key, got {result.status_code}"
|
|
)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 5. PUT /v1/measurement/series
|
|
# PydanticDateTimeSeries payload: {"data": {datetime_str: value, ...}, "dtype": "float64", "tz": "UTC"}
|
|
# ----------------------------------------------------------------------
|
|
series_payload = {
|
|
"data": {
|
|
"2026-03-08T10:00:00+00:00": 500.0,
|
|
"2026-03-08T11:00:00+00:00": 600.0,
|
|
"2026-03-08T12:00:00+00:00": 700.0,
|
|
},
|
|
"dtype": "float64",
|
|
"tz": "UTC",
|
|
}
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/series",
|
|
params={"key": "pv2_emr"},
|
|
json=series_payload,
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to PUT series: {result.text}"
|
|
|
|
series_response = result.json()
|
|
assert "data" in series_response
|
|
assert isinstance(series_response["data"], dict)
|
|
assert len(series_response["data"]) >= 3
|
|
|
|
# Verify the data round-trips correctly
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": "pv2_emr",
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
fetched = result.json()
|
|
fetched_values = list(fetched["data"].values())
|
|
assert 500.0 in fetched_values
|
|
assert 600.0 in fetched_values
|
|
assert 700.0 in fetched_values
|
|
|
|
# Non-existent key must be rejected
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/series",
|
|
params={"key": "non_existent_key"},
|
|
json=series_payload,
|
|
)
|
|
assert result.status_code == HTTPStatus.NOT_FOUND, (
|
|
f"Expected 404 for unknown series PUT key, got {result.status_code}"
|
|
)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 6. PUT /v1/measurement/dataframe
|
|
# PydanticDateTimeDataFrame payload:
|
|
# {"data": {datetime_str: {"col1": val, ...}, ...}, "dtypes": {}, "tz": ..., "datetime_columns": [...]}
|
|
# ----------------------------------------------------------------------
|
|
dataframe_payload = {
|
|
"data": {
|
|
"2026-03-08T00:00:00+00:00": {"pv1_emr": 100.5, "load1_emr": 50.2},
|
|
"2026-03-08T01:00:00+00:00": {"pv1_emr": 200.3, "load1_emr": 45.1},
|
|
"2026-03-08T02:00:00+00:00": {"pv1_emr": 300.7, "load1_emr": 48.9},
|
|
},
|
|
"dtypes": {"pv1_emr": "float64", "load1_emr": "float64"},
|
|
"tz": "UTC",
|
|
"datetime_columns": [],
|
|
}
|
|
result = requests.put(f"{server}/v1/measurement/dataframe", json=dataframe_payload)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to PUT dataframe: {result.text}"
|
|
|
|
# Verify data was loaded for both columns
|
|
for key in ("pv1_emr", "load1_emr"):
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": key,
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to verify series for {key}"
|
|
series_response = result.json()
|
|
assert len(series_response["data"]) >= 3, f"Expected >=3 data points for {key}"
|
|
|
|
# Invalid dataframe structure (row columns inconsistent) must be rejected
|
|
invalid_dataframe_payload = {
|
|
"data": {
|
|
"2026-03-08T00:00:00+00:00": {"pv1_emr": 100.0},
|
|
"2026-03-08T01:00:00+00:00": {"pv1_emr": 200.0, "load1_emr": 45.0}, # extra column
|
|
},
|
|
"dtypes": {},
|
|
"tz": "UTC",
|
|
"datetime_columns": [],
|
|
}
|
|
result = requests.put(f"{server}/v1/measurement/dataframe", json=invalid_dataframe_payload)
|
|
assert result.status_code == HTTPStatus.UNPROCESSABLE_ENTITY, (
|
|
f"Expected 422 for inconsistent dataframe columns, got {result.status_code}"
|
|
)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 7. PUT /v1/measurement/data
|
|
# PydanticDateTimeData payload (RootModel):
|
|
# Dict[str, Union[str, List[Union[float, int, str, None]]]]
|
|
# Columnar format: keys are column names (or special "start_datetime"/"interval"),
|
|
# values are flat lists of equal length. Datetime index is given via start_datetime + interval.
|
|
# ----------------------------------------------------------------------
|
|
data_payload = {
|
|
"start_datetime": "2026-03-09T00:00:00+00:00",
|
|
"interval": "1 hour",
|
|
"pv1_emr": [400.2, 450.1],
|
|
"load1_emr": [60.5, 55.3],
|
|
"pv2_emr": [150.8, 175.2],
|
|
}
|
|
result = requests.put(f"{server}/v1/measurement/data", json=data_payload)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to PUT data dict: {result.text}"
|
|
|
|
# Verify all three keys received the values
|
|
for key, expected_values in (
|
|
("pv1_emr", [400.2, 450.1]),
|
|
("load1_emr", [60.5, 55.3]),
|
|
("pv2_emr", [150.8, 175.2]),
|
|
):
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": key,
|
|
"processing": "raw",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed to verify {key} after data PUT"
|
|
fetched = result.json()
|
|
fetched_values = list(fetched["data"].values())
|
|
for expected in expected_values:
|
|
assert expected in fetched_values, (
|
|
f"Expected {expected} in {key} series, got {fetched_values}"
|
|
)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 8. Edge case: invalid datetime in value PUT
|
|
# ----------------------------------------------------------------------
|
|
result = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": "not-a-datetime",
|
|
"key": "pv1_emr",
|
|
"value": "1000.0",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.BAD_REQUEST, (
|
|
f"Expected 400 for invalid datetime, got {result.status_code}"
|
|
)
|
|
|
|
def test_measurement_high_frequency_and_duplicates(self, server_setup_for_class):
|
|
"""Simulate production-like high-frequency measurement updates."""
|
|
|
|
server = server_setup_for_class["server"]
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 1. Configure measurement keys
|
|
# ----------------------------------------------------------------------
|
|
config = {
|
|
"database": {
|
|
"provider": "LMDB",
|
|
},
|
|
"measurement": {
|
|
"pv_production_emr_keys": [
|
|
"pv1_emr_kwh",
|
|
"pv2_emr_kwh",
|
|
],
|
|
"load_emr_keys": [
|
|
"load1_emr_kwh",
|
|
"load2_emr_kwh",
|
|
],
|
|
}
|
|
}
|
|
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "pv1_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "pv2_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "load1_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "load2_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 2. Simulate high-frequency writes (1 Hz, mixed keys)
|
|
# ----------------------------------------------------------------------
|
|
base_time = "2026-04-10T00:00:00"
|
|
|
|
timestamps = [
|
|
"2026-04-10T00:00:16",
|
|
"2026-04-10T00:00:46",
|
|
"2026-04-10T00:01:46",
|
|
"2026-04-10T00:02:32",
|
|
"2026-04-10T00:02:32", # duplicate
|
|
"2026-04-10T00:03:32",
|
|
"2026-04-10T00:03:32", # duplicate
|
|
]
|
|
|
|
test_cases = [
|
|
# valid
|
|
("pv1_emr_kwh", -687),
|
|
("pv2_emr_kwh", 0.76),
|
|
("load1_emr_kwh", 500),
|
|
("load2_emr_kwh", 0.0),
|
|
|
|
# invalid
|
|
("invalid-key-1", 123),
|
|
("invalid-key-2", 456),
|
|
]
|
|
|
|
results = []
|
|
|
|
for ts in timestamps:
|
|
for key, value in test_cases:
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": ts, # NOTE: naive datetime like production
|
|
"key": key,
|
|
"value": str(value),
|
|
},
|
|
)
|
|
results.append((ts, key, response.status_code))
|
|
|
|
result = requests.post(f"{server}/v1/admin/database/save")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 3. Assertions: system must behave robustly
|
|
# ----------------------------------------------------------------------
|
|
|
|
# A. Invalid keys must consistently return 404
|
|
for ts, key, status in results:
|
|
if key.startswith("invalid"):
|
|
assert status == HTTPStatus.NOT_FOUND
|
|
else:
|
|
assert status != HTTPStatus.NOT_FOUND
|
|
|
|
# B. Valid keys must NEVER produce 500
|
|
for ts, key, status in results:
|
|
if key in ("pv1_emr_kwh", "pv2_emr_kwh"):
|
|
assert status != HTTPStatus.INTERNAL_SERVER_ERROR, (
|
|
f"500 error for key={key}, ts={ts}"
|
|
)
|
|
|
|
# C. Duplicates must be handled gracefully (200 or 409 or similar, but not 500)
|
|
duplicate_failures = [
|
|
(ts, key, status)
|
|
for ts, key, status in results
|
|
if ts in ("2026-04-10T00:02:32", "2026-04-10T00:03:32")
|
|
and key == "pv1_emr_kwh"
|
|
and status == HTTPStatus.INTERNAL_SERVER_ERROR
|
|
]
|
|
|
|
assert not duplicate_failures, f"Duplicate timestamp caused 500: {duplicate_failures}"
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 4. Verify data integrity (no explosion / corruption)
|
|
# ----------------------------------------------------------------------
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": "pv1_emr_kwh",
|
|
"processing": "raw",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()["data"]
|
|
|
|
# Should not contain excessive duplicates
|
|
assert len(data) <= len(set(timestamps)), "Duplicate timestamps not handled properly"
|
|
|
|
@pytest.mark.parametrize("db_provider", ["LMDB", "SQLite", None])
|
|
def test_measurement_realtime_stream(self, db_provider, server_setup_for_class):
|
|
"""Simulate real production stream: 1 Hz updates with jitter, duplicates, and out-of-order timestamps."""
|
|
|
|
server = server_setup_for_class["server"]
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 1. Configure measurement keys and measurement database
|
|
# ----------------------------------------------------------------------
|
|
config = {
|
|
"database": {
|
|
"provider": db_provider,
|
|
},
|
|
"measurement": {
|
|
"pv_production_emr_keys": ["pv1_emr_kwh"],
|
|
"load_emr_keys": ["load1_emr_kwh"],
|
|
}
|
|
}
|
|
|
|
result = requests.put(f"{server}/v1/config", json=config)
|
|
assert result.status_code == HTTPStatus.OK, f"Failed: {result.status_code} {result}"
|
|
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "pv1_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
result = requests.delete(f"{server}/v1/measurement/range", params={"key": "load1_emr_kwh"})
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 2. Real-time simulation
|
|
# ----------------------------------------------------------------------
|
|
start = to_datetime().replace(microsecond=0)
|
|
|
|
sent_timestamps = []
|
|
errors = []
|
|
|
|
for i in range(20): # run ~20 seconds
|
|
now = to_datetime().replace(microsecond=0)
|
|
|
|
# --- main timestamp ---
|
|
ts = str(now)
|
|
|
|
# --- simulate normal write ---
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": ts,
|
|
"key": "pv1_emr_kwh",
|
|
"value": str(random.uniform(0, 1000)),
|
|
},
|
|
)
|
|
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
errors.append(("main", ts, response.text))
|
|
|
|
sent_timestamps.append(ts)
|
|
|
|
# ------------------------------------------------------------------
|
|
# Inject real-world problems
|
|
# ------------------------------------------------------------------
|
|
|
|
# 0. Other key
|
|
if random.random() < 0.5:
|
|
now_same_second = now.add(microseconds=430)
|
|
ts_same_second = str(now_same_second)
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": ts_same_second,
|
|
"key": "load1_emr_kwh",
|
|
"value": str(random.uniform(0, 1000)),
|
|
},
|
|
)
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
errors.append(("other", ts, response.text))
|
|
|
|
|
|
# 1. Duplicate timestamp (same second, slightly later)
|
|
if random.random() < 0.5:
|
|
time.sleep(0.05) # slight delay
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": ts,
|
|
"key": "pv1_emr_kwh",
|
|
"value": str(random.uniform(0, 1000)),
|
|
},
|
|
)
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
errors.append(("duplicate", ts, response.text))
|
|
|
|
# 2. Out-of-order timestamp (older data arrives late)
|
|
if len(sent_timestamps) > 2 and random.random() < 0.5:
|
|
old_ts = random.choice(sent_timestamps[:-1])
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": old_ts,
|
|
"key": "pv1_emr_kwh",
|
|
"value": str(random.uniform(0, 1000)),
|
|
},
|
|
)
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
errors.append(("out_of_order", old_ts, response.text))
|
|
|
|
# 3. Same second burst (multiple writes in same second)
|
|
if random.random() < 0.5:
|
|
for _ in range(random.randint(2, 4)):
|
|
response = requests.put(
|
|
f"{server}/v1/measurement/value",
|
|
params={
|
|
"datetime": ts,
|
|
"key": "pv1_emr_kwh",
|
|
"value": str(random.uniform(0, 1000)),
|
|
},
|
|
)
|
|
if response.status_code == HTTPStatus.INTERNAL_SERVER_ERROR:
|
|
errors.append(("burst", ts, response.text))
|
|
|
|
# 4. Assure database in memory data is saved to database
|
|
if i in (0, 4, 9, 14, 19):
|
|
response = requests.post(f"{server}/v1/admin/database/save")
|
|
assert response.status_code == HTTPStatus.OK
|
|
|
|
# small delay to simulate real system (~1 Hz)
|
|
time.sleep(1)
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 3. Assertions
|
|
# ----------------------------------------------------------------------
|
|
|
|
assert not errors, f"500 errors occurred: {errors}"
|
|
|
|
# ----------------------------------------------------------------------
|
|
# 4. Verify resulting data
|
|
# ----------------------------------------------------------------------
|
|
result = requests.get(
|
|
f"{server}/v1/measurement/series",
|
|
params={
|
|
"key": "pv1_emr_kwh",
|
|
"processing": "raw",
|
|
},
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
data = result.json()["data"]
|
|
|
|
# sanity: should not explode in size
|
|
assert len(data) == len(set(sent_timestamps))
|
|
|
|
parsed = [to_datetime(ts) for ts in data.keys()]
|
|
assert all(parsed[i] <= parsed[i+1] for i in range(len(parsed)-1)), \
|
|
"Timestamps are not sorted"
|
|
|
|
unique_keys = set(data.keys())
|
|
assert len(unique_keys) == len(data), \
|
|
"Duplicate timestamps detected in API output"
|
|
|
|
def test_admin_cache(self, server_setup_for_class, is_system_test):
|
|
"""Test whether cache is reconstructed from cached files."""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
|
|
result = requests.get(f"{server}/v1/admin/cache")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache = result.json()
|
|
|
|
if is_system_test:
|
|
# There should be some cache data
|
|
assert cache != {}
|
|
|
|
# Save cache
|
|
result = requests.post(f"{server}/v1/admin/cache/save")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache_saved = result.json()
|
|
assert cache_saved == cache
|
|
|
|
# Clear expired cache - should clear nothing as all cache files expire in the future
|
|
result = requests.post(f"{server}/v1/admin/cache/clear-expired")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache_cleared = result.json()
|
|
assert cache_cleared == cache
|
|
|
|
# Force clear cache
|
|
result = requests.post(f"{server}/v1/admin/cache/clear")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache_cleared = result.json()
|
|
assert cache_cleared == {}
|
|
|
|
# Try to load already deleted cache entries
|
|
result = requests.post(f"{server}/v1/admin/cache/load")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache_loaded = result.json()
|
|
assert cache_loaded == {}
|
|
|
|
# Cache should still be empty
|
|
result = requests.get(f"{server}/v1/admin/cache")
|
|
assert result.status_code == HTTPStatus.OK
|
|
cache = result.json()
|
|
assert cache == {}
|
|
|
|
def test_deprecated_strompreis(self, server_setup_for_class, is_system_test):
|
|
"""Test deprecated /strompreis endpoint.
|
|
|
|
Deprecated /strompreis aggregates 15-minute spot prices to hourly means.
|
|
"""
|
|
server = server_setup_for_class["server"]
|
|
eos_dir = server_setup_for_class["eos_dir"]
|
|
start_datetime = to_datetime().start_of("day")
|
|
end_datetime = start_datetime.add(days=2)
|
|
|
|
# Reset electricity market price
|
|
result = requests.delete(
|
|
f"{server}/v1/prediction/range",
|
|
params = {
|
|
"key": "elecprice_marketprice_wh",
|
|
}
|
|
)
|
|
assert result.status_code in (HTTPStatus.OK, HTTPStatus.NOT_FOUND)
|
|
|
|
if not is_system_test:
|
|
return
|
|
|
|
# Call /strompreis
|
|
result = requests.get(f"{server}/strompreis")
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
strompreis_data = result.json()
|
|
assert len(strompreis_data) == 48
|
|
|
|
# Get the same data by v1 interface
|
|
result = requests.get(
|
|
f"{server}/v1/prediction/list",
|
|
params = {
|
|
"key": "elecprice_marketprice_wh",
|
|
"start_datetime": to_datetime(start_datetime, as_string=True),
|
|
"end_datetime": to_datetime(end_datetime, as_string=True),
|
|
"interval" : "1 hour",
|
|
"fill_method" : "ffill",
|
|
"resample_method" : "interval_mean",
|
|
}
|
|
)
|
|
assert result.status_code == HTTPStatus.OK
|
|
|
|
v1_data = result.json()
|
|
assert strompreis_data == pytest.approx(v1_data)
|