def test_read_query_name(self): packet = Packet() packet.buffer = bytearray( '\x06google\x03com\x00'.encode('UTF-8')) + bytearray([0xC0, 0x07]) assert packet.read_query_name() == "google.com" assert packet.position == 12 assert packet.read_query_name() == "com" assert packet.position == 14
def test_read(self): packet = Packet() assert packet.read_byte() == 0b0 packet.seek(0) packet.buffer = b'\xC0\x0C\x00\x05' assert len(packet.buffer) == 4 assert packet.read_byte() == 0b11000000 assert packet.position == 1 packet.seek(0) assert packet.read_uint16() == 49164 assert packet.position == 2 packet.seek(0) assert packet.read_uint32() == 3222011909 assert packet.position == 4 packet.seek(0) with pytest.raises(IndexError): packet.read_uint32() packet.read_uint32()