Fix modem_callerid test warning (#57760)

This commit is contained in:
Martin Hjelmare 2021-10-15 18:36:26 +02:00 committed by GitHub
parent 0f2b5ea28e
commit 5aba8a7c81
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 7 deletions

View File

@ -11,15 +11,16 @@ CONF_DATA = {CONF_DEVICE: DEFAULT_PORT}
IMPORT_DATA = {"sensor": {"platform": "modem_callerid"}}
def _patch_init_modem(mocked_modem):
def _patch_init_modem():
return patch(
"homeassistant.components.modem_callerid.PhoneModem",
return_value=mocked_modem,
autospec=True,
)
def _patch_config_flow_modem(mocked_modem):
return patch(
"homeassistant.components.modem_callerid.config_flow.PhoneModem",
autospec=True,
return_value=mocked_modem,
)

View File

@ -1,5 +1,5 @@
"""Test Modem Caller ID integration."""
from unittest.mock import AsyncMock, patch
from unittest.mock import patch
from phone_modem import exceptions
@ -19,8 +19,7 @@ async def test_setup_config(hass: HomeAssistant):
data=CONF_DATA,
)
entry.add_to_hass(hass)
mocked_modem = AsyncMock()
with _patch_init_modem(mocked_modem):
with _patch_init_modem():
await hass.config_entries.async_setup(entry.entry_id)
assert entry.state == ConfigEntryState.LOADED
@ -50,8 +49,7 @@ async def test_unload_config_entry(hass: HomeAssistant):
data=CONF_DATA,
)
entry.add_to_hass(hass)
mocked_modem = AsyncMock()
with _patch_init_modem(mocked_modem):
with _patch_init_modem():
await hass.config_entries.async_setup(entry.entry_id)
assert len(hass.config_entries.async_entries(DOMAIN)) == 1
assert entry.state is ConfigEntryState.LOADED