blob: 056238f9b11a2fbf3e1c1bba3e8698292333174a (
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
|
#!/usr/bin/python3-dbg
# -*- coding: utf-8 -*-
from chrysacase import ChrysalideTestCase
from pychrysalide.analysis.contents import MemoryContent
from pychrysalide.format import KnownFormat
class TestKnownFormat(ChrysalideTestCase):
"""TestCase for format.KnownFormat."""
def testKnownFormatConstructor(self):
"""Build Load a simple content for a flat format."""
with self.assertRaisesRegex(RuntimeError, 'pychrysalide.format.KnownFormat is an abstract class'):
fmt = KnownFormat()
class MyKnownFormat(KnownFormat):
pass
with self.assertRaisesRegex(TypeError, 'function takes exactly 1 argument .0 given.'):
fmt = MyKnownFormat()
class MyKnownFormat2(KnownFormat):
pass
with self.assertRaisesRegex(TypeError, 'unable to convert the provided argument to binary content'):
fmt = MyKnownFormat2(123)
class MyKnownFormatReady(KnownFormat):
_key = 'rdy'
def __init2__(self, cnt):
super(MyKnownFormatReady, self).__init2__(cnt)
data = b'\x00\x00\x00\xef'
cnt = MemoryContent(data)
fmt = MyKnownFormatReady(cnt)
self.assertIsNotNone(fmt)
self.assertEqual(fmt.key, 'rdy')
|
