mirror of
https://github.com/androguard/androguard.git
synced 2024-11-23 05:00:11 +00:00
99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
# -*- coding: utf8- -*-
|
|
import os
|
|
import tempfile
|
|
import unittest
|
|
|
|
from androguard.misc import clean_file_name
|
|
|
|
|
|
class MiscTest(unittest.TestCase):
|
|
def testCleanPath(self):
|
|
self.assertEqual(
|
|
"foobarfoo_", clean_file_name("foobarfoo ", unique=False)
|
|
)
|
|
self.assertEqual(
|
|
"foobarsdf_", clean_file_name("foobarsdf.", unique=False)
|
|
)
|
|
self.assertEqual("_init_", clean_file_name("<init>", unique=False))
|
|
if os.name == "nt":
|
|
self.assertEqual(
|
|
"C:\\" + "a" * 230,
|
|
clean_file_name("C:\\" + "a" * 999, unique=False),
|
|
)
|
|
self.assertEqual(
|
|
"C:\\" + "a" * 226 + ".foo",
|
|
clean_file_name("C:\\" + "a" * 999 + ".foo", unique=False),
|
|
)
|
|
else:
|
|
self.assertEqual(
|
|
"/some/path/" + "a" * 230,
|
|
clean_file_name("/some/path/" + "a" * 999, unique=False),
|
|
)
|
|
self.assertEqual(
|
|
"/some/path/" + "a" * 226 + ".foo",
|
|
clean_file_name(
|
|
"/some/path/" + "a" * 999 + ".foo", unique=False
|
|
),
|
|
)
|
|
|
|
with tempfile.NamedTemporaryFile() as fp:
|
|
self.assertEqual(
|
|
fp.name + "_0", clean_file_name(fp.name, unique=True)
|
|
)
|
|
|
|
def testClassNameFormatting(self):
|
|
from androguard.core.bytecode import get_package_class_name
|
|
|
|
self.assertEqual(
|
|
get_package_class_name('Ljava/lang/Object;'),
|
|
('java.lang', 'Object'),
|
|
)
|
|
self.assertEqual(
|
|
get_package_class_name('[Ljava/lang/Object;'),
|
|
('java.lang', 'Object'),
|
|
)
|
|
self.assertEqual(
|
|
get_package_class_name('[[Ljava/lang/Object;'),
|
|
('java.lang', 'Object'),
|
|
)
|
|
self.assertEqual(
|
|
get_package_class_name(
|
|
'[[[[[[[[[[[[[[[[[[[[[[[Ljava/lang/Object;'
|
|
),
|
|
('java.lang', 'Object'),
|
|
)
|
|
self.assertEqual(
|
|
get_package_class_name('[[[[[[[[[[[[[[[[[[[[[[[LObject;'),
|
|
('', 'Object'),
|
|
)
|
|
self.assertEqual(get_package_class_name('LFoobar;'), ('', 'Foobar'))
|
|
self.assertEqual(
|
|
get_package_class_name(
|
|
'Lsdflkjdsklfjsdkjfklsdjfkljsdkflsd/shdfjksdhkjfhsdkjfsh;'
|
|
),
|
|
('sdflkjdsklfjsdkjfklsdjfkljsdkflsd', 'shdfjksdhkjfhsdkjfsh'),
|
|
)
|
|
self.assertEqual(get_package_class_name('L;'), ('', ''))
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('Foobar')
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('java.lang.Object')
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('LOLjava.lang.Object')
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('[[LOLjava.lang.Object')
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('java.lang.Object;')
|
|
|
|
with self.assertRaises(ValueError):
|
|
get_package_class_name('[java.lang.Object;')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|