def test_getmemsz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_memsz) data = ByteArray() data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_memsz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_memsz)
def test_prepare(self): ef = DataElfSegment() ef.prepare(37) self.assertEqual(ef.prepared, True) self.assertEqual(ef.offset, 37) self.assertRaises(InvalidArgument, ef.prepare, 12) ef = DataElfSegment() self.assertRaises(InvalidArgument, ef.prepare, 37, 12) ef = HeaderElfSegment() self.assertRaises(InvalidArgument, ef.prepare, 12) ef.prepare(37, 12) self.assertEqual(ef.get_memsz(), 12) self.assertEqual(ef.offset, 37)