Reorganize drop_connect tests (#107148)

tests/components/drop_connect/test_init.py

@@ -0,0 +1,66 @@
+"""Test DROP initialisation."""
+
+from homeassistant.config_entries import ConfigEntryState
+from homeassistant.const import STATE_UNAVAILABLE, STATE_UNKNOWN
+from homeassistant.core import HomeAssistant
+
+from .common import (
+    TEST_DATA_HUB,
+    TEST_DATA_HUB_RESET,
+    TEST_DATA_HUB_TOPIC,
+    config_entry_hub,
+)
+
+from tests.common import async_fire_mqtt_message
+from tests.typing import MqttMockHAClient
+
+
+async def test_bad_json(hass: HomeAssistant, mqtt_mock: MqttMockHAClient) -> None:
+    """Test bad JSON."""
+    entry = config_entry_hub()
+    entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(entry.entry_id)
+
+    current_flow_sensor_name = "sensor.hub_drop_1_c0ffee_water_flow_rate"
+    assert hass.states.get(current_flow_sensor_name).state == STATE_UNKNOWN
+
+    async_fire_mqtt_message(hass, TEST_DATA_HUB_TOPIC, "{BAD JSON}")
+    await hass.async_block_till_done()
+    assert hass.states.get(current_flow_sensor_name).state == STATE_UNKNOWN
+
+
+async def test_unload(hass: HomeAssistant, mqtt_mock: MqttMockHAClient) -> None:
+    """Test entity unload."""
+    # Load the hub device
+    entry = config_entry_hub()
+    entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(entry.entry_id)
+
+    current_flow_sensor_name = "sensor.hub_drop_1_c0ffee_water_flow_rate"
+    assert hass.states.get(current_flow_sensor_name).state == STATE_UNKNOWN
+
+    async_fire_mqtt_message(hass, TEST_DATA_HUB_TOPIC, TEST_DATA_HUB_RESET)
+    await hass.async_block_till_done()
+    assert hass.states.get(current_flow_sensor_name).state == "0.0"
+
+    async_fire_mqtt_message(hass, TEST_DATA_HUB_TOPIC, TEST_DATA_HUB)
+    await hass.async_block_till_done()
+
+    assert hass.states.get(current_flow_sensor_name).state == "5.77"
+
+    # Unload the device
+    await hass.config_entries.async_unload(entry.entry_id)
+    assert entry.state is ConfigEntryState.NOT_LOADED
+
+    # Verify sensor is unavailable
+    assert hass.states.get(current_flow_sensor_name).state == STATE_UNAVAILABLE
+
+
+async def test_no_mqtt(hass: HomeAssistant) -> None:
+    """Test no MQTT."""
+    entry = config_entry_hub()
+    entry.add_to_hass(hass)
+    assert await hass.config_entries.async_setup(entry.entry_id) is False
+
+    protect_mode_select_name = "select.hub_drop_1_c0ffee_protect_mode"
+    assert hass.states.get(protect_mode_select_name) is None