diff options
| author | Cyrille Bagard <nocbos@gmail.com> | 2019-01-04 18:50:07 (GMT) |
|---|---|---|
| committer | Cyrille Bagard <nocbos@gmail.com> | 2019-01-04 18:50:07 (GMT) |
| commit | b9fe3a839e9212b809d64c11bf92b703adb18fb2 (patch) | |
| tree | 86a1c42f305265372bb4d697dad9c99f2ea3715a /tests/analysis/contents/restricted.py | |
| parent | 0b2cada60941d109bde2f54a198f4c3a799e0606 (diff) | |
Updated the test suite for the current API.
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 6484299..08aa968 100644 --- a/tests/analysis/contents/restricted.py +++ b/tests/analysis/contents/restricted.py @@ -7,6 +7,7 @@ from chrysacase import ChrysalideTestCase +from pychrysalide import arch from pychrysalide.analysis.contents import FileContent, RestrictedContent from pychrysalide.arch import vmpa, mrange import tempfile @@ -63,10 +64,10 @@ class TestRestrictedContent(ChrysalideTestCase): val = rcnt.read_u8(start) self.assertEqual(val, 0x16) - val = rcnt.read_u16(start, vmpa.SRE_LITTLE) + val = rcnt.read_u16(start, arch.SRE_LITTLE) self.assertEqual(val, 0x1817) - val = rcnt.read_u32(start, vmpa.SRE_LITTLE) + val = rcnt.read_u32(start, arch.SRE_LITTLE) self.assertEqual(val, 0x24232221) @@ -110,15 +111,15 @@ class TestRestrictedContent(ChrysalideTestCase): self.assertEqual(val, 0x15) start = vmpa(12, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u16(start, vmpa.SRE_LITTLE) + val = rcnt.read_u16(start, arch.SRE_LITTLE) self.assertEqual(val, 0x1615) start = vmpa(12, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u32(start, vmpa.SRE_LITTLE) + val = rcnt.read_u32(start, arch.SRE_LITTLE) self.assertEqual(val, 0x18171615) start = vmpa(12, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u64(start, vmpa.SRE_LITTLE) + val = rcnt.read_u64(start, arch.SRE_LITTLE) self.assertEqual(val, 0x2423222118171615) start = vmpa(23, vmpa.VMPA_NO_VIRTUAL) @@ -126,15 +127,15 @@ class TestRestrictedContent(ChrysalideTestCase): self.assertEqual(val, 0x28) start = vmpa(22, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u16(start, vmpa.SRE_LITTLE) + val = rcnt.read_u16(start, arch.SRE_LITTLE) self.assertEqual(val, 0x2827) start = vmpa(20, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u32(start, vmpa.SRE_LITTLE) + val = rcnt.read_u32(start, arch.SRE_LITTLE) self.assertEqual(val, 0x28272625) start = vmpa(16, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u64(start, vmpa.SRE_LITTLE) + val = rcnt.read_u64(start, arch.SRE_LITTLE) self.assertEqual(val, 0x2827262524232221) @@ -201,9 +202,9 @@ class TestRestrictedContent(ChrysalideTestCase): with self.assertRaisesRegex(Exception, 'Invalid read access.'): start = vmpa(11, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u16(start, vmpa.SRE_LITTLE) + val = rcnt.read_u16(start, arch.SRE_LITTLE) with self.assertRaisesRegex(Exception, 'Invalid read access.'): start = vmpa(23, vmpa.VMPA_NO_VIRTUAL) - val = rcnt.read_u16(start, vmpa.SRE_LITTLE) + val = rcnt.read_u16(start, arch.SRE_LITTLE) |