From 8cc77b154715bbe94e0bd93708292551c46adc98 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Mon, 13 Mar 2017 10:07:56 -0700 Subject: Avoid re-encoding for message crc check --- test/test_protocol.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) (limited to 'test/test_protocol.py') diff --git a/test/test_protocol.py b/test/test_protocol.py index aa3dd17..0203614 100644 --- a/test/test_protocol.py +++ b/test/test_protocol.py @@ -67,6 +67,30 @@ def test_decode_message(): assert decoded_message == msg +def test_decode_message_validate_crc(): + encoded = b''.join([ + struct.pack('>i', -1427009701), # CRC + struct.pack('>bb', 0, 0), # Magic, flags + struct.pack('>i', 3), # Length of key + b'key', # key + struct.pack('>i', 4), # Length of value + b'test', # value + ]) + decoded_message = Message.decode(encoded) + assert decoded_message.validate_crc() is True + + encoded = b''.join([ + struct.pack('>i', 1234), # Incorrect CRC + struct.pack('>bb', 0, 0), # Magic, flags + struct.pack('>i', 3), # Length of key + b'key', # key + struct.pack('>i', 4), # Length of value + b'test', # value + ]) + decoded_message = Message.decode(encoded) + assert decoded_message.validate_crc() is False + + def test_encode_message_set(): messages = [ Message(b'v1', key=b'k1'), -- cgit v1.2.1