tests: run_test() now uses PipelineTask

tests/test_filters.py

@@ -4,7 +4,6 @@
 # SPDX-License-Identifier: BSD 2-Clause License
 #
 
-import asyncio
 import unittest
 
 from pipecat.frames.frames import (
@@ -19,7 +18,7 @@ from pipecat.processors.filters.frame_filter import FrameFilter
 from pipecat.processors.filters.function_filter import FunctionFilter
 from pipecat.processors.filters.identity_filter import IdentityFilter
 from pipecat.processors.filters.wake_check_filter import WakeCheckFilter
-from pipecat.tests.utils import EndTestFrame, run_test
+from pipecat.tests.utils import run_test
 
 
 class TestIdentifyFilter(unittest.IsolatedAsyncioTestCase):
@@ -27,27 +26,44 @@ class TestIdentifyFilter(unittest.IsolatedAsyncioTestCase):
         filter = IdentityFilter()
         frames_to_send = [UserStartedSpeakingFrame(), UserStoppedSpeakingFrame()]
         expected_returned_frames = [UserStartedSpeakingFrame, UserStoppedSpeakingFrame]
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
 
 class TestFrameFilter(unittest.IsolatedAsyncioTestCase):
     async def test_text_frame(self):
-        filter = FrameFilter(types=(TextFrame, EndTestFrame))
+        filter = FrameFilter(types=(TextFrame,))
         frames_to_send = [TextFrame(text="Hello Pipecat!")]
         expected_returned_frames = [TextFrame]
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
     async def test_end_frame(self):
-        filter = FrameFilter(types=(EndFrame, EndTestFrame))
+        filter = FrameFilter(types=(EndFrame,))
         frames_to_send = [EndFrame()]
         expected_returned_frames = [EndFrame]
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+            send_end_frame=False,
+        )
 
     async def test_system_frame(self):
-        filter = FrameFilter(types=(EndTestFrame,))
+        filter = FrameFilter(types=())
         frames_to_send = [UserStartedSpeakingFrame()]
         expected_returned_frames = [UserStartedSpeakingFrame]
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
 
 class TestFunctionFilter(unittest.IsolatedAsyncioTestCase):
@@ -58,7 +74,11 @@ class TestFunctionFilter(unittest.IsolatedAsyncioTestCase):
         filter = FunctionFilter(filter=passthrough)
         frames_to_send = [TextFrame(text="Hello Pipecat!")]
         expected_returned_frames = [TextFrame]
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
     async def test_no_passthrough(self):
         async def no_passthrough(frame: Frame):
@@ -66,14 +86,12 @@ class TestFunctionFilter(unittest.IsolatedAsyncioTestCase):
 
         filter = FunctionFilter(filter=no_passthrough)
         frames_to_send = [TextFrame(text="Hello Pipecat!")]
-        expected_returned_frames = [TextFrame]
-        try:
-            await asyncio.wait_for(
-                run_test(filter, frames_to_send, expected_returned_frames), timeout=0.5
-            )
-            assert False
-        except asyncio.TimeoutError:
-            pass
+        expected_returned_frames = []
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
 
 class TestWakeCheckFilter(unittest.IsolatedAsyncioTestCase):
@@ -81,7 +99,11 @@ class TestWakeCheckFilter(unittest.IsolatedAsyncioTestCase):
         filter = WakeCheckFilter(wake_phrases=["Hey, Pipecat"])
         frames_to_send = [TranscriptionFrame(user_id="test", text="Phrase 1", timestamp="")]
         expected_returned_frames = []
-        await run_test(filter, frames_to_send, expected_returned_frames)
+        await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
 
     async def test_wake_word(self):
         filter = WakeCheckFilter(wake_phrases=["Hey, Pipecat"])
@@ -90,5 +112,9 @@ class TestWakeCheckFilter(unittest.IsolatedAsyncioTestCase):
             TranscriptionFrame(user_id="test", text="Phrase 1", timestamp=""),
         ]
         expected_returned_frames = [TranscriptionFrame, TranscriptionFrame]
-        (received_down, _) = await run_test(filter, frames_to_send, expected_returned_frames)
+        (received_down, _) = await run_test(
+            filter,
+            frames_to_send=frames_to_send,
+            expected_down_frames=expected_returned_frames,
+        )
         assert received_down[-1].text == "Phrase 1"