def test_get_program_header(self): data = ByteArray() data.memsz = 0 seg = DataElfSegment(data=data) self.assertRaises(Unprepared, seg.get_program_header, "<", 32) seg.prepare(0) self.assertRaises(InvalidArgument, seg.get_program_header, "<", 33) self.assertRaises(InvalidArgument, seg.get_program_header, "ads", 32) ph = seg.get_program_header("<", 32) self.assertEquals(ph.todata(), Elf32ProgramHeader("<").todata())
def test_get_program_header(self): data = ByteArray() data.memsz = 0 seg = DataElfSegment(None, data = data) self.assertRaises(Unprepared, seg.get_program_header, '<', 32) seg.prepare(0) self.assertRaises(InvalidArgument, seg.get_program_header, '<', 33) self.assertRaises(InvalidArgument, seg.get_program_header, 'ads', 32) ph = seg.get_program_header('<', 32) self.assertEquals(ph.todata(), Elf32ProgramHeader('<').todata())