Fix vizio async mock fixtures on Python 3.8.0 and .1 (#39926)

This commit is contained in:
Ville Skyttä 2020-09-14 09:35:30 +03:00 committed by GitHub
parent bda66d1929
commit aad6a24f28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -22,7 +22,7 @@ from .const import (
MockStartPairingResponse, MockStartPairingResponse,
) )
from tests.async_mock import patch from tests.async_mock import AsyncMock, patch
class MockInput: class MockInput:
@ -53,7 +53,7 @@ def vizio_get_unique_id_fixture():
"""Mock get vizio unique ID.""" """Mock get vizio unique ID."""
with patch( with patch(
"homeassistant.components.vizio.config_flow.VizioAsync.get_unique_id", "homeassistant.components.vizio.config_flow.VizioAsync.get_unique_id",
return_value=UNIQUE_ID, AsyncMock(return_value=UNIQUE_ID),
): ):
yield yield
@ -83,7 +83,7 @@ def vizio_connect_fixture():
"""Mock valid vizio device and entry setup.""" """Mock valid vizio device and entry setup."""
with patch( with patch(
"homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config", "homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config",
return_value=True, AsyncMock(return_value=True),
): ):
yield yield
@ -156,7 +156,7 @@ def vizio_cant_connect_fixture():
"""Mock vizio device can't connect with valid auth.""" """Mock vizio device can't connect with valid auth."""
with patch( with patch(
"homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config", "homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config",
return_value=False, AsyncMock(return_value=False),
): ):
yield yield