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_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_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)
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)
