def test_read_empty(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = '\x00' * buf_size actual = buf.read() self.assertEqual(expected, actual)
def test_write_read_with_size(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) actual = buf.read(size=5) self.assertEqual(expected[:5], actual)
def test_get_pointer_with_negative_size(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) size = -1 self.assertRaises(exceptions.BufferInvalidSize, buf.get_pointer, size=size)
def test_get_pointer_with_negative_offset(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) offset = -1 self.assertRaises(exceptions.BufferInvalidOffset, buf.get_pointer, offset=offset)
def test_write_read(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) actual = buf.read() # Without size read returns all contents, so expect nulls after data expected += '\x00' * (buf_size - len(expected)) self.assertEqual(expected, actual[:len(expected)])
def test_write_read_with_offset(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) offset = 6 actual = buf.read(offset=offset) expected = expected[6:] expected += '\x00' * (buf_size - len(expected) - offset) self.assertEqual(expected, actual)
def test_get_pointer_with_size(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) size = 5 ptr = buf.get_pointer(size=size) actual = ptr.contents.raw self.assertEqual(expected[:5], actual[:len(expected)])
def test_write_read_with_size_and_offset(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) offset = 6 size = 5 actual = buf.read(size=size, offset=offset) expected = expected[offset:offset + size] self.assertEqual(expected, actual)
def test_get_pointer(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) ptr = buf.get_pointer() actual = ptr.contents.raw # Without size read returns all contents, so expect nulls after data expected += '\x00' * (buf_size - len(expected)) self.assertEqual(expected, actual[:len(expected)])
def test_get_pointer_with_offset(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) offset = 6 ptr = buf.get_pointer(offset=offset) actual = ptr.contents.raw expected = expected[6:] expected += '\x00' * (buf_size - len(expected) - offset) self.assertEqual(expected, actual)
def test_get_pointer_with_size_and_offset(self): buf_size = 20 buf = AlignedBuffer(buf_size) expected = "Hello world!" buf.write(expected) offset = 6 size = 5 ptr = buf.get_pointer(size=size, offset=offset) actual = ptr.contents.raw expected = expected[offset:offset + size] self.assertEqual(expected, actual)