Extend tests

home-assistant · MartinHjelmare · May 4, 2021 · Jul 12, 2020 · Jul 13, 2020 · Jul 13, 2020
commit bd0f86c5f07d95808335e9de1d1a06c7bd76096c
diff --git a/tests/components/buienradar/test_config_flow.py b/tests/components/buienradar/test_config_flow.py
@@ -68,7 +68,6 @@ async def test_config_flow_already_configured_weather(hass):
 
 async def test_import_camera(hass):
     """Test import of camera."""
-
     with patch(
         "homeassistant.components.buienradar.async_setup_entry", return_value=True
     ):

diff --git a/tests/components/buienradar/test_init.py b/tests/components/buienradar/test_init.py
@@ -3,6 +3,7 @@
 
 from homeassistant.components.buienradar import async_setup
 from homeassistant.components.buienradar.const import DOMAIN
+from homeassistant.helpers.entity_registry import async_get_registry
 
 
 async def test_import_all(hass):
@@ -45,7 +46,18 @@ async def test_import_all(hass):
 
 async def test_import_camera(hass):
     """Test import of camera platform."""
-    config = {"camera 1": [{"platform": "buienradar", "country_code": "NL"}]}
+    entity_registry = await async_get_registry(hass)
+    entity_registry.async_get_or_create(
+        domain="camera",
+        platform="buienradar",
+        unique_id="512_NL",
+        original_name="test_name",
+    )
+    await hass.async_block_till_done()
+
+    config = {
+        "camera 1": [{"platform": "buienradar", "country_code": "NL", "dimension": 512}]
+    }
 
     with patch(
         "homeassistant.components.buienradar.async_setup_entry", return_value=True
@@ -68,3 +80,12 @@ async def test_import_camera(hass):
         "delta": 600,
         "name": "Buienradar",
     }
+
+    entity_id = entity_registry.async_get_entity_id(
+        "camera",
+        "buienradar",
+        f"{hass.config.latitude:2.6f}{hass.config.longitude:2.6f}",
+    )
+    assert entity_id
+    entity = entity_registry.async_get(entity_id)
+    assert entity.original_name == "test_name"