test: 添加反汇编之后的正则匹配测试
This commit is contained in:
@@ -90,6 +90,23 @@ class TestBackdoorDetection(unittest.TestCase):
|
|||||||
with self.assertRaises(ValueError):
|
with self.assertRaises(ValueError):
|
||||||
detectGPT(content)
|
detectGPT(content)
|
||||||
|
|
||||||
|
def test_find_dangerous_functions_pyc(self):
|
||||||
|
file_content = """import os
|
||||||
|
os.system('rm -rf /')
|
||||||
|
"""
|
||||||
|
file_extension = ".pyc"
|
||||||
|
|
||||||
|
expected_result = {
|
||||||
|
"high": [(2, "os.system('rm -rf /')")],
|
||||||
|
"medium": [],
|
||||||
|
"low": [],
|
||||||
|
"none": [],
|
||||||
|
}
|
||||||
|
|
||||||
|
result = find_dangerous_functions(file_content, file_extension)
|
||||||
|
|
||||||
|
self.assertEqual(result, expected_result)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
Reference in New Issue
Block a user