Black
This commit is contained in:
Paulus Schoutsen
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
|
@ -7,16 +7,21 @@ from homeassistant import data_entry_flow, config_entries
|
|||
from homeassistant.components.adguard import config_flow
|
||||
from homeassistant.components.adguard.const import DOMAIN
|
||||
from homeassistant.const import (
|
||||
CONF_HOST, CONF_PASSWORD, CONF_PORT, CONF_SSL, CONF_USERNAME,
|
||||
CONF_VERIFY_SSL)
|
||||
CONF_HOST,
|
||||
CONF_PASSWORD,
|
||||
CONF_PORT,
|
||||
CONF_SSL,
|
||||
CONF_USERNAME,
|
||||
CONF_VERIFY_SSL,
|
||||
)
|
||||
|
||||
from tests.common import MockConfigEntry, mock_coro
|
||||
|
||||
FIXTURE_USER_INPUT = {
|
||||
CONF_HOST: '127.0.0.1',
|
||||
CONF_HOST: "127.0.0.1",
|
||||
CONF_PORT: 3000,
|
||||
CONF_USERNAME: 'user',
|
||||
CONF_PASSWORD: 'pass',
|
||||
CONF_USERNAME: "user",
|
||||
CONF_PASSWORD: "pass",
|
||||
CONF_SSL: True,
|
||||
CONF_VERIFY_SSL: True,
|
||||
}
|
||||
|
|
@ -28,15 +33,15 @@ async def test_show_authenticate_form(hass):
|
|||
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'
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
|
||||
async def test_connection_error(hass, aioclient_mock):
|
||||
"""Test we show user form on AdGuard Home connection error."""
|
||||
aioclient_mock.get(
|
||||
"{}://{}:{}/control/status".format(
|
||||
'https' if FIXTURE_USER_INPUT[CONF_SSL] else 'http',
|
||||
"https" if FIXTURE_USER_INPUT[CONF_SSL] else "http",
|
||||
FIXTURE_USER_INPUT[CONF_HOST],
|
||||
FIXTURE_USER_INPUT[CONF_PORT],
|
||||
),
|
||||
|
|
@ -47,40 +52,38 @@ async def test_connection_error(hass, aioclient_mock):
|
|||
flow.hass = hass
|
||||
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT)
|
||||
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result['step_id'] == 'user'
|
||||
assert result['errors'] == {'base': 'connection_error'}
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result["step_id"] == "user"
|
||||
assert result["errors"] == {"base": "connection_error"}
|
||||
|
||||
|
||||
async def test_full_flow_implementation(hass, aioclient_mock):
|
||||
"""Test registering an integration and finishing flow works."""
|
||||
aioclient_mock.get(
|
||||
"{}://{}:{}/control/status".format(
|
||||
'https' if FIXTURE_USER_INPUT[CONF_SSL] else 'http',
|
||||
"https" if FIXTURE_USER_INPUT[CONF_SSL] else "http",
|
||||
FIXTURE_USER_INPUT[CONF_HOST],
|
||||
FIXTURE_USER_INPUT[CONF_PORT],
|
||||
),
|
||||
json={'version': '1.0'},
|
||||
headers={'Content-Type': 'application/json'},
|
||||
json={"version": "1.0"},
|
||||
headers={"Content-Type": "application/json"},
|
||||
)
|
||||
|
||||
flow = config_flow.AdGuardHomeFlowHandler()
|
||||
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'
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result["step_id"] == "user"
|
||||
|
||||
result = await flow.async_step_user(user_input=FIXTURE_USER_INPUT)
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result['title'] == FIXTURE_USER_INPUT[CONF_HOST]
|
||||
assert result['data'][CONF_HOST] == FIXTURE_USER_INPUT[CONF_HOST]
|
||||
assert result['data'][CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD]
|
||||
assert result['data'][CONF_PORT] == FIXTURE_USER_INPUT[CONF_PORT]
|
||||
assert result['data'][CONF_SSL] == FIXTURE_USER_INPUT[CONF_SSL]
|
||||
assert result['data'][CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME]
|
||||
assert (
|
||||
result['data'][CONF_VERIFY_SSL] == FIXTURE_USER_INPUT[CONF_VERIFY_SSL]
|
||||
)
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result["title"] == FIXTURE_USER_INPUT[CONF_HOST]
|
||||
assert result["data"][CONF_HOST] == FIXTURE_USER_INPUT[CONF_HOST]
|
||||
assert result["data"][CONF_PASSWORD] == FIXTURE_USER_INPUT[CONF_PASSWORD]
|
||||
assert result["data"][CONF_PORT] == FIXTURE_USER_INPUT[CONF_PORT]
|
||||
assert result["data"][CONF_SSL] == FIXTURE_USER_INPUT[CONF_SSL]
|
||||
assert result["data"][CONF_USERNAME] == FIXTURE_USER_INPUT[CONF_USERNAME]
|
||||
assert result["data"][CONF_VERIFY_SSL] == FIXTURE_USER_INPUT[CONF_VERIFY_SSL]
|
||||
|
||||
|
||||
async def test_integration_already_exists(hass):
|
||||
|
|
@ -88,155 +91,140 @@ async def test_integration_already_exists(hass):
|
|||
MockConfigEntry(domain=DOMAIN).add_to_hass(hass)
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
DOMAIN, context={'source': 'user'}
|
||||
DOMAIN, context={"source": "user"}
|
||||
)
|
||||
assert result['type'] == 'abort'
|
||||
assert result['reason'] == 'single_instance_allowed'
|
||||
assert result["type"] == "abort"
|
||||
assert result["reason"] == "single_instance_allowed"
|
||||
|
||||
|
||||
async def test_hassio_single_instance(hass):
|
||||
"""Test we only allow a single config flow."""
|
||||
MockConfigEntry(domain='adguard', data={
|
||||
'host': 'mock-adguard',
|
||||
'port': '3000'
|
||||
}).add_to_hass(hass)
|
||||
MockConfigEntry(
|
||||
domain="adguard", data={"host": "mock-adguard", "port": "3000"}
|
||||
).add_to_hass(hass)
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'adguard',
|
||||
data={
|
||||
'addon': 'AdGuard Home Addon',
|
||||
'host': 'mock-adguard',
|
||||
'port': '3000',
|
||||
},
|
||||
context={'source': 'hassio'}
|
||||
"adguard",
|
||||
data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": "3000"},
|
||||
context={"source": "hassio"},
|
||||
)
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result['reason'] == 'single_instance_allowed'
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result["reason"] == "single_instance_allowed"
|
||||
|
||||
|
||||
async def test_hassio_update_instance_not_running(hass):
|
||||
"""Test we only allow a single config flow."""
|
||||
entry = MockConfigEntry(domain='adguard', data={
|
||||
'host': 'mock-adguard',
|
||||
'port': '3000'
|
||||
})
|
||||
entry = MockConfigEntry(
|
||||
domain="adguard", data={"host": "mock-adguard", "port": "3000"}
|
||||
)
|
||||
entry.add_to_hass(hass)
|
||||
assert entry.state == config_entries.ENTRY_STATE_NOT_LOADED
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'adguard',
|
||||
"adguard",
|
||||
data={
|
||||
'addon': 'AdGuard Home Addon',
|
||||
'host': 'mock-adguard-updated',
|
||||
'port': '3000',
|
||||
"addon": "AdGuard Home Addon",
|
||||
"host": "mock-adguard-updated",
|
||||
"port": "3000",
|
||||
},
|
||||
context={'source': 'hassio'}
|
||||
context={"source": "hassio"},
|
||||
)
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result['reason'] == 'existing_instance_updated'
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result["reason"] == "existing_instance_updated"
|
||||
|
||||
|
||||
async def test_hassio_update_instance_running(hass):
|
||||
"""Test we only allow a single config flow."""
|
||||
entry = MockConfigEntry(domain='adguard', data={
|
||||
'host': 'mock-adguard',
|
||||
'port': '3000',
|
||||
'verify_ssl': False,
|
||||
'username': None,
|
||||
'password': None,
|
||||
'ssl': False,
|
||||
})
|
||||
entry = MockConfigEntry(
|
||||
domain="adguard",
|
||||
data={
|
||||
"host": "mock-adguard",
|
||||
"port": "3000",
|
||||
"verify_ssl": False,
|
||||
"username": None,
|
||||
"password": None,
|
||||
"ssl": False,
|
||||
},
|
||||
)
|
||||
entry.add_to_hass(hass)
|
||||
|
||||
with patch.object(
|
||||
hass.config_entries, 'async_forward_entry_setup',
|
||||
side_effect=lambda *_: mock_coro(True)
|
||||
hass.config_entries,
|
||||
"async_forward_entry_setup",
|
||||
side_effect=lambda *_: mock_coro(True),
|
||||
) as mock_load:
|
||||
assert await hass.config_entries.async_setup(entry.entry_id)
|
||||
assert entry.state == config_entries.ENTRY_STATE_LOADED
|
||||
assert len(mock_load.mock_calls) == 2
|
||||
|
||||
with patch.object(
|
||||
hass.config_entries, 'async_forward_entry_unload',
|
||||
side_effect=lambda *_: mock_coro(True)
|
||||
hass.config_entries,
|
||||
"async_forward_entry_unload",
|
||||
side_effect=lambda *_: mock_coro(True),
|
||||
) as mock_unload, patch.object(
|
||||
hass.config_entries, 'async_forward_entry_setup',
|
||||
side_effect=lambda *_: mock_coro(True)
|
||||
hass.config_entries,
|
||||
"async_forward_entry_setup",
|
||||
side_effect=lambda *_: mock_coro(True),
|
||||
) as mock_load:
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'adguard',
|
||||
"adguard",
|
||||
data={
|
||||
'addon': 'AdGuard Home Addon',
|
||||
'host': 'mock-adguard-updated',
|
||||
'port': '3000',
|
||||
"addon": "AdGuard Home Addon",
|
||||
"host": "mock-adguard-updated",
|
||||
"port": "3000",
|
||||
},
|
||||
context={'source': 'hassio'}
|
||||
context={"source": "hassio"},
|
||||
)
|
||||
assert len(mock_unload.mock_calls) == 2
|
||||
assert len(mock_load.mock_calls) == 2
|
||||
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result['reason'] == 'existing_instance_updated'
|
||||
assert entry.data['host'] == 'mock-adguard-updated'
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_ABORT
|
||||
assert result["reason"] == "existing_instance_updated"
|
||||
assert entry.data["host"] == "mock-adguard-updated"
|
||||
|
||||
|
||||
async def test_hassio_confirm(hass, aioclient_mock):
|
||||
"""Test we can finish a config flow."""
|
||||
aioclient_mock.get(
|
||||
"http://mock-adguard:3000/control/status",
|
||||
json={'version': '1.0'},
|
||||
headers={'Content-Type': 'application/json'},
|
||||
json={"version": "1.0"},
|
||||
headers={"Content-Type": "application/json"},
|
||||
)
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'adguard',
|
||||
data={
|
||||
'addon': 'AdGuard Home Addon',
|
||||
'host': 'mock-adguard',
|
||||
'port': 3000,
|
||||
},
|
||||
context={'source': 'hassio'},
|
||||
"adguard",
|
||||
data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000},
|
||||
context={"source": "hassio"},
|
||||
)
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result['step_id'] == 'hassio_confirm'
|
||||
assert result['description_placeholders'] == {
|
||||
'addon': 'AdGuard Home Addon'
|
||||
}
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result["step_id"] == "hassio_confirm"
|
||||
assert result["description_placeholders"] == {"addon": "AdGuard Home Addon"}
|
||||
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result['flow_id'], {}
|
||||
)
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result['title'] == 'AdGuard Home Addon'
|
||||
assert result['data'][CONF_HOST] == 'mock-adguard'
|
||||
assert result['data'][CONF_PASSWORD] is None
|
||||
assert result['data'][CONF_PORT] == 3000
|
||||
assert result['data'][CONF_SSL] is False
|
||||
assert result['data'][CONF_USERNAME] is None
|
||||
assert result['data'][CONF_VERIFY_SSL]
|
||||
result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
|
||||
assert result["title"] == "AdGuard Home Addon"
|
||||
assert result["data"][CONF_HOST] == "mock-adguard"
|
||||
assert result["data"][CONF_PASSWORD] is None
|
||||
assert result["data"][CONF_PORT] == 3000
|
||||
assert result["data"][CONF_SSL] is False
|
||||
assert result["data"][CONF_USERNAME] is None
|
||||
assert result["data"][CONF_VERIFY_SSL]
|
||||
|
||||
|
||||
async def test_hassio_connection_error(hass, aioclient_mock):
|
||||
"""Test we show hassio confirm form on AdGuard Home connection error."""
|
||||
aioclient_mock.get(
|
||||
"http://mock-adguard:3000/control/status",
|
||||
exc=aiohttp.ClientError,
|
||||
"http://mock-adguard:3000/control/status", exc=aiohttp.ClientError
|
||||
)
|
||||
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'adguard',
|
||||
data={
|
||||
'addon': 'AdGuard Home Addon',
|
||||
'host': 'mock-adguard',
|
||||
'port': 3000,
|
||||
},
|
||||
context={'source': 'hassio'},
|
||||
"adguard",
|
||||
data={"addon": "AdGuard Home Addon", "host": "mock-adguard", "port": 3000},
|
||||
context={"source": "hassio"},
|
||||
)
|
||||
|
||||
result = await hass.config_entries.flow.async_configure(
|
||||
result['flow_id'], {}
|
||||
)
|
||||
result = await hass.config_entries.flow.async_configure(result["flow_id"], {})
|
||||
|
||||
assert result['type'] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result['step_id'] == 'hassio_confirm'
|
||||
assert result['errors'] == {'base': 'connection_error'}
|
||||
assert result["type"] == data_entry_flow.RESULT_TYPE_FORM
|
||||
assert result["step_id"] == "hassio_confirm"
|
||||
assert result["errors"] == {"base": "connection_error"}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue