From eb339b9793af90f05021e46649846b2f4f9668cb Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Sat, 23 Jan 2021 19:53:45 +0100 Subject: [PATCH] Fix flaky SRP_Energy test (#45471) --- tests/components/srp_energy/test_config_flow.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/components/srp_energy/test_config_flow.py b/tests/components/srp_energy/test_config_flow.py index adad1878326..c63843723b1 100644 --- a/tests/components/srp_energy/test_config_flow.py +++ b/tests/components/srp_energy/test_config_flow.py @@ -18,7 +18,14 @@ async def test_form(hass): assert result["errors"] == {} # Fill submit form data for config entry - with patch("homeassistant.components.srp_energy.config_flow.SrpEnergyClient"): + with patch( + "homeassistant.components.srp_energy.config_flow.SrpEnergyClient" + ), patch( + "homeassistant.components.srp_energy.async_setup", return_value=True + ) as mock_setup, patch( + "homeassistant.components.srp_energy.async_setup_entry", + return_value=True, + ) as mock_setup_entry: result = await hass.config_entries.flow.async_configure( result["flow_id"], @@ -29,6 +36,9 @@ async def test_form(hass): assert result["title"] == "Test" assert result["data"][CONF_IS_TOU] is False + assert len(mock_setup.mock_calls) == 1 + assert len(mock_setup_entry.mock_calls) == 1 + async def test_form_invalid_auth(hass): """Test user config with invalid auth."""