def test_write_string(self): sw = StreamWriter() value = 'hello' sw.write_string(value) expected = bytes([0x05, 0x00, 0x00, 0x00, ord('h'), ord('e'), ord('l'), ord('l'), ord('o')]) self.assertEqual(9, sw.get_size()) self.assertEqual(expected, sw.get_stream_value()) sw.write_string(value) expected = bytes([0x05, 0x00, 0x00, 0x00, ord('h'), ord('e'), ord('l'), ord('l'), ord('o'), 0x00, 0x00, 0x00, 0x05, 0x00, 0x00, 0x00, ord('h'), ord('e'), ord('l'), ord('l'), ord('o')]) self.assertEqual(21, sw.get_size()) self.assertEqual(expected, sw.get_stream_value()) sw.close()
def test_write_bytes(self): sw = StreamWriter() value = bytes([0x01, 0x02, 0x03]) sw.write_bytes(value) expected = bytes([0x03, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03]) self.assertEqual(7, sw.get_size()) self.assertEqual(expected, sw.get_stream_value()) sw.write_bytes(value) expected = bytes([0x03, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03, 0x00, 0x03, 0x00, 0x00, 0x00, 0x01, 0x02, 0x03]) self.assertEqual(15, sw.get_size()) self.assertEqual(expected, sw.get_stream_value()) sw.close()
def test_write_uint8(self): sw = StreamWriter() value = 0x17 sw.write_uint8(value) self.assertEqual(1, sw.get_size()) expected = bytes([0x17]) self.assertEqual(expected, sw.get_stream_value()) sw.close()
def test_reset(self): sw = StreamWriter() value = 0x17 sw.write_uint64(value) sw.write_uint64(value) sw.reset() expected = b'' self.assertEqual(expected, sw.get_stream_value()) self.assertEqual(0, sw.get_size()) sw.close()