109 lines
3.0 KiB
Python
109 lines
3.0 KiB
Python
import asyncio
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from wyoming.asr import Transcript
|
|
from wyoming.audio import AudioChunk
|
|
from wyoming.client import AsyncClient
|
|
from wyoming.event import Event
|
|
from wyoming.satellite import RunSatellite
|
|
from wyoming.wake import Detection
|
|
|
|
from wyoming_satellite import (
|
|
EventSettings,
|
|
MicSettings,
|
|
SatelliteSettings,
|
|
WakeSettings,
|
|
WakeStreamingSatellite,
|
|
)
|
|
|
|
from .shared import AUDIO_CHUNK
|
|
|
|
_LOGGER = logging.getLogger()
|
|
|
|
|
|
class MicClient(AsyncClient):
|
|
async def read_event(self) -> Optional[Event]:
|
|
# Send 30ms of audio every 30ms
|
|
await asyncio.sleep(AUDIO_CHUNK.seconds)
|
|
return AUDIO_CHUNK.event()
|
|
|
|
async def write_event(self, event: Event) -> None:
|
|
# Output only
|
|
pass
|
|
|
|
|
|
class WakeClient(AsyncClient):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self._detection_event = asyncio.Event()
|
|
|
|
async def read_event(self) -> Optional[Event]:
|
|
await self._detection_event.wait()
|
|
return Detection().event()
|
|
|
|
async def write_event(self, event: Event) -> None:
|
|
if AudioChunk.is_type(event.type):
|
|
self._detection_event.set()
|
|
|
|
|
|
class EventClient(AsyncClient):
|
|
def __init__(self) -> None:
|
|
super().__init__()
|
|
self.wake_event = asyncio.Event()
|
|
|
|
async def read_event(self) -> Optional[Event]:
|
|
# Input only
|
|
return None
|
|
|
|
async def write_event(self, event: Event) -> None:
|
|
if Detection.is_type(event.type):
|
|
self.wake_event.set()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_multiple_wakeups(tmp_path: Path) -> None:
|
|
mic_client = MicClient()
|
|
wake_client = WakeClient()
|
|
event_client = EventClient()
|
|
|
|
with patch(
|
|
"wyoming_satellite.satellite.SatelliteBase._make_mic_client",
|
|
return_value=mic_client,
|
|
), patch(
|
|
"wyoming_satellite.satellite.SatelliteBase._make_wake_client",
|
|
return_value=wake_client,
|
|
), patch(
|
|
"wyoming_satellite.satellite.SatelliteBase._make_event_client",
|
|
return_value=event_client,
|
|
):
|
|
satellite = WakeStreamingSatellite(
|
|
SatelliteSettings(
|
|
mic=MicSettings(uri="test"),
|
|
wake=WakeSettings(uri="test"),
|
|
event=EventSettings(uri="test"),
|
|
)
|
|
)
|
|
|
|
# Fake server connection
|
|
satellite.server_id = "test"
|
|
|
|
satellite_task = asyncio.create_task(satellite.run(), name="satellite")
|
|
await satellite.event_from_server(RunSatellite().event())
|
|
|
|
await asyncio.wait_for(event_client.wake_event.wait(), timeout=1)
|
|
event_client.wake_event.clear()
|
|
|
|
# Stop streaming
|
|
await satellite.event_from_server(Transcript("test").event())
|
|
|
|
# Should not trigger again within refractory period (default: 5 sec)
|
|
# with pytest.raises(asyncio.TimeoutError):
|
|
# await asyncio.wait_for(event_client.wake_event.wait(), timeout=0.15)
|
|
|
|
await satellite.stop()
|
|
await satellite_task
|