evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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

Browse source
This commit is contained in:
epenet 2024-06-27 19:54:44 +02:00 • committed by GitHub
parent eb2d2ce1b3
commit f4b76406f2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 34 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

32
tests/scripts/test_auth.py
View file

@ -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