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

refactor: fix some things shown by newer VS syntax checker, mostly type related

Browse Source
This commit is contained in:
Dobin Rutishauser
2025-01-25 18:07:24 +01:00
parent 1d3324686f
commit 919aca0d70
11 changed files with 73 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_superpe.py
+5 -3
View File
@@ -29,7 +29,8 @@ class SuperPeTest(unittest.TestCase):
self.assertEqual(code_sect.Misc_VirtualSize, 0x11B0CE)
# Text Section 2 (PeSection)
code_pesect: PeSection = superpe.get_section_by_name(".text")
code_pesect = superpe.get_section_by_name(".text")
self.assertIsNotNone(code_pesect)
self.assertEqual(code_pesect.name, ".text")
self.assertEqual(code_pesect.virt_addr, 0x1000)
self.assertEqual(code_pesect.virt_size, 0x11B0CE)
@@ -43,7 +44,7 @@ class SuperPeTest(unittest.TestCase):
self.assertEqual(base_reloc.offset, 0x618)
# IAT
iat_entries: Dict[str, IatEntry] = superpe.get_iat_entries()
iat_entries: Dict[str, List[IatEntry]] = superpe.get_iat_entries()
self.assertEqual(len(iat_entries), 24)
self.assertTrue("kernel32.dll" in iat_entries)
self.assertTrue("uxtheme.dll" in iat_entries)
@@ -89,6 +90,7 @@ class SuperPeTest(unittest.TestCase):
# Text Section 2 (PeSection)
code_pesect: PeSection = superpe.get_section_by_name(".text")
self.assertIsNotNone(code_pesect)
self.assertEqual(code_pesect.name, ".text")
self.assertEqual(code_pesect.virt_addr, 0x1000)
self.assertEqual(code_pesect.virt_size, 0x12D08)
@@ -102,7 +104,7 @@ class SuperPeTest(unittest.TestCase):
self.assertEqual(base_reloc.offset, 0xCE8)
# IAT
iat_entries: Dict[str, IatEntry] = superpe.get_iat_entries()
iat_entries: Dict[str, List[IatEntry]] = superpe.get_iat_entries()
self.assertEqual(len(iat_entries), 2)
self.assertTrue("kernel32.dll" in iat_entries)
self.assertTrue("msvcrt.dll" in iat_entries)