From 8d89de22d817dec6a0eac38f6fdce331c9244de2 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 5 Nov 2024 18:12:09 +0530 Subject: [PATCH] Update tests to incorporate the completed boolean in segments Signed-off-by: makaveli10 --- tests/test_client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_client.py b/tests/test_client.py index 836b616..4610ea9 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -68,15 +68,15 @@ class TestClientCallbacks(BaseTestCase): message = json.dumps({ "uid": self.client.uid, "segments": [ - {"start": 0, "end": 1, "text": "Test transcript"}, - {"start": 1, "end": 2, "text": "Test transcript 2"}, - {"start": 2, "end": 3, "text": "Test transcript 3"} + {"start": 0, "end": 1, "text": "Test transcript", "completed": True}, + {"start": 1, "end": 2, "text": "Test transcript 2", "completed": True}, + {"start": 2, "end": 3, "text": "Test transcript 3", "completed": True} ] }) self.client.on_message(self.mock_ws_app, message) # Assert that the transcript was updated correctly - self.assertEqual(len(self.client.transcript), 2) + self.assertEqual(len(self.client.transcript), 3) self.assertEqual(self.client.transcript[1]['text'], "Test transcript 2") def test_on_close(self):