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( assert await setup_config_entry(
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz 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 assert fritz().login.call_count == 1
next_update = dt_util.utcnow() + timedelta(seconds=200) next_update = dt_util.utcnow() + timedelta(seconds=200)
async_fire_time_changed(hass, next_update) async_fire_time_changed(hass, next_update)
await hass.async_block_till_done() 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 assert fritz().login.call_count == 1
@ -170,16 +170,16 @@ async def test_update_error(hass: HomeAssistant, fritz: Mock):
device.get_colors.return_value = { device.get_colors.return_value = {
"Red": [("100", "70", "10"), ("100", "50", "10"), ("100", "30", "10")] "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( assert not await setup_config_entry(
hass, MOCK_CONFIG[FB_DOMAIN][CONF_DEVICES][0], ENTITY_ID, device, fritz 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 assert fritz().login.call_count == 1
next_update = dt_util.utcnow() + timedelta(seconds=200) next_update = dt_util.utcnow() + timedelta(seconds=200)
async_fire_time_changed(hass, next_update) async_fire_time_changed(hass, next_update)
await hass.async_block_till_done() 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 assert fritz().login.call_count == 2