diff options
Diffstat (limited to 'tests/analysis/contents/restricted.py')
| -rw-r--r-- | tests/analysis/contents/restricted.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/tests/analysis/contents/restricted.py b/tests/analysis/contents/restricted.py index 023e600..8c3cb50 100644 --- a/tests/analysis/contents/restricted.py +++ b/tests/analysis/contents/restricted.py @@ -7,6 +7,7 @@ from chrysacase import ChrysalideTestCase +from pychrysalide import SourceEndian from pychrysalide.analysis import BinContent from pychrysalide.analysis.contents import FileContent, RestrictedContent from pychrysalide.arch import vmpa, mrange @@ -64,10 +65,10 @@ class TestRestrictedContent(ChrysalideTestCase): val = rcnt.read_u8(start) self.assertEqual(val, 0x16) - val = rcnt.read_u16(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u16(start, SourceEndian.LITTLE) self.assertEqual(val, 0x1817) - val = rcnt.read_u32(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u32(start, SourceEndian.LITTLE) self.assertEqual(val, 0x24232221) @@ -111,15 +112,15 @@ class TestRestrictedContent(ChrysalideTestCase): self.assertEqual(val, 0x15) start = vmpa(12, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u16(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u16(start, SourceEndian.LITTLE) self.assertEqual(val, 0x1615) start = vmpa(12, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u32(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u32(start, SourceEndian.LITTLE) self.assertEqual(val, 0x18171615) start = vmpa(12, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u64(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u64(start, SourceEndian.LITTLE) self.assertEqual(val, 0x2423222118171615) start = vmpa(23, vmpa.VmpaSpecialValue.NO_VIRTUAL) @@ -127,15 +128,15 @@ class TestRestrictedContent(ChrysalideTestCase): self.assertEqual(val, 0x28) start = vmpa(22, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u16(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u16(start, SourceEndian.LITTLE) self.assertEqual(val, 0x2827) start = vmpa(20, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u32(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u32(start, SourceEndian.LITTLE) self.assertEqual(val, 0x28272625) start = vmpa(16, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u64(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u64(start, SourceEndian.LITTLE) self.assertEqual(val, 0x2827262524232221) @@ -202,12 +203,12 @@ class TestRestrictedContent(ChrysalideTestCase): with self.assertRaisesRegex(Exception, 'Invalid read access.'): start = vmpa(11, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u16(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u16(start, SourceEndian.LITTLE) with self.assertRaisesRegex(Exception, 'Invalid read access.'): start = vmpa(23, vmpa.VmpaSpecialValue.NO_VIRTUAL) - val = rcnt.read_u16(start, BinContent.SourceEndian.LITTLE) + val = rcnt.read_u16(start, SourceEndian.LITTLE) def testDescription(self): |