resourse map never found test

erev0s · Jul 8, 2024 · a7c0099 · a7c0099
1 parent a13e754
commit a7c0099
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tests/res/manifest/AndroidManifest_resource_map_not_found.xml b/tests/res/manifest/AndroidManifest_resource_map_not_found.xml
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -1,3 +1,4 @@
+import glob
 import io
 import os
 import unittest
@@ -6,7 +7,7 @@
 from apkInspector.extract import extract_file_based_on_header_info
 from apkInspector.headers import EndOfCentralDirectoryRecord, CentralDirectory, ZipEntry, LocalHeaderRecord
 from apkInspector.indicators import apk_tampering_check
-from apkInspector.axml import get_manifest
+from apkInspector.axml import get_manifest, parse_apk_for_manifest
 
 
 class ApkInspectorTestCase(unittest.TestCase):
@@ -246,6 +247,14 @@ def test_inconsistencies_in_central_local_entries(self):
             self.assertIn('classes.dex', res['zip tampering']['unique_entries'])
             self.assertIn('classes.deP', res['zip tampering']['unique_entries'])
 
+    def test_manifest_invalid_res_map(self):
+        test_dir = os.path.dirname(os.path.abspath(__file__))
+        with open(os.path.join(test_dir, 'res', 'manifest', 'AndroidManifest_resource_map_not_found.xml'), "rb") as mf:
+            with self.assertRaises(ValueError) as context:
+                get_manifest(io.BytesIO(mf.read()))
+            self.assertEqual(str(context.exception), "Resource Map header was not detected.")
+
+
 
 if __name__ == '__main__':
     unittest.main()