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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 1 deletions

View file

@ -5,6 +5,12 @@ from pytest import fixture
from tests.async_mock import MagicMock, patch
@fixture(autouse=True)
def mock_ffmpeg(hass):
"""Mock ffmpeg is loaded."""
hass.config.components.add("ffmpeg")
@fixture
def canary(hass):
"""Mock the CanaryApi for easier testing."""

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)