diff --git a/homeassistant/components/pi_hole/__init__.py b/homeassistant/components/pi_hole/__init__.py index 95351083b5a..8ee21af7858 100644 --- a/homeassistant/components/pi_hole/__init__.py +++ b/homeassistant/components/pi_hole/__init__.py @@ -1,31 +1,31 @@ """The pi_hole component.""" import logging -import voluptuous as vol from hole import Hole from hole.exceptions import HoleError +import voluptuous as vol +from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN from homeassistant.const import ( + CONF_API_KEY, CONF_HOST, CONF_NAME, - CONF_API_KEY, CONF_SSL, CONF_VERIFY_SSL, ) -from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN from homeassistant.helpers import config_validation as cv from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.discovery import async_load_platform from homeassistant.util import Throttle from .const import ( - DOMAIN, CONF_LOCATION, DEFAULT_HOST, DEFAULT_LOCATION, DEFAULT_NAME, DEFAULT_SSL, DEFAULT_VERIFY_SSL, + DOMAIN, MIN_TIME_BETWEEN_UPDATES, SERVICE_DISABLE, SERVICE_DISABLE_ATTR_DURATION, diff --git a/homeassistant/components/pi_hole/sensor.py b/homeassistant/components/pi_hole/sensor.py index 4e80e9767a6..f76e756aec4 100644 --- a/homeassistant/components/pi_hole/sensor.py +++ b/homeassistant/components/pi_hole/sensor.py @@ -4,10 +4,10 @@ import logging from homeassistant.helpers.entity import Entity from .const import ( - DOMAIN as PIHOLE_DOMAIN, ATTR_BLOCKED_DOMAINS, - SENSOR_LIST, + DOMAIN as PIHOLE_DOMAIN, SENSOR_DICT, + SENSOR_LIST, ) LOGGER = logging.getLogger(__name__) diff --git a/tests/components/pi_hole/test_init.py b/tests/components/pi_hole/test_init.py index f30422bfea9..b2f4b3f28af 100644 --- a/tests/components/pi_hole/test_init.py +++ b/tests/components/pi_hole/test_init.py @@ -1,11 +1,13 @@ """Test pi_hole component.""" +from unittest.mock import patch + from asynctest import CoroutineMock from hole import Hole from homeassistant.components import pi_hole + from tests.common import async_setup_component -from unittest.mock import patch def mock_pihole_data_call(Hole):