diff --git a/homeassistant/components/sharkiq/config_flow.py b/homeassistant/components/sharkiq/config_flow.py index 235f09b11da..b2b85d6cf36 100644 --- a/homeassistant/components/sharkiq/config_flow.py +++ b/homeassistant/components/sharkiq/config_flow.py @@ -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, ) diff --git a/homeassistant/components/sharkiq/update_coordinator.py b/homeassistant/components/sharkiq/update_coordinator.py index b19f12eb403..c498307ac9d 100644 --- a/homeassistant/components/sharkiq/update_coordinator.py +++ b/homeassistant/components/sharkiq/update_coordinator.py @@ -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, ) ) diff --git a/tests/components/sharkiq/test_config_flow.py b/tests/components/sharkiq/test_config_flow.py index 5cf6bee25c4..0d1612f857b 100644 --- a/tests/components/sharkiq/test_config_flow.py +++ b/tests/components/sharkiq/test_config_flow.py @@ -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"