def test_no_max_size(self): decoder = PerMessageDeflate( is_server=False, server_no_context_takeover=False, client_no_context_takeover=False, server_max_window_bits=15, client_max_window_bits=15, mem_level=None, ) # 2000 'x' characters compressed compressed_data = b'\xab\xa8\x18\x05\xa3`\x14\x8c\x82Q0\nF\xc1P\x07\x00\xcf@\xa9\xae' decoder.start_decompress_message() data = decoder.decompress_message_data(compressed_data) self.assertEqual(data, b"x" * 2000)
def test_max_size(self): decoder = PerMessageDeflate( is_server=False, server_no_context_takeover=False, client_no_context_takeover=False, server_max_window_bits=15, client_max_window_bits=15, mem_level=8, max_message_size=10, ) # 2000 'x' characters compressed compressed_data = b'\xab\xa8\x18\x05\xa3`\x14\x8c\x82Q0\nF\xc1P\x07\x00\xcf@\xa9\xae' decoder.start_decompress_message() data = decoder.decompress_message_data(compressed_data) # since we set max_message_size, we should only get that # many bytes back. self.assertEqual(data, b"x" * 10)