Improve type annotations in websocket_api tests (#100198)

tests/components/websocket_api/test_commands.py

@@ -92,7 +92,9 @@ def _apply_entities_changes(state_dict: dict, change_dict: dict) -> None:
             del state_dict[STATE_KEY_LONG_NAMES[key]][item]
 
 
-async def test_fire_event(hass: HomeAssistant, websocket_client) -> None:
+async def test_fire_event(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test fire event command."""
     runs = []
 
@@ -121,7 +123,9 @@ async def test_fire_event(hass: HomeAssistant, websocket_client) -> None:
     assert runs[0].data == {"hello": "world"}
 
 
-async def test_fire_event_without_data(hass: HomeAssistant, websocket_client) -> None:
+async def test_fire_event_without_data(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test fire event command."""
     runs = []
 
@@ -149,7 +153,9 @@ async def test_fire_event_without_data(hass: HomeAssistant, websocket_client) ->
     assert runs[0].data == {}
 
 
-async def test_call_service(hass: HomeAssistant, websocket_client) -> None:
+async def test_call_service(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test call service command."""
     calls = async_mock_service(hass, "domain_test", "test_service")
 
@@ -179,7 +185,7 @@ async def test_call_service(hass: HomeAssistant, websocket_client) -> None:
 
 @pytest.mark.parametrize("command", ("call_service", "call_service_action"))
 async def test_call_service_blocking(
-    hass: HomeAssistant, websocket_client, command
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket, command
 ) -> None:
     """Test call service commands block, except for homeassistant restart / stop."""
     with patch(
@@ -256,7 +262,9 @@ async def test_call_service_blocking(
     )
 
 
-async def test_call_service_target(hass: HomeAssistant, websocket_client) -> None:
+async def test_call_service_target(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test call service command with target."""
     calls = async_mock_service(hass, "domain_test", "test_service")
 
@@ -316,7 +324,9 @@ async def test_call_service_target_template(
     assert msg["error"]["code"] == const.ERR_INVALID_FORMAT
 
 
-async def test_call_service_not_found(hass: HomeAssistant, websocket_client) -> None:
+async def test_call_service_not_found(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test call service command."""
     await websocket_client.send_json(
         {
@@ -433,7 +443,9 @@ async def test_call_service_schema_validation_error(
     assert len(calls) == 0
 
 
-async def test_call_service_error(hass: HomeAssistant, websocket_client) -> None:
+async def test_call_service_error(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test call service command with error."""
 
     @callback
@@ -526,7 +538,9 @@ async def test_subscribe_unsubscribe_events(
     assert sum(hass.bus.async_listeners().values()) == init_count
 
 
-async def test_get_states(hass: HomeAssistant, websocket_client) -> None:
+async def test_get_states(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test get_states command."""
     hass.states.async_set("greeting.hello", "world")
     hass.states.async_set("greeting.bye", "universe")
@@ -545,7 +559,9 @@ async def test_get_states(hass: HomeAssistant, websocket_client) -> None:
     assert msg["result"] == states
 
 
-async def test_get_services(hass: HomeAssistant, websocket_client) -> None:
+async def test_get_services(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test get_services command."""
     for id_ in (5, 6):
         await websocket_client.send_json({"id": id_, "type": "get_services"})
@@ -557,7 +573,9 @@ async def test_get_services(hass: HomeAssistant, websocket_client) -> None:
         assert msg["result"] == hass.services.async_services()
 
 
-async def test_get_config(hass: HomeAssistant, websocket_client) -> None:
+async def test_get_config(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test get_config command."""
     await websocket_client.send_json({"id": 5, "type": "get_config"})
 
@@ -584,7 +602,7 @@ async def test_get_config(hass: HomeAssistant, websocket_client) -> None:
     assert msg["result"] == hass.config.as_dict()
 
 
-async def test_ping(websocket_client) -> None:
+async def test_ping(websocket_client: MockHAClientWebSocket) -> None:
     """Test get_panels command."""
     await websocket_client.send_json({"id": 5, "type": "ping"})
 
@@ -637,7 +655,7 @@ async def test_call_service_context_with_user(
 
 
 async def test_subscribe_requires_admin(
-    websocket_client, hass_admin_user: MockUser
+    websocket_client: MockHAClientWebSocket, hass_admin_user: MockUser
 ) -> None:
     """Test subscribing events without being admin."""
     hass_admin_user.groups = []
@@ -668,7 +686,9 @@ async def test_states_filters_visible(
     assert msg["result"][0]["entity_id"] == "test.entity"
 
 
-async def test_get_states_not_allows_nan(hass: HomeAssistant, websocket_client) -> None:
+async def test_get_states_not_allows_nan(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test get_states command converts NaN to None."""
     hass.states.async_set("greeting.hello", "world")
     hass.states.async_set("greeting.bad", "data", {"hello": float("NaN")})
@@ -691,7 +711,9 @@ async def test_get_states_not_allows_nan(hass: HomeAssistant, websocket_client)
 
 
 async def test_subscribe_unsubscribe_events_whitelist(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe events on whitelist."""
     hass_admin_user.groups = []
@@ -728,7 +750,9 @@ async def test_subscribe_unsubscribe_events_whitelist(
 
 
 async def test_subscribe_unsubscribe_events_state_changed(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe state_changed events."""
     hass_admin_user.groups = []
@@ -754,7 +778,9 @@ async def test_subscribe_unsubscribe_events_state_changed(
 
 
 async def test_subscribe_entities_with_unserializable_state(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe entities with an unserializeable state."""
 
@@ -871,7 +897,9 @@ async def test_subscribe_entities_with_unserializable_state(
 
 
 async def test_subscribe_unsubscribe_entities(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe entities."""
 
@@ -1037,7 +1065,9 @@ async def test_subscribe_unsubscribe_entities(
 
 
 async def test_subscribe_unsubscribe_entities_specific_entities(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe entities with a list of entity ids."""
 
@@ -1376,7 +1406,7 @@ async def test_render_template_with_error(
 )
 async def test_render_template_with_timeout_and_error(
     hass: HomeAssistant,
-    websocket_client,
+    websocket_client: MockHAClientWebSocket,
     caplog: pytest.LogCaptureFixture,
     template: str,
     expected_events: list[dict[str, str]],
@@ -1592,7 +1622,7 @@ async def test_render_template_strict_with_timeout_and_error_2(
 )
 async def test_render_template_error_in_template_code(
     hass: HomeAssistant,
-    websocket_client,
+    websocket_client: MockHAClientWebSocket,
     caplog: pytest.LogCaptureFixture,
     template: str,
     expected_events_1: list[dict[str, str]],
@@ -1691,7 +1721,9 @@ async def test_render_template_error_in_template_code_2(
 
 
 async def test_render_template_with_delayed_error(
-    hass: HomeAssistant, websocket_client, caplog: pytest.LogCaptureFixture
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test a template with an error that only happens after a state change.
 
@@ -1815,7 +1847,9 @@ async def test_render_template_with_delayed_error_2(
 
 
 async def test_render_template_with_timeout(
-    hass: HomeAssistant, websocket_client, caplog: pytest.LogCaptureFixture
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    caplog: pytest.LogCaptureFixture,
 ) -> None:
     """Test a template that will timeout."""
 
@@ -1859,7 +1893,9 @@ async def test_render_template_returns_with_match_all(
     assert msg["success"]
 
 
-async def test_manifest_list(hass: HomeAssistant, websocket_client) -> None:
+async def test_manifest_list(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test loading manifests."""
     http = await async_get_integration(hass, "http")
     websocket_api = await async_get_integration(hass, "websocket_api")
@@ -1897,7 +1933,9 @@ async def test_manifest_list_specific_integrations(
     ]
 
 
-async def test_manifest_get(hass: HomeAssistant, websocket_client) -> None:
+async def test_manifest_get(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test getting a manifest."""
     hue = await async_get_integration(hass, "hue")
 
@@ -1924,7 +1962,9 @@ async def test_manifest_get(hass: HomeAssistant, websocket_client) -> None:
 
 
 async def test_entity_source_admin(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Check that we fetch sources correctly."""
     platform = MockEntityPlatform(hass)
@@ -1963,7 +2003,9 @@ async def test_entity_source_admin(
     }
 
 
-async def test_subscribe_trigger(hass: HomeAssistant, websocket_client) -> None:
+async def test_subscribe_trigger(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test subscribing to a trigger."""
     init_count = sum(hass.bus.async_listeners().values())
 
@@ -2017,7 +2059,9 @@ async def test_subscribe_trigger(hass: HomeAssistant, websocket_client) -> None:
     assert sum(hass.bus.async_listeners().values()) == init_count
 
 
-async def test_test_condition(hass: HomeAssistant, websocket_client) -> None:
+async def test_test_condition(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test testing a condition."""
     hass.states.async_set("hello.world", "paulus")
 
@@ -2077,7 +2121,9 @@ async def test_test_condition(hass: HomeAssistant, websocket_client) -> None:
     assert msg["result"]["result"] is False
 
 
-async def test_execute_script(hass: HomeAssistant, websocket_client) -> None:
+async def test_execute_script(
+    hass: HomeAssistant, websocket_client: MockHAClientWebSocket
+) -> None:
     """Test testing a condition."""
     calls = async_mock_service(
         hass, "domain_test", "test_service", response={"hello": "world"}
@@ -2226,7 +2272,9 @@ async def test_execute_script_with_dynamically_validated_action(
 
 
 async def test_subscribe_unsubscribe_bootstrap_integrations(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe bootstrap_integrations."""
     await websocket_client.send_json(
@@ -2248,7 +2296,9 @@ async def test_subscribe_unsubscribe_bootstrap_integrations(
 
 
 async def test_integration_setup_info(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test subscribe/unsubscribe bootstrap_integrations."""
     hass.data[DATA_SETUP_TIME] = {
@@ -2284,7 +2334,9 @@ async def test_integration_setup_info(
         ("action", [{"service": "domain_test.test_service"}]),
     ),
 )
-async def test_validate_config_works(websocket_client, key, config) -> None:
+async def test_validate_config_works(
+    websocket_client: MockHAClientWebSocket, key, config
+) -> None:
     """Test config validation."""
     await websocket_client.send_json({"id": 7, "type": "validate_config", key: config})
 
@@ -2323,7 +2375,9 @@ async def test_validate_config_works(websocket_client, key, config) -> None:
         ),
     ),
 )
-async def test_validate_config_invalid(websocket_client, key, config, error) -> None:
+async def test_validate_config_invalid(
+    websocket_client: MockHAClientWebSocket, key, config, error
+) -> None:
     """Test config validation."""
     await websocket_client.send_json({"id": 7, "type": "validate_config", key: config})
 
@@ -2335,7 +2389,9 @@ async def test_validate_config_invalid(websocket_client, key, config, error) ->
 
 
 async def test_message_coalescing(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test enabling message coalescing."""
     await websocket_client.send_json(
@@ -2407,7 +2463,9 @@ async def test_message_coalescing(
 
 
 async def test_message_coalescing_not_supported_by_websocket_client(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test enabling message coalescing not supported by websocket client."""
     await websocket_client.send_json({"id": 7, "type": "subscribe_entities"})
@@ -2449,7 +2507,9 @@ async def test_message_coalescing_not_supported_by_websocket_client(
 
 
 async def test_client_message_coalescing(
-    hass: HomeAssistant, websocket_client, hass_admin_user: MockUser
+    hass: HomeAssistant,
+    websocket_client: MockHAClientWebSocket,
+    hass_admin_user: MockUser,
 ) -> None:
     """Test client message coalescing."""
     await websocket_client.send_json(