8ba4ee1012
* Add Airly integration * Update .coveragerc file * Remove AVAILABLE_CONDITIONS and fix device_class * Don't create client on every update * Rename client to session * Rename state_attributes to device_state_attributes * Remove log latitude and longitude * Fix try...except * Change latitude and longitude to HA defaults * _show_config_form doesn't need coroutine * Simplify config_flow errors handlig * Preetier * Remove unnecessary condition * Change sensor platform to air_quality * Remove PM1 * Make unique_id more unique * Remove , * Add tests for config_flow * Move conf to CONFIG * Remove domain from unique_id * Change the way update of attrs * Language and attrs * Fix attrs * Add aiohttp error handling * Throttle as decorator * Suggested change * Suggested change * Invert condition * Cleaning * Add tests * Polish no sesnor error handling * Better strings * Fix test_invalid_api_key * Fix documentation url * Remove unnecessary test * Remove language option * Fix test_invalid_api_key once again * Sort imports * Remove splits in strings
93 lines
2.9 KiB
Python
93 lines
2.9 KiB
Python
"""Define tests for the Airly config flow."""
|
|
import json
|
|
|
|
from airly.exceptions import AirlyError
|
|
from asynctest import patch
|
|
|
|
from homeassistant import data_entry_flow
|
|
from homeassistant.const import CONF_API_KEY, CONF_LATITUDE, CONF_LONGITUDE, CONF_NAME
|
|
from homeassistant.components.airly import config_flow
|
|
from homeassistant.components.airly.const import DOMAIN
|
|
|
|
from tests.common import load_fixture, MockConfigEntry
|
|
|
|
CONFIG = {
|
|
CONF_NAME: "abcd",
|
|
CONF_API_KEY: "foo",
|
|
CONF_LATITUDE: 123,
|
|
CONF_LONGITUDE: 456,
|
|
}
|
|
|
|
|
|
async def test_show_form(hass):
|
|
"""Test that the form is served with no input."""
|
|
flow = config_flow.AirlyFlowHandler()
|
|
flow.hass = hass
|
|
|
|
result = await flow.async_step_user(user_input=None)
|
|
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
|
assert result["step_id"] == "user"
|
|
|
|
|
|
async def test_invalid_api_key(hass):
|
|
"""Test that errors are shown when API key is invalid."""
|
|
with patch(
|
|
"airly._private._RequestsHandler.get",
|
|
side_effect=AirlyError(403, {"message": "Invalid authentication credentials"}),
|
|
):
|
|
flow = config_flow.AirlyFlowHandler()
|
|
flow.hass = hass
|
|
|
|
result = await flow.async_step_user(user_input=CONFIG)
|
|
|
|
assert result["errors"] == {"base": "auth"}
|
|
|
|
|
|
async def test_invalid_location(hass):
|
|
"""Test that errors are shown when location is invalid."""
|
|
with patch(
|
|
"airly._private._RequestsHandler.get",
|
|
return_value=json.loads(load_fixture("airly_no_station.json")),
|
|
):
|
|
flow = config_flow.AirlyFlowHandler()
|
|
flow.hass = hass
|
|
|
|
result = await flow.async_step_user(user_input=CONFIG)
|
|
|
|
assert result["errors"] == {"base": "wrong_location"}
|
|
|
|
|
|
async def test_duplicate_error(hass):
|
|
"""Test that errors are shown when duplicates are added."""
|
|
|
|
with patch(
|
|
"airly._private._RequestsHandler.get",
|
|
return_value=json.loads(load_fixture("airly_valid_station.json")),
|
|
):
|
|
MockConfigEntry(domain=DOMAIN, data=CONFIG).add_to_hass(hass)
|
|
flow = config_flow.AirlyFlowHandler()
|
|
flow.hass = hass
|
|
|
|
result = await flow.async_step_user(user_input=CONFIG)
|
|
|
|
assert result["errors"] == {CONF_NAME: "name_exists"}
|
|
|
|
|
|
async def test_create_entry(hass):
|
|
"""Test that the user step works."""
|
|
|
|
with patch(
|
|
"airly._private._RequestsHandler.get",
|
|
return_value=json.loads(load_fixture("airly_valid_station.json")),
|
|
):
|
|
flow = config_flow.AirlyFlowHandler()
|
|
flow.hass = hass
|
|
|
|
result = await flow.async_step_user(user_input=CONFIG)
|
|
|
|
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
|
assert result["title"] == CONFIG[CONF_NAME]
|
|
assert result["data"][CONF_LATITUDE] == CONFIG[CONF_LATITUDE]
|
|
assert result["data"][CONF_LONGITUDE] == CONFIG[CONF_LONGITUDE]
|
|
assert result["data"][CONF_API_KEY] == CONFIG[CONF_API_KEY]
|