def test_setdata(self): ef = DataElfSegment() data = ByteArray("foo") ef.set_data(data) self.assertEqual(ef.get_file_data(), data) ef = SectionedElfSegment(sections=[]) self.assertRaises(InvalidArgument, ef.set_data, data)
def test_add_section(self): seg = DataElfSegment() data = ByteArray("foo") seg.set_data(data) section = UnpreparedElfSection() self.assertRaises(InvalidArgument, seg.add_section, section) seg = SectionedElfSegment() seg.add_section(section) self.assertEqual(section in seg.sections, True)
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_getfilesz(self): seg = DataElfSegment() self.assertRaises(InvalidArgument, seg.get_filesz) data = ByteArray("1234567890") data.memsz = 10 seg.set_data(data) self.assertEquals(seg.get_filesz(), 10) seg = HeaderElfSegment() self.assertRaises(Unprepared, seg.get_filesz) seg = SectionedElfSegment() self.assertEqual(seg.get_filesz(), 0)