mirror of
https://github.com/home-assistant/core.git
synced 2025-04-23 08:47:57 +00:00
Add additional stream HLS payload tests (#46517)
* Add tests for HLS playlist view details * Add tests for hls playlist payloads * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> * Update tests/components/stream/test_hls.py Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com> Co-authored-by: uvjustin <46082645+uvjustin@users.noreply.github.com>
This commit is contained in:
Allen Porter
GitHub
parent
cfdaadf5d9
commit
3f4828f5e1
@ -1,11 +1,15 @@
|
||||
"""The tests for hls streams."""
|
||||
from datetime import timedelta
|
||||
import io
|
||||
from unittest.mock import patch
|
||||
from urllib.parse import urlparse
|
||||
|
||||
import av
|
||||
import pytest
|
||||
|
||||
from homeassistant.components.stream import create_stream
|
||||
from homeassistant.components.stream.const import MAX_SEGMENTS, NUM_PLAYLIST_SEGMENTS
|
||||
from homeassistant.components.stream.core import Segment
|
||||
from homeassistant.const import HTTP_NOT_FOUND
|
||||
from homeassistant.setup import async_setup_component
|
||||
import homeassistant.util.dt as dt_util
|
||||
@ -13,8 +17,61 @@ import homeassistant.util.dt as dt_util
|
||||
from tests.common import async_fire_time_changed
|
||||
from tests.components.stream.common import generate_h264_video
|
||||
|
||||
STREAM_SOURCE = "some-stream-source"
|
||||
SEQUENCE_BYTES = io.BytesIO(b"some-bytes")
|
||||
DURATION = 10
|
||||
|
||||
async def test_hls_stream(hass, hass_client, stream_worker_sync):
|
||||
|
||||
class HlsClient:
|
||||
"""Test fixture for fetching the hls stream."""
|
||||
|
||||
def __init__(self, http_client, parsed_url):
|
||||
"""Initialize HlsClient."""
|
||||
self.http_client = http_client
|
||||
self.parsed_url = parsed_url
|
||||
|
||||
async def get(self, path=None):
|
||||
"""Fetch the hls stream for the specified path."""
|
||||
url = self.parsed_url.path
|
||||
if path:
|
||||
# Strip off the master playlist suffix and replace with path
|
||||
url = "/".join(self.parsed_url.path.split("/")[:-1]) + path
|
||||
return await self.http_client.get(url)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def hls_stream(hass, hass_client):
|
||||
"""Create test fixture for creating an HLS client for a stream."""
|
||||
|
||||
async def create_client_for_stream(stream):
|
||||
http_client = await hass_client()
|
||||
parsed_url = urlparse(stream.endpoint_url("hls"))
|
||||
return HlsClient(http_client, parsed_url)
|
||||
|
||||
return create_client_for_stream
|
||||
|
||||
|
||||
def playlist_response(sequence, segments):
|
||||
"""Create a an hls playlist response for tests to assert on."""
|
||||
response = [
|
||||
"#EXTM3U",
|
||||
"#EXT-X-VERSION:7",
|
||||
"#EXT-X-TARGETDURATION:10",
|
||||
'#EXT-X-MAP:URI="init.mp4"',
|
||||
f"#EXT-X-MEDIA-SEQUENCE:{sequence}",
|
||||
]
|
||||
for segment in segments:
|
||||
response.extend(
|
||||
[
|
||||
"#EXTINF:10.0000,",
|
||||
f"./segment/{segment}.m4s",
|
||||
]
|
||||
)
|
||||
response.append("")
|
||||
return "\n".join(response)
|
||||
|
||||
|
||||
async def test_hls_stream(hass, hls_stream, stream_worker_sync):
|
||||
"""
|
||||
Test hls stream.
|
||||
|
||||
@ -32,27 +89,22 @@ async def test_hls_stream(hass, hass_client, stream_worker_sync):
|
||||
# Request stream
|
||||
stream.add_provider("hls")
|
||||
stream.start()
|
||||
url = stream.endpoint_url("hls")
|
||||
|
||||
http_client = await hass_client()
|
||||
hls_client = await hls_stream(stream)
|
||||
|
||||
# Fetch playlist
|
||||
parsed_url = urlparse(url)
|
||||
playlist_response = await http_client.get(parsed_url.path)
|
||||
playlist_response = await hls_client.get()
|
||||
assert playlist_response.status == 200
|
||||
|
||||
# Fetch init
|
||||
playlist = await playlist_response.text()
|
||||
playlist_url = "/".join(parsed_url.path.split("/")[:-1])
|
||||
init_url = playlist_url + "/init.mp4"
|
||||
init_response = await http_client.get(init_url)
|
||||
init_response = await hls_client.get("/init.mp4")
|
||||
assert init_response.status == 200
|
||||
|
||||
# Fetch segment
|
||||
playlist = await playlist_response.text()
|
||||
playlist_url = "/".join(parsed_url.path.split("/")[:-1])
|
||||
segment_url = playlist_url + "/" + playlist.splitlines()[-1]
|
||||
segment_response = await http_client.get(segment_url)
|
||||
segment_url = "/" + playlist.splitlines()[-1]
|
||||
segment_response = await hls_client.get(segment_url)
|
||||
assert segment_response.status == 200
|
||||
|
||||
stream_worker_sync.resume()
|
||||
@ -61,7 +113,7 @@ async def test_hls_stream(hass, hass_client, stream_worker_sync):
|
||||
stream.stop()
|
||||
|
||||
# Ensure playlist not accessible after stream ends
|
||||
fail_response = await http_client.get(parsed_url.path)
|
||||
fail_response = await hls_client.get()
|
||||
assert fail_response.status == HTTP_NOT_FOUND
|
||||
|
||||
|
||||
@ -176,3 +228,85 @@ async def test_stream_keepalive(hass):
|
||||
|
||||
# Stop stream, if it hasn't quit already
|
||||
stream.stop()
|
||||
|
||||
|
||||
async def test_hls_playlist_view_no_output(hass, hass_client, hls_stream):
|
||||
"""Test rendering the hls playlist with no output segments."""
|
||||
await async_setup_component(hass, "stream", {"stream": {}})
|
||||
|
||||
stream = create_stream(hass, STREAM_SOURCE)
|
||||
stream.add_provider("hls")
|
||||
|
||||
hls_client = await hls_stream(stream)
|
||||
|
||||
# Fetch playlist
|
||||
resp = await hls_client.get("/playlist.m3u8")
|
||||
assert resp.status == 404
|
||||
|
||||
|
||||
async def test_hls_playlist_view(hass, hls_stream, stream_worker_sync):
|
||||
"""Test rendering the hls playlist with 1 and 2 output segments."""
|
||||
await async_setup_component(hass, "stream", {"stream": {}})
|
||||
|
||||
stream = create_stream(hass, STREAM_SOURCE)
|
||||
stream_worker_sync.pause()
|
||||
hls = stream.add_provider("hls")
|
||||
|
||||
hls.put(Segment(1, SEQUENCE_BYTES, DURATION))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
hls_client = await hls_stream(stream)
|
||||
|
||||
resp = await hls_client.get("/playlist.m3u8")
|
||||
assert resp.status == 200
|
||||
assert await resp.text() == playlist_response(sequence=1, segments=[1])
|
||||
|
||||
hls.put(Segment(2, SEQUENCE_BYTES, DURATION))
|
||||
await hass.async_block_till_done()
|
||||
resp = await hls_client.get("/playlist.m3u8")
|
||||
assert resp.status == 200
|
||||
assert await resp.text() == playlist_response(sequence=1, segments=[1, 2])
|
||||
|
||||
stream_worker_sync.resume()
|
||||
stream.stop()
|
||||
|
||||
|
||||
async def test_hls_max_segments(hass, hls_stream, stream_worker_sync):
|
||||
"""Test rendering the hls playlist with more segments than the segment deque can hold."""
|
||||
await async_setup_component(hass, "stream", {"stream": {}})
|
||||
|
||||
stream = create_stream(hass, STREAM_SOURCE)
|
||||
stream_worker_sync.pause()
|
||||
hls = stream.add_provider("hls")
|
||||
|
||||
hls_client = await hls_stream(stream)
|
||||
|
||||
# Produce enough segments to overfill the output buffer by one
|
||||
for sequence in range(1, MAX_SEGMENTS + 2):
|
||||
hls.put(Segment(sequence, SEQUENCE_BYTES, DURATION))
|
||||
await hass.async_block_till_done()
|
||||
|
||||
resp = await hls_client.get("/playlist.m3u8")
|
||||
assert resp.status == 200
|
||||
|
||||
# Only NUM_PLAYLIST_SEGMENTS are returned in the playlist.
|
||||
start = MAX_SEGMENTS + 2 - NUM_PLAYLIST_SEGMENTS
|
||||
assert await resp.text() == playlist_response(
|
||||
sequence=start, segments=range(start, MAX_SEGMENTS + 2)
|
||||
)
|
||||
|
||||
# Fetch the actual segments with a fake byte payload
|
||||
with patch(
|
||||
"homeassistant.components.stream.hls.get_m4s", return_value=b"fake-payload"
|
||||
):
|
||||
# The segment that fell off the buffer is not accessible
|
||||
segment_response = await hls_client.get("/segment/1.m4s")
|
||||
assert segment_response.status == 404
|
||||
|
||||
# However all segments in the buffer are accessible, even those that were not in the playlist.
|
||||
for sequence in range(2, MAX_SEGMENTS + 2):
|
||||
segment_response = await hls_client.get(f"/segment/{sequence}.m4s")
|
||||
assert segment_response.status == 200
|
||||
|
||||
stream_worker_sync.resume()
|
||||
stream.stop()
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user