Add capsys to enforce-type-hints plugin (#120653)

tests/scripts/test_auth.py

@@ -6,6 +6,7 @@ from typing import Any
 from unittest.mock import Mock, patch
 
 import pytest
+from typing_extensions import Generator
 
 from homeassistant.auth.providers import homeassistant as hass_auth
 from homeassistant.core import HomeAssistant
@@ -15,7 +16,7 @@ from tests.common import register_auth_provider
 
 
 @pytest.fixture(autouse=True)
-def reset_log_level():
+def reset_log_level() -> Generator[None]:
     """Reset log level after each test case."""
     logger = logging.getLogger("homeassistant.core")
     orig_level = logger.level
@@ -24,7 +25,7 @@ def reset_log_level():
 
 
 @pytest.fixture
-def provider(hass):
+def provider(hass: HomeAssistant) -> hass_auth.HassAuthProvider:
     """Home Assistant auth provider."""
     provider = hass.loop.run_until_complete(
         register_auth_provider(hass, {"type": "homeassistant"})
@@ -33,7 +34,11 @@ def provider(hass):
     return provider
 
 
-async def test_list_user(hass: HomeAssistant, provider, capsys) -> None:
+async def test_list_user(
+    hass: HomeAssistant,
+    provider: hass_auth.HassAuthProvider,
+    capsys: pytest.CaptureFixture[str],
+) -> None:
     """Test we can list users."""
     data = provider.data
     data.add_auth("test-user", "test-pass")
@@ -47,7 +52,10 @@ async def test_list_user(hass: HomeAssistant, provider, capsys) -> None:
 
 
 async def test_add_user(
-    hass: HomeAssistant, provider, capsys, hass_storage: dict[str, Any]
+    hass: HomeAssistant,
+    provider: hass_auth.HassAuthProvider,
+    capsys: pytest.CaptureFixture[str],
+    hass_storage: dict[str, Any],
 ) -> None:
     """Test we can add a user."""
     data = provider.data
@@ -64,7 +72,11 @@ async def test_add_user(
     data.validate_login("paulus", "test-pass")
 
 
-async def test_validate_login(hass: HomeAssistant, provider, capsys) -> None:
+async def test_validate_login(
+    hass: HomeAssistant,
+    provider: hass_auth.HassAuthProvider,
+    capsys: pytest.CaptureFixture[str],
+) -> None:
     """Test we can validate a user login."""
     data = provider.data
     data.add_auth("test-user", "test-pass")
@@ -89,7 +101,10 @@ async def test_validate_login(hass: HomeAssistant, provider, capsys) -> None:
 
 
 async def test_change_password(
-    hass: HomeAssistant, provider, capsys, hass_storage: dict[str, Any]
+    hass: HomeAssistant,
+    provider: hass_auth.HassAuthProvider,
+    capsys: pytest.CaptureFixture[str],
+    hass_storage: dict[str, Any],
 ) -> None:
     """Test we can change a password."""
     data = provider.data
@@ -108,7 +123,10 @@ async def test_change_password(
 
 
 async def test_change_password_invalid_user(
-    hass: HomeAssistant, provider, capsys, hass_storage: dict[str, Any]
+    hass: HomeAssistant,
+    provider: hass_auth.HassAuthProvider,
+    capsys: pytest.CaptureFixture[str],
+    hass_storage: dict[str, Any],
 ) -> None:
     """Test changing password of non-existing user."""
     data = provider.data