mirror of
https://github.com/esphome/esphome.git
synced 2026-10-22 05:46:12 +00:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
"""Integration test for the socket Ipv4Resolve helper on host."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
|
|
import pytest
|
|
|
|
from .types import APIClientConnectedFactory, RunCompiledFunction
|
|
from .wait_utils import wait_or_fail
|
|
|
|
CHECKS = (
|
|
"Literal resolve",
|
|
"Forget drops address",
|
|
"IPv6 literal rejected",
|
|
"Hostname resolve",
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_socket_ipv4_resolve(
|
|
yaml_config: str,
|
|
run_compiled: RunCompiledFunction,
|
|
api_client_connected: APIClientConnectedFactory,
|
|
) -> None:
|
|
"""Exercise Ipv4Resolve literals, forget, failure, and getaddrinfo on host."""
|
|
test_complete = asyncio.Event()
|
|
results: dict[str, bool] = {}
|
|
|
|
def on_log_line(line: str) -> None:
|
|
if "IPv4 resolve test complete" in line:
|
|
test_complete.set()
|
|
return
|
|
for check in CHECKS:
|
|
if f"{check}:" in line:
|
|
results[check] = "PASSED" in line
|
|
|
|
async with (
|
|
run_compiled(yaml_config, line_callback=on_log_line),
|
|
api_client_connected() as client,
|
|
):
|
|
device_info = await client.device_info()
|
|
assert device_info is not None
|
|
assert device_info.name == "socket-ipv4-resolve-test"
|
|
|
|
await wait_or_fail(test_complete, 10.0, "IPv4 resolve test timed out")
|
|
|
|
for check in CHECKS:
|
|
assert results.get(check), f"{check} check failed or never ran"
|