import unittest

from pipecat.pipeline.frames import AudioFrame, TextFrame, TranscriptionFrame
from pipecat.serializers.protobuf_serializer import ProtobufFrameSerializer


class TestProtobufFrameSerializer(unittest.IsolatedAsyncioTestCase):
    def setUp(self):
        self.serializer = ProtobufFrameSerializer()

    async def test_roundtrip(self):
        text_frame = TextFrame(text='hello world')
        frame = self.serializer.deserialize(
            self.serializer.serialize(text_frame))
        self.assertEqual(frame, TextFrame(text='hello world'))

        transcription_frame = TranscriptionFrame(
            text="Hello there!", participantId="123", timestamp="2021-01-01")
        frame = self.serializer.deserialize(
            self.serializer.serialize(transcription_frame))
        self.assertEqual(frame, transcription_frame)

        audio_frame = AudioFrame(data=b'1234567890')
        frame = self.serializer.deserialize(
            self.serializer.serialize(audio_frame))
        self.assertEqual(frame, audio_frame)


if __name__ == "__main__":
    unittest.main()