Format sharkiq with black (#39422)

This commit is contained in:
springstan 2020-08-30 15:56:59 +02:00 committed by GitHub
parent be489e83a1
commit 4294d10789
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 22 additions and 9 deletions

View File

@ -95,7 +95,9 @@ class SharkIqConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
return self.async_abort(reason=errors["base"])
return self.async_show_form(
step_id="reauth", data_schema=SHARKIQ_SCHEMA, errors=errors,
step_id="reauth",
data_schema=SHARKIQ_SCHEMA,
errors=errors,
)

View File

@ -88,7 +88,9 @@ class SharkIqUpdateCoordinator(DataUpdateCoordinator):
if not matching_flows:
self.hass.async_create_task(
self.hass.config_entries.flow.async_init(
DOMAIN, context=flow_context, data=self._config_entry.data,
DOMAIN,
context=flow_context,
data=self._config_entry.data,
)
)

View File

@ -31,10 +31,12 @@ async def test_form(hass):
with patch("sharkiqpy.AylaApi.async_sign_in", return_value=True), patch(
"homeassistant.components.sharkiq.async_setup", return_value=True
) as mock_setup, patch(
"homeassistant.components.sharkiq.async_setup_entry", return_value=True,
"homeassistant.components.sharkiq.async_setup_entry",
return_value=True,
) as mock_setup_entry:
result2 = await hass.config_entries.flow.async_configure(
result["flow_id"], CONFIG,
result["flow_id"],
CONFIG,
)
await hass.async_block_till_done()
@ -61,7 +63,8 @@ async def test_form_invalid_auth(hass):
return_value=mocked_ayla,
):
result2 = await hass.config_entries.flow.async_configure(
result["flow_id"], CONFIG,
result["flow_id"],
CONFIG,
)
assert result2["type"] == "form"
@ -80,7 +83,8 @@ async def test_form_cannot_connect(hass):
return_value=mocked_ayla,
):
result2 = await hass.config_entries.flow.async_configure(
result["flow_id"], CONFIG,
result["flow_id"],
CONFIG,
)
assert result2["type"] == "form"
@ -109,7 +113,8 @@ async def test_form_other_error(hass):
async def test_reauth(hass):
"""Test reauth flow."""
with patch(
"homeassistant.components.sharkiq.vacuum.async_setup_entry", return_value=True,
"homeassistant.components.sharkiq.vacuum.async_setup_entry",
return_value=True,
), patch("sharkiqpy.AylaApi.async_sign_in", return_value=True):
mock_config = MockConfigEntry(domain=DOMAIN, unique_id=UNIQUE_ID, data=CONFIG)
mock_config.add_to_hass(hass)
@ -124,14 +129,18 @@ async def test_reauth(hass):
with patch("sharkiqpy.AylaApi.async_sign_in", side_effect=SharkIqAuthError):
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": "reauth", "unique_id": UNIQUE_ID}, data=CONFIG,
DOMAIN,
context={"source": "reauth", "unique_id": UNIQUE_ID},
data=CONFIG,
)
assert result["type"] == "form"
assert result["errors"] == {"base": "invalid_auth"}
with patch("sharkiqpy.AylaApi.async_sign_in", side_effect=RuntimeError):
result = await hass.config_entries.flow.async_init(
DOMAIN, context={"source": "reauth", "unique_id": UNIQUE_ID}, data=CONFIG,
DOMAIN,
context={"source": "reauth", "unique_id": UNIQUE_ID},
data=CONFIG,
)
assert result["type"] == "abort"