diff --git a/tests/components/websocket_api/conftest.py b/tests/components/websocket_api/conftest.py index 382de3142e8..65b9232821f 100644 --- a/tests/components/websocket_api/conftest.py +++ b/tests/components/websocket_api/conftest.py @@ -1,9 +1,9 @@ """Fixtures for websocket tests.""" import pytest -from homeassistant.setup import async_setup_component -from homeassistant.components.websocket_api.http import URL from homeassistant.components.websocket_api.auth import TYPE_AUTH_REQUIRED +from homeassistant.components.websocket_api.http import URL +from homeassistant.setup import async_setup_component @pytest.fixture diff --git a/tests/components/websocket_api/test_auth.py b/tests/components/websocket_api/test_auth.py index 00387506020..ccc033ccc72 100644 --- a/tests/components/websocket_api/test_auth.py +++ b/tests/components/websocket_api/test_auth.py @@ -1,18 +1,17 @@ """Test auth of websocket API.""" from unittest.mock import patch -from homeassistant.components.websocket_api.const import ( - URL, - SIGNAL_WEBSOCKET_CONNECTED, - SIGNAL_WEBSOCKET_DISCONNECTED, -) from homeassistant.components.websocket_api.auth import ( TYPE_AUTH, TYPE_AUTH_INVALID, TYPE_AUTH_OK, TYPE_AUTH_REQUIRED, ) - +from homeassistant.components.websocket_api.const import ( + SIGNAL_WEBSOCKET_CONNECTED, + SIGNAL_WEBSOCKET_DISCONNECTED, + URL, +) from homeassistant.setup import async_setup_component from tests.common import mock_coro diff --git a/tests/components/websocket_api/test_commands.py b/tests/components/websocket_api/test_commands.py index 1de5b8bb2c1..58d904c8f4b 100644 --- a/tests/components/websocket_api/test_commands.py +++ b/tests/components/websocket_api/test_commands.py @@ -1,14 +1,14 @@ """Tests for WebSocket API commands.""" from async_timeout import timeout -from homeassistant.core import callback -from homeassistant.components.websocket_api.const import URL +from homeassistant.components.websocket_api import const from homeassistant.components.websocket_api.auth import ( TYPE_AUTH, TYPE_AUTH_OK, TYPE_AUTH_REQUIRED, ) -from homeassistant.components.websocket_api import const +from homeassistant.components.websocket_api.const import URL +from homeassistant.core import callback from homeassistant.exceptions import HomeAssistantError from homeassistant.setup import async_setup_component diff --git a/tests/components/websocket_api/test_init.py b/tests/components/websocket_api/test_init.py index ed3319ed4f4..7cda3200e7b 100644 --- a/tests/components/websocket_api/test_init.py +++ b/tests/components/websocket_api/test_init.py @@ -1,6 +1,6 @@ """Tests for the Home Assistant Websocket API.""" import asyncio -from unittest.mock import patch, Mock +from unittest.mock import Mock, patch from aiohttp import WSMsgType import pytest diff --git a/tests/components/websocket_api/test_sensor.py b/tests/components/websocket_api/test_sensor.py index 84b73060698..2c711737851 100644 --- a/tests/components/websocket_api/test_sensor.py +++ b/tests/components/websocket_api/test_sensor.py @@ -2,9 +2,10 @@ from homeassistant.bootstrap import async_setup_component -from tests.common import assert_setup_component from .test_auth import test_auth_active_with_token +from tests.common import assert_setup_component + async def test_websocket_api( hass, no_auth_websocket_client, hass_access_token, legacy_auth