blob: b6aad8f00d0f76ac633ed34cb3932fd7fb8a09a5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/python3-dbg
# -*- coding: utf-8 -*-
# Tests minimalistes pour valider la gestion des erreurs relevées.
from chrysacase import ChrysalideTestCase
from pychrysalide.arch import vmpa, mrange
from pychrysalide.format import BinFormat
from pychrysalide.format import BinSymbol
import os
import sys
class SimpleFormat(BinFormat):
pass
class TestFormatErrors(ChrysalideTestCase):
"""TestCase for format.BinFormat."""
def create_fake_symbol(self, index):
saddr = vmpa(index * 0x10, vmpa.VMPA_NO_VIRTUAL)
srange = mrange(saddr, 0x3)
symbol = BinSymbol(BinSymbol.STP_ENTRY_POINT, srange)
return symbol
def testBasicSymbolOperations(self):
"""Deal with the basic operations related to symbols in a binary format."""
sf = SimpleFormat()
self.assertTrue(len(list(sf.symbols)) == 0)
symbols = [ self.create_fake_symbol(i) for i in range(4) ]
s0, s1, s2, s3 = symbols
for s in symbols:
sf.add_symbol(s)
self.assertTrue(len(list(sf.symbols)) == len(symbols))
sf.remove_symbol(s2)
self.assertTrue(list(sf.symbols) == [s0, s1, s3])
def testBadParamsForAdding(self):
"""Check if bad parameters fail for adding a new symbol."""
sf = SimpleFormat()
with self.assertRaises(TypeError):
sf.add_symbol('s')
def testWrongRemoval(self):
"""Try to remove a wrong symbol from a format."""
sf = SimpleFormat()
s23 = self.create_fake_symbol(23)
sf.remove_symbol(s23)
|
