Guard for no discovery info in command_line (#123717)

homeassistant/components/command_line/binary_sensor.py

@@ -40,6 +40,8 @@ async def async_setup_platform(
     discovery_info: DiscoveryInfoType | None = None,
 ) -> None:
     """Set up the Command line Binary Sensor."""
+    if not discovery_info:
+        return
 
     discovery_info = cast(DiscoveryInfoType, discovery_info)
     binary_sensor_config = discovery_info

homeassistant/components/command_line/cover.py

@@ -37,6 +37,8 @@ async def async_setup_platform(
     discovery_info: DiscoveryInfoType | None = None,
 ) -> None:
     """Set up cover controlled by shell commands."""
+    if not discovery_info:
+        return
 
     covers = []
     discovery_info = cast(DiscoveryInfoType, discovery_info)

homeassistant/components/command_line/notify.py

@@ -21,8 +21,10 @@ def get_service(
     hass: HomeAssistant,
     config: ConfigType,
     discovery_info: DiscoveryInfoType | None = None,
-) -> CommandLineNotificationService:
+) -> CommandLineNotificationService | None:
     """Get the Command Line notification service."""
+    if not discovery_info:
+        return None
 
     discovery_info = cast(DiscoveryInfoType, discovery_info)
     notify_config = discovery_info

homeassistant/components/command_line/sensor.py

@@ -48,6 +48,8 @@ async def async_setup_platform(
     discovery_info: DiscoveryInfoType | None = None,
 ) -> None:
     """Set up the Command Sensor."""
+    if not discovery_info:
+        return
 
     discovery_info = cast(DiscoveryInfoType, discovery_info)
     sensor_config = discovery_info

homeassistant/components/command_line/switch.py

@@ -36,9 +36,9 @@ async def async_setup_platform(
     discovery_info: DiscoveryInfoType | None = None,
 ) -> None:
     """Find and return switches controlled by shell commands."""
-
     if not discovery_info:
         return
+
     switches = []
     discovery_info = cast(DiscoveryInfoType, discovery_info)
     entities: dict[str, dict[str, Any]] = {

tests/components/command_line/test_binary_sensor.py

@@ -56,6 +56,24 @@ async def test_setup_integration_yaml(
     assert entity_state.name == "Test"
 
 
+async def test_setup_platform_yaml(hass: HomeAssistant) -> None:
+    """Test setting up the platform with platform yaml."""
+    await setup.async_setup_component(
+        hass,
+        "binary_sensor",
+        {
+            "binary_sensor": {
+                "platform": "command_line",
+                "command": "echo 1",
+                "payload_on": "1",
+                "payload_off": "0",
+            }
+        },
+    )
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all()) == 0
+
+
 @pytest.mark.parametrize(
     "get_config",
     [

tests/components/command_line/test_cover.py

@@ -36,6 +36,24 @@ from . import mock_asyncio_subprocess_run
 from tests.common import async_fire_time_changed
 
 
+async def test_setup_platform_yaml(hass: HomeAssistant) -> None:
+    """Test setting up the platform with platform yaml."""
+    await setup.async_setup_component(
+        hass,
+        "cover",
+        {
+            "cover": {
+                "platform": "command_line",
+                "command": "echo 1",
+                "payload_on": "1",
+                "payload_off": "0",
+            }
+        },
+    )
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all()) == 0
+
+
 async def test_no_poll_when_cover_has_no_command_state(hass: HomeAssistant) -> None:
     """Test that the cover does not polls when there's no state command."""
 

tests/components/command_line/test_notify.py

@@ -16,6 +16,24 @@ from homeassistant.components.notify import DOMAIN as NOTIFY_DOMAIN
 from homeassistant.core import HomeAssistant
 
 
+async def test_setup_platform_yaml(hass: HomeAssistant) -> None:
+    """Test setting up the platform with platform yaml."""
+    await setup.async_setup_component(
+        hass,
+        "notify",
+        {
+            "notify": {
+                "platform": "command_line",
+                "command": "echo 1",
+                "payload_on": "1",
+                "payload_off": "0",
+            }
+        },
+    )
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all()) == 0
+
+
 @pytest.mark.parametrize(
     "get_config",
     [

tests/components/command_line/test_sensor.py

@@ -27,6 +27,24 @@ from . import mock_asyncio_subprocess_run
 from tests.common import async_fire_time_changed
 
 
+async def test_setup_platform_yaml(hass: HomeAssistant) -> None:
+    """Test setting up the platform with platform yaml."""
+    await setup.async_setup_component(
+        hass,
+        "sensor",
+        {
+            "sensor": {
+                "platform": "command_line",
+                "command": "echo 1",
+                "payload_on": "1",
+                "payload_off": "0",
+            }
+        },
+    )
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all()) == 0
+
+
 @pytest.mark.parametrize(
     "get_config",
     [

tests/components/command_line/test_switch.py

@@ -37,6 +37,24 @@ from . import mock_asyncio_subprocess_run
 from tests.common import async_fire_time_changed
 
 
+async def test_setup_platform_yaml(hass: HomeAssistant) -> None:
+    """Test setting up the platform with platform yaml."""
+    await setup.async_setup_component(
+        hass,
+        "switch",
+        {
+            "switch": {
+                "platform": "command_line",
+                "command": "echo 1",
+                "payload_on": "1",
+                "payload_off": "0",
+            }
+        },
+    )
+    await hass.async_block_till_done()
+    assert len(hass.states.async_all()) == 0
+
+
 async def test_state_integration_yaml(hass: HomeAssistant) -> None:
     """Test with none state."""
     with tempfile.TemporaryDirectory() as tempdirname: