Use snapshot assertion for EasyEnergy diagnostics test (#98909)

pull/97467/head
Joost Lekkerkerker 2023-08-24 09:19:36 +02:00 committed by GitHub
parent 8d576c900d
commit 602a80c35c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 74 additions and 62 deletions

View File

@ -0,0 +1,63 @@
# serializer version: 1
# name: test_diagnostics
dict({
'energy_return': dict({
'average_price': 0.14599,
'current_hour_price': 0.18629,
'highest_price_time': '2023-01-19T16:00:00+00:00',
'lowest_price_time': '2023-01-19T02:00:00+00:00',
'max_price': 0.20394,
'min_price': 0.10172,
'next_hour_price': 0.20394,
'percentage_of_max': 91.35,
}),
'energy_usage': dict({
'average_price': 0.17665,
'current_hour_price': 0.22541,
'highest_price_time': '2023-01-19T16:00:00+00:00',
'lowest_price_time': '2023-01-19T02:00:00+00:00',
'max_price': 0.24677,
'min_price': 0.12308,
'next_hour_price': 0.24677,
'percentage_of_max': 91.34,
}),
'entry': dict({
'title': 'energy',
}),
'gas': dict({
'current_hour_price': 0.7253,
'next_hour_price': 0.7253,
}),
})
# ---
# name: test_diagnostics_no_gas_today
dict({
'energy_return': dict({
'average_price': 0.14599,
'current_hour_price': 0.18629,
'highest_price_time': '2023-01-19T16:00:00+00:00',
'lowest_price_time': '2023-01-19T02:00:00+00:00',
'max_price': 0.20394,
'min_price': 0.10172,
'next_hour_price': 0.20394,
'percentage_of_max': 91.35,
}),
'energy_usage': dict({
'average_price': 0.17665,
'current_hour_price': 0.22541,
'highest_price_time': '2023-01-19T16:00:00+00:00',
'lowest_price_time': '2023-01-19T02:00:00+00:00',
'max_price': 0.24677,
'min_price': 0.12308,
'next_hour_price': 0.24677,
'percentage_of_max': 91.34,
}),
'entry': dict({
'title': 'energy',
}),
'gas': dict({
'current_hour_price': None,
'next_hour_price': None,
}),
})
# ---

View File

@ -3,6 +3,7 @@ from unittest.mock import MagicMock
from easyenergy import EasyEnergyNoDataError
import pytest
from syrupy import SnapshotAssertion
from homeassistant.components.homeassistant import SERVICE_UPDATE_ENTITY
from homeassistant.const import ATTR_ENTITY_ID
@ -19,39 +20,13 @@ async def test_diagnostics(
hass: HomeAssistant,
hass_client: ClientSessionGenerator,
init_integration: MockConfigEntry,
snapshot: SnapshotAssertion,
) -> None:
"""Test diagnostics."""
assert await get_diagnostics_for_config_entry(
hass, hass_client, init_integration
) == {
"entry": {
"title": "energy",
},
"energy_usage": {
"current_hour_price": 0.22541,
"next_hour_price": 0.24677,
"average_price": 0.17665,
"max_price": 0.24677,
"min_price": 0.12308,
"highest_price_time": "2023-01-19T16:00:00+00:00",
"lowest_price_time": "2023-01-19T02:00:00+00:00",
"percentage_of_max": 91.34,
},
"energy_return": {
"current_hour_price": 0.18629,
"next_hour_price": 0.20394,
"average_price": 0.14599,
"max_price": 0.20394,
"min_price": 0.10172,
"highest_price_time": "2023-01-19T16:00:00+00:00",
"lowest_price_time": "2023-01-19T02:00:00+00:00",
"percentage_of_max": 91.35,
},
"gas": {
"current_hour_price": 0.7253,
"next_hour_price": 0.7253,
},
}
assert (
await get_diagnostics_for_config_entry(hass, hass_client, init_integration)
== snapshot
)
@pytest.mark.freeze_time("2023-01-19 15:00:00")
@ -60,6 +35,7 @@ async def test_diagnostics_no_gas_today(
hass_client: ClientSessionGenerator,
mock_easyenergy: MagicMock,
init_integration: MockConfigEntry,
snapshot: SnapshotAssertion,
) -> None:
"""Test diagnostics, no gas sensors available."""
await async_setup_component(hass, "homeassistant", {})
@ -73,34 +49,7 @@ async def test_diagnostics_no_gas_today(
)
await hass.async_block_till_done()
assert await get_diagnostics_for_config_entry(
hass, hass_client, init_integration
) == {
"entry": {
"title": "energy",
},
"energy_usage": {
"current_hour_price": 0.22541,
"next_hour_price": 0.24677,
"average_price": 0.17665,
"max_price": 0.24677,
"min_price": 0.12308,
"highest_price_time": "2023-01-19T16:00:00+00:00",
"lowest_price_time": "2023-01-19T02:00:00+00:00",
"percentage_of_max": 91.34,
},
"energy_return": {
"current_hour_price": 0.18629,
"next_hour_price": 0.20394,
"average_price": 0.14599,
"max_price": 0.20394,
"min_price": 0.10172,
"highest_price_time": "2023-01-19T16:00:00+00:00",
"lowest_price_time": "2023-01-19T02:00:00+00:00",
"percentage_of_max": 91.35,
},
"gas": {
"current_hour_price": None,
"next_hour_price": None,
},
}
assert (
await get_diagnostics_for_config_entry(hass, hass_client, init_integration)
== snapshot
)