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

Add typing to test files for Wallbox (#68635)

Browse source 
Co-authored-by: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
This commit is contained in:
hesselonline 2022-03-25 18:09:49 +01:00 • committed by GitHub
parent 3ef912b7a0
commit 4b22f04505
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 22 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/components/wallbox/__init__.py
View file

@ -23,6 +23,7 @@ from homeassistant.components.wallbox.const import (
DOMAIN,
)
from homeassistant.const import CONF_PASSWORD, CONF_USERNAME
from homeassistant.core import HomeAssistant
from .const import CONF_ERROR, CONF_JWT, CONF_STATUS, CONF_TTL, CONF_USER_ID
@ -71,7 +72,7 @@ entry = MockConfigEntry(
)
async def setup_integration(hass):
async def setup_integration(hass: HomeAssistant) -> None:
"""Test wallbox sensor class setup."""
entry.add_to_hass(hass)
@ -99,7 +100,7 @@ async def setup_integration(hass):
await hass.async_block_till_done()
async def setup_integration_connection_error(hass):
async def setup_integration_connection_error(hass: HomeAssistant) -> None:
"""Test wallbox sensor class setup with a connection error."""
with requests_mock.Mocker() as mock_request:
@ -125,7 +126,7 @@ async def setup_integration_connection_error(hass):
await hass.async_block_till_done()
async def setup_integration_read_only(hass):
async def setup_integration_read_only(hass: HomeAssistant) -> None:
"""Test wallbox sensor class setup for read only."""
with requests_mock.Mocker() as mock_request: