jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 05:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix mqtt test DeprecationWarnings (#97734)

Browse Source
This commit is contained in:
Marc Mueller 2023-08-04 09:21:36 +02:00 committed by GitHub
parent 1587ac2137
commit 37885400c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/components/mqtt/test_config_flow.py
View File

@ -1,5 +1,6 @@
"""Test config flow."""
from collections.abc import Generator
from collections.abc import Generator, Iterator
from contextlib import contextmanager
from pathlib import Path
from random import getrandbits
from ssl import SSLError
@ -136,19 +137,22 @@ def mock_process_uploaded_file(tmp_path: Path) -> Generator[MagicMock, None, Non
file_id_cert = str(uuid4())
file_id_key = str(uuid4())
def _mock_process_uploaded_file(hass: HomeAssistant, file_id) -> None:
@contextmanager
def _mock_process_uploaded_file(
hass: HomeAssistant, file_id: str
) -> Iterator[Path | None]:
if file_id == file_id_ca:
with open(tmp_path / "ca.crt", "wb") as cafile:
cafile.write(b"## mock CA certificate file ##")
return tmp_path / "ca.crt"
yield tmp_path / "ca.crt"
elif file_id == file_id_cert:
with open(tmp_path / "client.crt", "wb") as certfile:
certfile.write(b"## mock client certificate file ##")
return tmp_path / "client.crt"
yield tmp_path / "client.crt"
elif file_id == file_id_key:
with open(tmp_path / "client.key", "wb") as keyfile:
keyfile.write(b"## mock key file ##")
return tmp_path / "client.key"
yield tmp_path / "client.key"
else:
pytest.fail(f"Unexpected file_id: {file_id}")