Fix mfi test (#43067)

This commit is contained in:
Paulus Schoutsen
2020-11-11 00:03:16 +01:00
committed by GitHub
parent 7745408440
commit ec1b9c2099
2 changed files with 7 additions and 1 deletions

View File

@@ -28,7 +28,7 @@ async def test_setup_adds_proper_devices(hass):
with mock.patch(
"homeassistant.components.mfi.switch.MFiClient"
) as mock_client, mock.patch(
"homeassistant.components.mfi.switch.MfiSwitch"
"homeassistant.components.mfi.switch.MfiSwitch", side_effect=mfi.MfiSwitch
) as mock_switch:
ports = {
i: mock.MagicMock(model=model) for i, model in enumerate(mfi.SWITCH_MODELS)