From 91201155776a3c648b5f5221991145601c73e0cb Mon Sep 17 00:00:00 2001
From: CrazyMan2000 <45339639+CrazyMan2000@users.noreply.github.com>
Date: Tue, 9 Jul 2024 23:28:41 +0200
Subject: [PATCH] Split media player select input test (#121633)

---
 tests/components/alexa/test_capabilities.py | 41 ++++++++++++++-------
 1 file changed, 28 insertions(+), 13 deletions(-)

diff --git a/tests/components/alexa/test_capabilities.py b/tests/components/alexa/test_capabilities.py
index 9caa703335e..162149f095b 100644
--- a/tests/components/alexa/test_capabilities.py
+++ b/tests/components/alexa/test_capabilities.py
@@ -234,7 +234,6 @@ async def test_api_increase_color_temp(
         ("media_player", "GAME CONSOLE", ["tv", "game console", 10000], 1),
         ("media_player", "SATELLITE TV", ["satellite-tv", "game console", None], 0),
         ("media_player", "SATELLITE TV", ["satellite_tv", "game console"], 0),
-        ("media_player", "BAD DEVICE", ["satellite_tv", "game console"], None),
     ],
 )
 async def test_api_select_input(
@@ -255,18 +254,6 @@ async def test_api_select_input(
         },
     )
 
-    # test where no source matches
-    if idx is None:
-        await assert_request_fails(
-            "Alexa.InputController",
-            "SelectInput",
-            "media_player#test",
-            "media_player.select_source",
-            hass,
-            payload={"input": payload},
-        )
-        return
-
     call, _ = await assert_request_calls_service(
         "Alexa.InputController",
         "SelectInput",
@@ -278,6 +265,34 @@ async def test_api_select_input(
     assert call.data["source"] == source_list[idx]
 
 
+@pytest.mark.parametrize(
+    ("source_list"),
+    [(["satellite_tv", "game console"]), ([])],
+)
+async def test_api_select_input_fails(
+    hass: HomeAssistant,
+    source_list: list[Any],
+) -> None:
+    """Test api set input process fails."""
+    hass.states.async_set(
+        "media_player.test",
+        "off",
+        {
+            "friendly_name": "Test media player",
+            "source": "unknown",
+            "source_list": source_list,
+        },
+    )
+    await assert_request_fails(
+        "Alexa.InputController",
+        "SelectInput",
+        "media_player#test",
+        "media_player.select_source",
+        hass,
+        payload={"input": "BAD DEVICE"},
+    )
+
+
 @pytest.mark.parametrize(
     ("activity", "activity_list", "target_activity_index"),
     [