Black
This commit is contained in:
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
@ -7,55 +7,55 @@ from tests.common import mock_coro
|
|||
|
||||
async def test_config_flow_import(hass):
|
||||
"""Test that we automatically create a config flow."""
|
||||
assert not hass.config_entries.async_entries('owntracks')
|
||||
assert await async_setup_component(hass, 'owntracks', {
|
||||
'owntracks': {
|
||||
|
||||
}
|
||||
})
|
||||
assert not hass.config_entries.async_entries("owntracks")
|
||||
assert await async_setup_component(hass, "owntracks", {"owntracks": {}})
|
||||
await hass.async_block_till_done()
|
||||
assert hass.config_entries.async_entries('owntracks')
|
||||
assert hass.config_entries.async_entries("owntracks")
|
||||
|
||||
|
||||
async def test_config_flow_unload(hass):
|
||||
"""Test unloading a config flow."""
|
||||
with patch('homeassistant.config_entries.ConfigEntries'
|
||||
'.async_forward_entry_setup') as mock_forward:
|
||||
with patch(
|
||||
"homeassistant.config_entries.ConfigEntries" ".async_forward_entry_setup"
|
||||
) as mock_forward:
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'owntracks', context={'source': 'import'},
|
||||
data={}
|
||||
"owntracks", context={"source": "import"}, data={}
|
||||
)
|
||||
|
||||
assert len(mock_forward.mock_calls) == 1
|
||||
entry = result['result']
|
||||
entry = result["result"]
|
||||
|
||||
assert mock_forward.mock_calls[0][1][0] is entry
|
||||
assert mock_forward.mock_calls[0][1][1] == 'device_tracker'
|
||||
assert entry.data['webhook_id'] in hass.data['webhook']
|
||||
assert mock_forward.mock_calls[0][1][1] == "device_tracker"
|
||||
assert entry.data["webhook_id"] in hass.data["webhook"]
|
||||
|
||||
with patch('homeassistant.config_entries.ConfigEntries'
|
||||
'.async_forward_entry_unload', return_value=mock_coro()
|
||||
) as mock_unload:
|
||||
with patch(
|
||||
"homeassistant.config_entries.ConfigEntries" ".async_forward_entry_unload",
|
||||
return_value=mock_coro(),
|
||||
) as mock_unload:
|
||||
assert await hass.config_entries.async_unload(entry.entry_id)
|
||||
|
||||
assert len(mock_unload.mock_calls) == 1
|
||||
assert mock_forward.mock_calls[0][1][0] is entry
|
||||
assert mock_forward.mock_calls[0][1][1] == 'device_tracker'
|
||||
assert entry.data['webhook_id'] not in hass.data['webhook']
|
||||
assert mock_forward.mock_calls[0][1][1] == "device_tracker"
|
||||
assert entry.data["webhook_id"] not in hass.data["webhook"]
|
||||
|
||||
|
||||
async def test_with_cloud_sub(hass):
|
||||
"""Test creating a config flow while subscribed."""
|
||||
with patch('homeassistant.components.cloud.async_active_subscription',
|
||||
return_value=True), \
|
||||
patch('homeassistant.components.cloud.async_create_cloudhook',
|
||||
return_value=mock_coro('https://hooks.nabu.casa/ABCD')):
|
||||
with patch(
|
||||
"homeassistant.components.cloud.async_active_subscription", return_value=True
|
||||
), patch(
|
||||
"homeassistant.components.cloud.async_create_cloudhook",
|
||||
return_value=mock_coro("https://hooks.nabu.casa/ABCD"),
|
||||
):
|
||||
result = await hass.config_entries.flow.async_init(
|
||||
'owntracks', context={'source': 'user'},
|
||||
data={}
|
||||
"owntracks", context={"source": "user"}, data={}
|
||||
)
|
||||
|
||||
entry = result['result']
|
||||
assert entry.data['cloudhook']
|
||||
assert result['description_placeholders']['webhook_url'] == \
|
||||
'https://hooks.nabu.casa/ABCD'
|
||||
entry = result["result"]
|
||||
assert entry.data["cloudhook"]
|
||||
assert (
|
||||
result["description_placeholders"]["webhook_url"]
|
||||
== "https://hooks.nabu.casa/ABCD"
|
||||
)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue