evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Black

Browse source
This commit is contained in:
Paulus Schoutsen 2019-07-31 12:25:30 -07:00
parent da05dfe708
commit 4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
Download patch file Download diff file Expand all files Collapse all files

42
tests/components/iqvia/test_config_flow.py
View file

@ -10,35 +10,31 @@ from tests.common import MockConfigEntry, MockDependency
@pytest.fixture
def mock_pyiqvia():
"""Mock the pyiqvia library."""
with MockDependency('pyiqvia') as mock_pyiqvia_:
with MockDependency("pyiqvia") as mock_pyiqvia_:
yield mock_pyiqvia_
async def test_duplicate_error(hass):
"""Test that errors are shown when duplicates are added."""
conf = {
CONF_ZIP_CODE: '12345',
}
conf = {CONF_ZIP_CODE: "12345"}
MockConfigEntry(domain=DOMAIN, data=conf).add_to_hass(hass)
flow = config_flow.IQVIAFlowHandler()
flow.hass = hass
result = await flow.async_step_user(user_input=conf)
assert result['errors'] == {CONF_ZIP_CODE: 'identifier_exists'}
assert result["errors"] == {CONF_ZIP_CODE: "identifier_exists"}
async def test_invalid_zip_code(hass, mock_pyiqvia):
"""Test that an invalid ZIP code key throws an error."""
conf = {
CONF_ZIP_CODE: 'abcde',
}
conf = {CONF_ZIP_CODE: "abcde"}
flow = config_flow.IQVIAFlowHandler()
flow.hass = hass
result = await flow.async_step_user(user_input=conf)
assert result['errors'] == {CONF_ZIP_CODE: 'invalid_zip_code'}
assert result["errors"] == {CONF_ZIP_CODE: "invalid_zip_code"}
async def test_show_form(hass):
@ -48,39 +44,31 @@ async def test_show_form(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_step_import(hass, mock_pyiqvia):
"""Test that the import step works."""
conf = {
CONF_ZIP_CODE: '12345',
}
conf = {CONF_ZIP_CODE: "12345"}
flow = config_flow.IQVIAFlowHandler()
flow.hass = hass
result = await flow.async_step_import(import_config=conf)
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result['title'] == '12345'
assert result['data'] == {
CONF_ZIP_CODE: '12345',
}
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == "12345"
assert result["data"] == {CONF_ZIP_CODE: "12345"}
async def test_step_user(hass, mock_pyiqvia):
"""Test that the user step works."""
conf = {
CONF_ZIP_CODE: '12345',
}
conf = {CONF_ZIP_CODE: "12345"}
flow = config_flow.IQVIAFlowHandler()
flow.hass = hass
result = await flow.async_step_user(user_input=conf)
assert result['type'] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result['title'] == '12345'
assert result['data'] == {
CONF_ZIP_CODE: '12345',
}
assert result["type"] == data_entry_flow.RESULT_TYPE_CREATE_ENTRY
assert result["title"] == "12345"
assert result["data"] == {CONF_ZIP_CODE: "12345"}