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

Add type hints to integration tests (part 3) (#87844)

Browse source
This commit is contained in:
epenet 2023-02-10 16:05:01 +01:00 • committed by GitHub
parent 22bfb99db4
commit fa7acb4f0d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
50 changed files with 745 additions and 386 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/components/button/test_init.py
View file

@ -28,7 +28,11 @@ async def test_button(hass: HomeAssistant) -> None:
assert button.press.called
async def test_custom_integration(hass, caplog, enable_custom_integrations):
async def test_custom_integration(
hass: HomeAssistant,
caplog: pytest.LogCaptureFixture,
enable_custom_integrations: None,
) -> None:
"""Test we integration."""
platform = getattr(hass.components, f"test.{DOMAIN}")
platform.init()
@ -51,7 +55,9 @@ async def test_custom_integration(hass, caplog, enable_custom_integrations):
assert "The button has been pressed" in caplog.text
async def test_restore_state(hass, enable_custom_integrations):
async def test_restore_state(
hass: HomeAssistant, enable_custom_integrations: None
) -> None:
"""Test we restore state integration."""
mock_restore_cache(hass, (State("button.button_1", "2021-01-01T23:59:59+00:00"),))