From aad6a24f28379abf0d431e33b612120d352a72ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ville=20Skytt=C3=A4?= Date: Mon, 14 Sep 2020 09:35:30 +0300 Subject: [PATCH] Fix vizio async mock fixtures on Python 3.8.0 and .1 (#39926) --- tests/components/vizio/conftest.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/components/vizio/conftest.py b/tests/components/vizio/conftest.py index 08e5da5c9e5..c8a9083bb1a 100644 --- a/tests/components/vizio/conftest.py +++ b/tests/components/vizio/conftest.py @@ -22,7 +22,7 @@ from .const import ( MockStartPairingResponse, ) -from tests.async_mock import patch +from tests.async_mock import AsyncMock, patch class MockInput: @@ -53,7 +53,7 @@ def vizio_get_unique_id_fixture(): """Mock get vizio unique ID.""" with patch( "homeassistant.components.vizio.config_flow.VizioAsync.get_unique_id", - return_value=UNIQUE_ID, + AsyncMock(return_value=UNIQUE_ID), ): yield @@ -83,7 +83,7 @@ def vizio_connect_fixture(): """Mock valid vizio device and entry setup.""" with patch( "homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config", - return_value=True, + AsyncMock(return_value=True), ): yield @@ -156,7 +156,7 @@ def vizio_cant_connect_fixture(): """Mock vizio device can't connect with valid auth.""" with patch( "homeassistant.components.vizio.config_flow.VizioAsync.validate_ha_config", - return_value=False, + AsyncMock(return_value=False), ): yield