jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-18 18:57:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use freezegun in pvpc_hourly_pricing tests (#99040)

Browse Source
This commit is contained in:
Erik Montnemery 2023-08-25 16:06:14 +02:00 committed by GitHub
parent c827af5826
commit 452caee41a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 73 additions and 72 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

145
tests/components/pvpc_hourly_pricing/test_config_flow.py
View File

@ -1,7 +1,7 @@
"""Tests for the pvpc_hourly_pricing config_flow.""" """Tests for the pvpc_hourly_pricing config_flow."""
from datetime import datetime, timedelta from datetime import datetime, timedelta
from freezegun import freeze_time from freezegun.api import FrozenDateTimeFactory
from homeassistant import config_entries, data_entry_flow from homeassistant import config_entries, data_entry_flow
from homeassistant.components.pvpc_hourly_pricing import ( from homeassistant.components.pvpc_hourly_pricing import (
@ -25,7 +25,9 @@ _MOCK_TIME_VALID_RESPONSES = datetime(2023, 1, 6, 12, 0, tzinfo=dt_util.UTC)
async def test_config_flow( async def test_config_flow(
hass: HomeAssistant, pvpc_aioclient_mock: AiohttpClientMocker hass: HomeAssistant,
freezer: FrozenDateTimeFactory,
pvpc_aioclient_mock: AiohttpClientMocker,
) -> None: ) -> None:
"""Test config flow for pvpc_hourly_pricing. """Test config flow for pvpc_hourly_pricing.
@ -35,6 +37,7 @@ async def test_config_flow(
- Check removal and add again to check state restoration - Check removal and add again to check state restoration
- Configure options to change power and tariff to "2.0TD" - Configure options to change power and tariff to "2.0TD"
""" """
freezer.move_to(_MOCK_TIME_VALID_RESPONSES)
hass.config.set_time_zone("Europe/Madrid") hass.config.set_time_zone("Europe/Madrid")
tst_config = { tst_config = {
CONF_NAME: "test", CONF_NAME: "test",
@ -43,84 +46,82 @@ async def test_config_flow(
ATTR_POWER_P3: 5.75, ATTR_POWER_P3: 5.75,
} }
with freeze_time(_MOCK_TIME_VALID_RESPONSES) as mock_time: result = await hass.config_entries.flow.async_init(
result = await hass.config_entries.flow.async_init( DOMAIN, context={"source": config_entries.SOURCE_USER}
DOMAIN, context={"source": config_entries.SOURCE_USER} )
) assert result["type"] == data_entry_flow.FlowResultType.FORM
assert result["type"] == data_entry_flow.FlowResultType.FORM
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], tst_config result["flow_id"], tst_config
) )
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
await hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("sensor.test") state = hass.states.get("sensor.test")
check_valid_state(state, tariff=TARIFFS[1]) check_valid_state(state, tariff=TARIFFS[1])
assert pvpc_aioclient_mock.call_count == 1 assert pvpc_aioclient_mock.call_count == 1
# Check abort when configuring another with same tariff # Check abort when configuring another with same tariff
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": config_entries.SOURCE_USER} DOMAIN, context={"source": config_entries.SOURCE_USER}
) )
assert result["type"] == data_entry_flow.FlowResultType.FORM assert result["type"] == data_entry_flow.FlowResultType.FORM
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], tst_config result["flow_id"], tst_config
) )
assert result["type"] == data_entry_flow.FlowResultType.ABORT assert result["type"] == data_entry_flow.FlowResultType.ABORT
assert pvpc_aioclient_mock.call_count == 1 assert pvpc_aioclient_mock.call_count == 1
# Check removal # Check removal
registry = er.async_get(hass) registry = er.async_get(hass)
registry_entity = registry.async_get("sensor.test") registry_entity = registry.async_get("sensor.test")
assert await hass.config_entries.async_remove(registry_entity.config_entry_id) assert await hass.config_entries.async_remove(registry_entity.config_entry_id)
# and add it again with UI # and add it again with UI
result = await hass.config_entries.flow.async_init( result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": config_entries.SOURCE_USER} DOMAIN, context={"source": config_entries.SOURCE_USER}
) )
assert result["type"] == data_entry_flow.FlowResultType.FORM assert result["type"] == data_entry_flow.FlowResultType.FORM
result = await hass.config_entries.flow.async_configure( result = await hass.config_entries.flow.async_configure(
result["flow_id"], tst_config result["flow_id"], tst_config
) )
assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY assert result["type"] == data_entry_flow.FlowResultType.CREATE_ENTRY
await hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("sensor.test") state = hass.states.get("sensor.test")
check_valid_state(state, tariff=TARIFFS[1]) check_valid_state(state, tariff=TARIFFS[1])
assert pvpc_aioclient_mock.call_count == 2 assert pvpc_aioclient_mock.call_count == 2
assert state.attributes["period"] == "P3" assert state.attributes["period"] == "P3"
assert state.attributes["next_period"] == "P2" assert state.attributes["next_period"] == "P2"
assert state.attributes["available_power"] == 5750 assert state.attributes["available_power"] == 5750
# check options flow # check options flow
current_entries = hass.config_entries.async_entries(DOMAIN) current_entries = hass.config_entries.async_entries(DOMAIN)
assert len(current_entries) == 1 assert len(current_entries) == 1
config_entry = current_entries[0] config_entry = current_entries[0]
result = await hass.config_entries.options.async_init(config_entry.entry_id) result = await hass.config_entries.options.async_init(config_entry.entry_id)
assert result["type"] == data_entry_flow.FlowResultType.FORM assert result["type"] == data_entry_flow.FlowResultType.FORM
assert result["step_id"] == "init" assert result["step_id"] == "init"
result = await hass.config_entries.options.async_configure( result = await hass.config_entries.options.async_configure(
result["flow_id"], result["flow_id"],
user_input={ATTR_POWER: 3.0, ATTR_POWER_P3: 4.6}, user_input={ATTR_POWER: 3.0, ATTR_POWER_P3: 4.6},
) )
await hass.async_block_till_done() await hass.async_block_till_done()
state = hass.states.get("sensor.test") state = hass.states.get("sensor.test")
check_valid_state(state, tariff=TARIFFS[1]) check_valid_state(state, tariff=TARIFFS[1])
assert pvpc_aioclient_mock.call_count == 3 assert pvpc_aioclient_mock.call_count == 3
assert state.attributes["period"] == "P3" assert state.attributes["period"] == "P3"
assert state.attributes["next_period"] == "P2" assert state.attributes["next_period"] == "P2"
assert state.attributes["available_power"] == 4600 assert state.attributes["available_power"] == 4600
# check update failed # check update failed
ts_future = _MOCK_TIME_VALID_RESPONSES + timedelta(days=1) freezer.tick(timedelta(days=1))
mock_time.move_to(ts_future) async_fire_time_changed(hass)
async_fire_time_changed(hass, ts_future) await hass.async_block_till_done()
await hass.async_block_till_done() state = hass.states.get("sensor.test")
state = hass.states.get("sensor.test") check_valid_state(state, tariff=TARIFFS[0], value="unavailable")
check_valid_state(state, tariff=TARIFFS[0], value="unavailable") assert "period" not in state.attributes
assert "period" not in state.attributes assert pvpc_aioclient_mock.call_count == 4
assert pvpc_aioclient_mock.call_count == 4