Fix fritzbox tests (#58227)

This commit is contained in:
Lars 2021-10-22 17:29:36 +02:00 committed by GitHub
parent 51a10f88de
commit 416d87c01c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -152,14 +152,14 @@ async def test_update(hass: HomeAssistant, fritz: Mock):
assert await setup_config_entry(
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
)
assert device.update.call_count == 1
assert fritz().update_devices.call_count == 1
assert fritz().login.call_count == 1
next_update = dt_util.utcnow() + timedelta(seconds=200)
async_fire_time_changed(hass, next_update)
await hass.async_block_till_done()
assert device.update.call_count == 2
assert fritz().update_devices.call_count == 2
assert fritz().login.call_count == 1
@ -170,16 +170,16 @@ async def test_update_error(hass: HomeAssistant, fritz: Mock):
device.get_colors.return_value = {
"Red": [("100", "70", "10"), ("100", "50", "10"), ("100", "30", "10")]
}
device.update.side_effect = HTTPError("Boom")
fritz().update_devices.side_effect = HTTPError("Boom")
assert not await setup_config_entry(
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz
)
assert device.update.call_count == 1
assert fritz().update_devices.call_count == 1
assert fritz().login.call_count == 1
next_update = dt_util.utcnow() + timedelta(seconds=200)
async_fire_time_changed(hass, next_update)
await hass.async_block_till_done()
assert device.update.call_count == 2
assert fritz().update_devices.call_count == 2
assert fritz().login.call_count == 2