mirror of
https://github.com/esphome/esphome.git
synced 2026-08-29 09:13:28 +00:00
guillempagesJ. Nick Kostonpre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com>J. Nick KostonCo-authored-by: J. Nick Koston <nick@koston.org> Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Co-authored-by: J. Nick Koston <nick@home-assistant.io>
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
|
|
import pytest
|
|
|
|
from .online_image_utils import LEN_BMP_IMAGE, handle_http, make_download_watcher
|
|
from .types import APIClientConnectedFactory, RunCompiledFunction
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_online_image_bmp(
|
|
yaml_config: str,
|
|
run_compiled: RunCompiledFunction,
|
|
api_client_connected: APIClientConnectedFactory,
|
|
) -> None:
|
|
"""Esphome shouldn't block the main loop when a http response is slow"""
|
|
loop = asyncio.get_running_loop()
|
|
|
|
# Track http request
|
|
http_request_future = loop.create_future()
|
|
download_finished_future = loop.create_future()
|
|
downloaded_bytes_future = loop.create_future()
|
|
|
|
check_output = make_download_watcher(
|
|
downloaded_bytes_future, download_finished_future
|
|
)
|
|
|
|
server = await asyncio.start_server(
|
|
handle_http(http_request_future), "127.0.0.1", 0
|
|
)
|
|
http_server_port = server.sockets[0].getsockname()[1]
|
|
|
|
config = yaml_config.replace("HTTP_PORT", str(http_server_port))
|
|
|
|
# Run with log monitoring
|
|
async with (
|
|
server,
|
|
run_compiled(config, line_callback=check_output),
|
|
api_client_connected() as client,
|
|
):
|
|
# Verify device info
|
|
|
|
device_info = await client.device_info()
|
|
assert device_info is not None
|
|
assert device_info.name == "online-image-bmp"
|
|
|
|
# List services to find our test service
|
|
_, services = await client.list_entities_services()
|
|
|
|
# Find test service
|
|
request_service = next((s for s in services if s.name == "fetch_image"), None)
|
|
|
|
assert request_service is not None, "fetch_image service not found"
|
|
|
|
await client.execute_service(request_service, {})
|
|
|
|
async with asyncio.timeout(0.1):
|
|
await http_request_future
|
|
|
|
async with asyncio.timeout(0.5):
|
|
numbytes = await downloaded_bytes_future
|
|
assert numbytes == LEN_BMP_IMAGE
|
|
await download_finished_future
|