33 lines
943 B
Python
33 lines
943 B
Python
"""Go2rtc tests."""
|
|
|
|
from homeassistant.components.camera import Camera, CameraEntityFeature
|
|
|
|
|
|
class MockCamera(Camera):
|
|
"""Mock Camera Entity."""
|
|
|
|
_attr_name = "Test"
|
|
_attr_supported_features: CameraEntityFeature = CameraEntityFeature.STREAM
|
|
|
|
def __init__(self) -> None:
|
|
"""Initialize the mock entity."""
|
|
super().__init__()
|
|
self._stream_source: str | None = "rtsp://stream"
|
|
|
|
def set_stream_source(self, stream_source: str | None) -> None:
|
|
"""Set the stream source."""
|
|
self._stream_source = stream_source
|
|
|
|
async def stream_source(self) -> str | None:
|
|
"""Return the source of the stream.
|
|
|
|
This is used by cameras with CameraEntityFeature.STREAM
|
|
and StreamType.HLS.
|
|
"""
|
|
return self._stream_source
|
|
|
|
@property
|
|
def use_stream_for_stills(self) -> bool:
|
|
"""Always use the RTSP stream to generate snapshots."""
|
|
return True
|