admin/SuperMega
1
0
Fork 0
mirror of https://github.com/dobin/SuperMega synced 2026-06-02 17:27:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

tests: fix

Browse Source
This commit is contained in:
Dobin
2024-06-02 13:17:57 +01:00
parent 13320374d9
commit 46293bb60c
5 changed files with 47 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_superpe.py
+4 -4
View File
@@ -2,8 +2,8 @@ from typing import List, Dict
import unittest
import pefile
from pe.superpe import SuperPe, PeSection
from model.defs import *
from pe.superpe import SuperPe, PeSection
class SuperPeTest(unittest.TestCase):
@@ -14,7 +14,7 @@ class SuperPeTest(unittest.TestCase):
# Properties
self.assertFalse(superpe.is_dll())
self.assertTrue(superpe.is_64())
self.assertTrue(superpe.is_64())
self.assertFalse(superpe.is_dotnet())
self.assertEqual(superpe.get_entrypoint(), 0xE1D78)
self.assertIsNone(superpe.get_rwx_section())
@@ -55,8 +55,8 @@ class SuperPeTest(unittest.TestCase):
self.assertEqual(entry.iat_vaddr, 0x14011D528)
self.assertEqual(superpe.get_vaddr_of_iatentry("FileTimeToLocalFileTime"), 0x14011D528)
self.assertEqual(superpe.get_replacement_iat_for(
"kernel32.dll", "GetEnvironmentStringsW"), "FileTimeToLocalFileTime")
self.assertNotEqual(superpe.get_replacement_iat_for(
"kernel32.dll", "GetEnvironmentStringsW"), "GetEnvironmentStringsW")
# Exports
exports = superpe.get_exports_full()