feat: 添加unittest测试

tests/test_backdoor_detection.py

@@ -0,0 +1,55 @@
+import unittest
+from detection.backdoor_detection import find_dangerous_functions
+
+
+class TestBackdoorDetection(unittest.TestCase):
+    def test_high_risk_detection(self):
+        content = """import os
+        os.system('rm -rf /')   # high risk
+        exec('print("Hello")')  # high risk
+        eval('2 + 2')   # high risk
+        """
+        results = find_dangerous_functions(content)
+        self.assertIn((2, "os.system('rm -rf /')"), results["high"])
+        self.assertIn((3, "exec('print(\"Hello\")')"), results["high"])
+        self.assertIn((4, "eval('2 + 2')"), results["high"])
+
+    def test_medium_risk_detection(self):
+        content = """import subprocess
+        subprocess.run(['ls', '-l'])    # medium risk
+        import os
+        os.popen('ls')  # medium risk
+        """
+        results = find_dangerous_functions(content)
+        self.assertIn((2, "subprocess.run(['ls', '-l'])"), results["medium"])
+        self.assertIn((4, "os.popen('ls')"), results["medium"])
+
+    def test_no_risk_detection(self):
+        content = """a = 10
+        b = a + 5
+        print('This should not be detected as risky.')
+        """
+        results = find_dangerous_functions(content)
+        self.assertEqual(len(results["high"]), 0)
+        self.assertEqual(len(results["medium"]), 0)
+        self.assertEqual(len(results["low"]), 0)
+
+    def test_inclusion_of_comments(self):
+        content = """# Just a comment line
+        print('This is a safe line')
+        eval('2 + 2')  # This should be high risk
+        subprocess.run(['echo', 'hello'])  # This should be medium risk
+        """
+        results = find_dangerous_functions(content)
+        self.assertIn(
+            (3, "eval('2 + 2')"),
+            results["high"],
+        )
+        self.assertIn(
+            (4, "subprocess.run(['echo', 'hello'])"),
+            results["medium"],
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()