Merge pull request #246 from jwillemsen/jwi-codecovupdatecoord

Moved some test code to another method to increase coverage

Author: jwillemsen

tests/test_init.py

@@ -450,6 +450,21 @@ async def test_water_heater(
 
         assert len(responses.calls) == 9
 
+        # In order to call update_entity we need to setup the HA core
+        await async_setup_component(hass, "homeassistant", {})
+
+        # Try to call update_entity service to trigger an update but because we just did patches
+        # this is a noop
+        await hass.services.async_call(
+            HA_DOMAIN,
+            SERVICE_UPDATE_ENTITY,
+            {ATTR_ENTITY_ID: "water_heater.altherma"},
+            blocking=True,
+        )
+        await hass.async_block_till_done()
+
+        assert len(responses.calls) == 9
+
 
 @responses.activate
 async def test_climate(
@@ -933,14 +948,14 @@ async def test_climate(
         assert len(responses.calls) == 32
         assert hass.states.get("climate.werkkamer_room_temperature").state == HVACMode.DRY
 
+        # In order to call update_entity we need to setup the HA core
+        await async_setup_component(hass, "homeassistant", {})
+
         # We patch the scan_ignore method to zero so that the coordinator will pull again
         with patch(
             "custom_components.daikin_onecta.OnectaDataUpdateCoordinator.scan_ignore",
             return_value=0,
         ):
-            # In order to call update_entity we need to setup the HA core
-            await async_setup_component(hass, "homeassistant", {})
-
             with responses.RequestsMock() as rsps:
                 rsps.get(DAIKIN_API_URL + "/v1/gateway-devices", status=200, json=load_fixture_json("altherma"))
                 # Call update_entity service to trigger an update
@@ -954,15 +969,3 @@ async def test_climate(
 
                 assert len(rsps.calls) == 1
                 assert rsps.calls[0].request.url == DAIKIN_API_URL + "/v1/gateway-devices"
-
-                # Try to call update_entity service to trigger an update but because we just did this it
-                # is a noop
-                await hass.services.async_call(
-                    HA_DOMAIN,
-                    SERVICE_UPDATE_ENTITY,
-                    {ATTR_ENTITY_ID: "climate.werkkamer_room_temperature"},
-                    blocking=True,
-                )
-                await hass.async_block_till_done()
-
-                assert len(rsps.calls) == 1