From d523e96d23747888750309437ef9557462a5030d Mon Sep 17 00:00:00 2001 From: Sergei Pikhovkin Date: Thu, 5 Oct 2023 21:12:39 +0300 Subject: [PATCH] #2 Add Protocol.unpack_many --- omnicomm/tests/test_protocol.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/omnicomm/tests/test_protocol.py b/omnicomm/tests/test_protocol.py index 481358e..729f32c 100644 --- a/omnicomm/tests/test_protocol.py +++ b/omnicomm/tests/test_protocol.py @@ -146,16 +146,18 @@ def test_cmd1_broken_remain(self): def test_cmd2_no_remain(self): value = {'reg_id': 1234567890, 'firmware': 1} data = protocol.RegistrarProtocol.pack(commands.Cmd80(value)) - cmds, remain = protocol.ServerProtocol.unpack_many(data + data) - self.assertTrue(len(cmds) == 2) + original_cmds = [data, data] + cmds, remain = protocol.ServerProtocol.unpack_many(b''.join(original_cmds)) + self.assertTrue(len(cmds) == len(original_cmds)) self.assertTrue(remain == b'') def test_cmd2_broken_remain(self): value = {'reg_id': 1234567890, 'firmware': 1} data = protocol.RegistrarProtocol.pack(commands.Cmd80(value)) + original_cmds = [data, data] original_remain = data[:-1] - cmds, remain = protocol.ServerProtocol.unpack_many(data + data + original_remain) - self.assertTrue(len(cmds) == 2) + cmds, remain = protocol.ServerProtocol.unpack_many(b''.join(original_cmds) + original_remain) + self.assertTrue(len(cmds) == len(original_cmds)) self.assertTrue(remain == original_remain) def test_empty_data_error(self):