Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Add typing to tests with single hass argument (2) (#87675)

Browse Source 
* Add typing to tests with single hass argument (2)

* a few more
This commit is contained in:
epenet
2023-02-08 08:51:43 +01:00
committed by GitHub
parent 1bbc03d0ba
commit c98b4e3204
50 changed files with 816 additions and 671 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
tests/helpers/test_httpx_client.py
View File

@@ -5,25 +5,27 @@ from unittest.mock import Mock, patch
import httpx
import pytest
from homeassistant.core import EVENT_HOMEASSISTANT_CLOSE
from homeassistant.core import EVENT_HOMEASSISTANT_CLOSE, HomeAssistant
import homeassistant.helpers.httpx_client as client
async def test_get_async_client_with_ssl(hass):
async def test_get_async_client_with_ssl(hass: HomeAssistant) -> None:
"""Test init async client with ssl."""
client.get_async_client(hass)
assert isinstance(hass.data[client.DATA_ASYNC_CLIENT], httpx.AsyncClient)
async def test_get_async_client_without_ssl(hass):
async def test_get_async_client_without_ssl(hass: HomeAssistant) -> None:
"""Test init async client without ssl."""
client.get_async_client(hass, verify_ssl=False)
assert isinstance(hass.data[client.DATA_ASYNC_CLIENT_NOVERIFY], httpx.AsyncClient)
async def test_create_async_httpx_client_with_ssl_and_cookies(hass):
async def test_create_async_httpx_client_with_ssl_and_cookies(
hass: HomeAssistant,
) -> None:
"""Test init async client with ssl and cookies."""
client.get_async_client(hass)
@@ -32,7 +34,9 @@ async def test_create_async_httpx_client_with_ssl_and_cookies(hass):
assert hass.data[client.DATA_ASYNC_CLIENT] != httpx_client
async def test_create_async_httpx_client_without_ssl_and_cookies(hass):
async def test_create_async_httpx_client_without_ssl_and_cookies(
hass: HomeAssistant,
) -> None:
"""Test init async client without ssl and cookies."""
client.get_async_client(hass, verify_ssl=False)
@@ -43,7 +47,7 @@ async def test_create_async_httpx_client_without_ssl_and_cookies(hass):
assert hass.data[client.DATA_ASYNC_CLIENT_NOVERIFY] != httpx_client
async def test_get_async_client_cleanup(hass):
async def test_get_async_client_cleanup(hass: HomeAssistant) -> None:
"""Test init async client with ssl."""
client.get_async_client(hass)
@@ -55,7 +59,7 @@ async def test_get_async_client_cleanup(hass):
assert hass.data[client.DATA_ASYNC_CLIENT].is_closed
async def test_get_async_client_cleanup_without_ssl(hass):
async def test_get_async_client_cleanup_without_ssl(hass: HomeAssistant) -> None:
"""Test init async client without ssl."""
client.get_async_client(hass, verify_ssl=False)
@@ -67,7 +71,7 @@ async def test_get_async_client_cleanup_without_ssl(hass):
assert hass.data[client.DATA_ASYNC_CLIENT_NOVERIFY].is_closed
async def test_get_async_client_patched_close(hass):
async def test_get_async_client_patched_close(hass: HomeAssistant) -> None:
"""Test closing the async client does not work."""
with patch("httpx.AsyncClient.aclose") as mock_aclose:
@@ -80,7 +84,7 @@ async def test_get_async_client_patched_close(hass):
assert mock_aclose.call_count == 0
async def test_get_async_client_context_manager(hass):
async def test_get_async_client_context_manager(hass: HomeAssistant) -> None:
"""Test using the async client with a context manager does not close the session."""
with patch("httpx.AsyncClient.aclose") as mock_aclose: