From 342313b76db6ddc4d22e77de5db4d30df2c423d7 Mon Sep 17 00:00:00 2001 From: Stefan Allius Date: Sat, 21 Oct 2023 18:20:32 +0200 Subject: [PATCH] add more test cases --- app/tests/test_messages.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) diff --git a/app/tests/test_messages.py b/app/tests/test_messages.py index 05f5ef1..59dfd9b 100644 --- a/app/tests/test_messages.py +++ b/app/tests/test_messages.py @@ -75,6 +75,7 @@ def test_read_message(MsgContactInfo): assert m.msg_id==0 assert m.header_len==23 assert m.data_len==25 + m.close() def test_read_message_long_id(MsgContactInfo_LongId): @@ -93,6 +94,7 @@ def test_read_message_long_id(MsgContactInfo_LongId): m.read() # try to read rest of message, but there is no chunk available assert m.header_valid # must be valid, since header is complete but not the msg assert m.msg_count == 0 + m.close() def test_read_message_in_chunks(MsgContactInfo): @@ -111,6 +113,7 @@ def test_read_message_in_chunks(MsgContactInfo): m.read() # read rest of message assert not m.header_valid # must be invalid, since msg was handled and buffer flushed assert m.msg_count == 1 + m.close() def test_read_message_in_chunks2(MsgContactInfo): m = MemoryStream(MsgContactInfo, (4,10,0)) @@ -132,6 +135,7 @@ def test_read_message_in_chunks2(MsgContactInfo): pass assert m.msg_count == 1 assert not m.header_valid # must be invalid, since msg was handled and buffer flushed + m.close() def test_read_two_messages(Msg2ContactInfo): m = MemoryStream(Msg2ContactInfo, (0,)) @@ -151,6 +155,7 @@ def test_read_two_messages(Msg2ContactInfo): assert m.msg_id==0 assert m.header_len==23 assert m.data_len==25 + m.close() def test_ctrl_byte(): c = Control(0x91) @@ -161,3 +166,36 @@ def test_ctrl_byte(): assert c.is_resp() +def test_msg_interator(): + m1 = Message() + m2 = Message() + m3 = Message() + m3.close() + del m3 + test1 = 0 + test2 = 0 + for key in Message: + if key == m1: + test1+=1 + elif key == m2: + test2+=1 + else: + assert False + assert test1 == 1 + assert test2 == 1 + +def test_proxy_counter(): + m = Message() + assert m.new_data == {} + assert 'proxy' in m.db.stat + assert 0 == m.db.stat['proxy']['Unknown_Msg'] + + m.inc_counter('Unknown_Msg') + assert m.new_data == {'proxy': True} + assert 1 == m.db.stat['proxy']['Unknown_Msg'] + + m.new_data['proxy'] = False + m.dec_counter('Unknown_Msg') + assert m.new_data == {'proxy': True} + assert 0 == m.db.stat['proxy']['Unknown_Msg'] + m.close()