Exemplo n.º 1
0
    def test_remove_section(self):
        seg = SectionedElfSegment()
        section = UnpreparedElfSection()
        self.assertRaises(InvalidArgument, seg.remove_section, section)
        seg.add_section(section)
        self.assertEqual(section in seg.sections, True)
        seg.remove_section(section)
        self.assertEqual(section in seg.sections, False)

        self.assertRaises(InvalidArgument, seg.remove_section, section)

        seg = DataElfSegment()
        data = ByteArray("foo")
        seg.set_data(data)
        self.assertRaises(InvalidArgument, seg.remove_section, None)

        seg = HeaderElfSegment()
        self.assertRaises(InvalidArgument, seg.remove_section, None)
Exemplo n.º 2
0
    def test_remove_section(self):
        seg = SectionedElfSegment()
        section = UnpreparedElfSection()
        self.assertRaises(InvalidArgument, seg.remove_section, section)
        seg.add_section(section)
        self.assertEqual(section in seg.sections, True)
        seg.remove_section(section)
        self.assertEqual(section in seg.sections, False)

        self.assertRaises(InvalidArgument, seg.remove_section, section)

        seg = DataElfSegment()
        data = ByteArray("foo")
        seg.set_data(data)
        self.assertRaises(InvalidArgument, seg.remove_section, None)

        seg = HeaderElfSegment()
        self.assertRaises(InvalidArgument, seg.remove_section, None)