pull from main

server/tests/test_basic_rtc.py

@@ -1,63 +0,0 @@
-import pytest
-from unittest.mock import patch
-
-
-@pytest.mark.asyncio
-async def test_basic_rtc_server(aiohttp_server, event_loop):
-    # goal is to start the server, and send rtc audio to it
-    # validate the events received
-    import argparse
-    import json
-    from pathlib import Path
-    from reflector.server import create_app
-    from reflector.stream_client import StreamClient
-    from reflector.models import TitleSummaryOutput
-    from aiortc.contrib.signaling import add_signaling_arguments, create_signaling
-
-    # customize settings to have a mock LLM server
-    with patch("reflector.server.get_title_and_summary") as mock_llm:
-        # any response from mock_llm will be test topic
-        mock_llm.return_value = TitleSummaryOutput(["topic_test"])
-
-        # create the server
-        app = create_app()
-        server = await aiohttp_server(app)
-        url = f"http://{server.host}:{server.port}/offer"
-
-        # create signaling
-        parser = argparse.ArgumentParser()
-        add_signaling_arguments(parser)
-        args = parser.parse_args(["-s", "tcp-socket"])
-        signaling = create_signaling(args)
-
-        # create the client
-        path = Path(__file__).parent / "records" / "test_mathieu_hello.wav"
-        client = StreamClient(signaling, url=url, play_from=path.as_posix())
-        await client.start()
-
-        # we just want the first transcription
-        # and topic update messages
-
-        marks = {
-            "SHOW_TRANSCRIPTION": False,
-            "UPDATE_TOPICS": False,
-        }
-
-        async for rawmsg in client.get_reader():
-            msg = json.loads(rawmsg)
-            cmd = msg["cmd"]
-            if cmd == "SHOW_TRANSCRIPTION":
-                assert "text" in msg
-                assert "want to share my incredible experience" in msg["text"]
-            elif cmd == "UPDATE_TOPICS":
-                assert "topics" in msg
-                assert "topic_test" in msg["topics"]
-            marks[cmd] = True
-
-            # break if we have all the events we need
-            if all(marks.values()):
-                break
-
-        # stop the server
-        await server.close()
-        await client.stop()

server/tests/test_transcripts_rtc_ws.py

@@ -71,11 +71,15 @@ async def dummy_llm():
 
 
 @pytest.mark.asyncio
-async def test_transcript_rtc_and_websocket(dummy_transcript, dummy_llm):
+async def test_transcript_rtc_and_websocket(tmpdir, dummy_transcript, dummy_llm):
     # goal: start the server, exchange RTC, receive websocket events
     # because of that, we need to start the server in a thread
     # to be able to connect with aiortc
 
+    from reflector.settings import settings
+
+    settings.DATA_DIR = Path(tmpdir)
+
     # start server
     host = "127.0.0.1"
     port = 1255
@@ -189,3 +193,13 @@ async def test_transcript_rtc_and_websocket(dummy_transcript, dummy_llm):
     resp = await ac.get(f"/transcripts/{tid}")
     assert resp.status_code == 200
     assert resp.json()["status"] == "ended"
+
+    # check that audio is available
+    resp = await ac.get(f"/transcripts/{tid}/audio")
+    assert resp.status_code == 200
+    assert resp.headers["Content-Type"] == "audio/wav"
+
+    # check that audio/mp3 is available
+    resp = await ac.get(f"/transcripts/{tid}/audio/mp3")
+    assert resp.status_code == 200
+    assert resp.headers["Content-Type"] == "audio/mp3"