From 6de02fc1b979e8d8061e0a3b53660e671932da46 Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Wed, 2 Sep 2020 14:53:07 +0200 Subject: [PATCH] Fix some more usages of asynctest (#39570) --- tests/components/command_line/test_cover.py | 3 ++- tests/components/deconz/test_config_flow.py | 3 ++- tests/components/group/test_light.py | 4 +--- tests/components/homeassistant/triggers/test_time_pattern.py | 2 +- tests/components/min_max/test_sensor.py | 3 +-- tests/components/universal/test_media_player.py | 2 +- 6 files changed, 8 insertions(+), 9 deletions(-) diff --git a/tests/components/command_line/test_cover.py b/tests/components/command_line/test_cover.py index ceb3711f62b..508b1e1fb41 100644 --- a/tests/components/command_line/test_cover.py +++ b/tests/components/command_line/test_cover.py @@ -4,7 +4,6 @@ from os import path import tempfile from unittest import mock -from asynctest.mock import patch import pytest from homeassistant import config as hass_config @@ -19,6 +18,8 @@ from homeassistant.const import ( ) from homeassistant.setup import async_setup_component +from tests.async_mock import patch + @pytest.fixture def rs(hass): diff --git a/tests/components/deconz/test_config_flow.py b/tests/components/deconz/test_config_flow.py index b51869b01ab..43536a44bbe 100644 --- a/tests/components/deconz/test_config_flow.py +++ b/tests/components/deconz/test_config_flow.py @@ -1,7 +1,6 @@ """Tests for deCONZ config flow.""" import asyncio -from asynctest.mock import patch import pydeconz from homeassistant import data_entry_flow @@ -22,6 +21,8 @@ from homeassistant.const import CONF_API_KEY, CONF_HOST, CONF_PORT from .test_gateway import API_KEY, BRIDGEID, setup_deconz_integration +from tests.async_mock import patch + async def test_flow_discovered_bridges(hass, aioclient_mock): """Test that config flow works for discovered bridges.""" diff --git a/tests/components/group/test_light.py b/tests/components/group/test_light.py index dae975ead03..8855ca97626 100644 --- a/tests/components/group/test_light.py +++ b/tests/components/group/test_light.py @@ -1,8 +1,6 @@ """The tests for the Group Light platform.""" from os import path -from asynctest.mock import patch - from homeassistant import config as hass_config from homeassistant.components.group import DOMAIN, SERVICE_RELOAD import homeassistant.components.group.light as group @@ -34,7 +32,7 @@ from homeassistant.const import ( from homeassistant.setup import async_setup_component import tests.async_mock -from tests.async_mock import MagicMock +from tests.async_mock import MagicMock, patch async def test_default_state(hass): diff --git a/tests/components/homeassistant/triggers/test_time_pattern.py b/tests/components/homeassistant/triggers/test_time_pattern.py index f2a2cd29d79..0ef071aadb6 100644 --- a/tests/components/homeassistant/triggers/test_time_pattern.py +++ b/tests/components/homeassistant/triggers/test_time_pattern.py @@ -1,5 +1,4 @@ """The tests for the time_pattern automation.""" -from asynctest.mock import patch import pytest import voluptuous as vol @@ -8,6 +7,7 @@ import homeassistant.components.homeassistant.triggers.time_pattern as time_patt from homeassistant.setup import async_setup_component import homeassistant.util.dt as dt_util +from tests.async_mock import patch from tests.common import async_fire_time_changed, async_mock_service, mock_component from tests.components.automation import common diff --git a/tests/components/min_max/test_sensor.py b/tests/components/min_max/test_sensor.py index 8f532a6ec07..52de3dfa1ab 100644 --- a/tests/components/min_max/test_sensor.py +++ b/tests/components/min_max/test_sensor.py @@ -3,8 +3,6 @@ from os import path import statistics import unittest -from asynctest.mock import patch - from homeassistant import config as hass_config from homeassistant.components.min_max import DOMAIN from homeassistant.const import ( @@ -18,6 +16,7 @@ from homeassistant.const import ( ) from homeassistant.setup import async_setup_component, setup_component +from tests.async_mock import patch from tests.common import get_test_home_assistant diff --git a/tests/components/universal/test_media_player.py b/tests/components/universal/test_media_player.py index 5333f72b2a9..38949c32ebb 100644 --- a/tests/components/universal/test_media_player.py +++ b/tests/components/universal/test_media_player.py @@ -4,7 +4,6 @@ from copy import copy from os import path import unittest -from asynctest.mock import patch from voluptuous.error import MultipleInvalid from homeassistant import config as hass_config @@ -23,6 +22,7 @@ from homeassistant.const import ( ) from homeassistant.setup import async_setup_component +from tests.async_mock import patch from tests.common import get_test_home_assistant, mock_service