Add config flow for integration sensor (#68288)

tests/components/integration/test_init.py

@@ -0,0 +1,61 @@
+"""Test the Integration - Riemann sum integral integration."""
+import pytest
+
+from homeassistant.components.integration.const import DOMAIN
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import entity_registry as er
+
+from tests.common import MockConfigEntry
+
+
+@pytest.mark.parametrize("platform", ("sensor",))
+async def test_setup_and_remove_config_entry(
+    hass: HomeAssistant,
+    platform: str,
+) -> None:
+    """Test setting up and removing a config entry."""
+    input_sensor_entity_id = "sensor.input"
+    registry = er.async_get(hass)
+    integration_entity_id = f"{platform}.my_integration"
+
+    # Setup the config entry
+    config_entry = MockConfigEntry(
+        data={},
+        domain=DOMAIN,
+        options={
+            "method": "trapezoidal",
+            "name": "My integration",
+            "round": 1.0,
+            "source": "sensor.input",
+            "unit_prefix": "k",
+            "unit_time": "min",
+        },
+        title="My integration",
+    )
+    config_entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    # Check the entity is registered in the entity registry
+    assert registry.async_get(integration_entity_id) is not None
+
+    # Check the platform is setup correctly
+    state = hass.states.get(integration_entity_id)
+    assert state.state == "unknown"
+    assert "unit_of_measurement" not in state.attributes
+    assert state.attributes["source"] == "sensor.input"
+
+    hass.states.async_set(input_sensor_entity_id, 10, {"unit_of_measurement": "cat"})
+    hass.states.async_set(input_sensor_entity_id, 11, {"unit_of_measurement": "cat"})
+    await hass.async_block_till_done()
+    state = hass.states.get(integration_entity_id)
+    assert state.state != "unknown"
+    assert state.attributes["unit_of_measurement"] == "kcatmin"
+
+    # Remove the config entry
+    assert await hass.config_entries.async_remove(config_entry.entry_id)
+    await hass.async_block_till_done()
+
+    # Check the state and entity registry entry are removed
+    assert hass.states.get(integration_entity_id) is None
+    assert registry.async_get(integration_entity_id) is None