sangge/BackDoorBuster
sangge/BackDoorBuster
1
0
Fork 0
Code Issues Pull Requests Projects 1 Activity

fix: fix some error

Browse Source
This commit is contained in:
sangge-redmi 2024-06-03 21:17:42 +08:00
parent 4a55822a8f
commit 4f5c67b32e
2 changed files with 18 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/final_tests_util.py
View File

@ -1,5 +1,5 @@
from typing import Tuple from typing import Tuple
from git import Repo from git import Repo # type: ignore
import random import random
from pathlib import Path from pathlib import Path

20
tests/test_final_tests.py
View File

@ -1,9 +1,8 @@
import unittest import unittest
import os
import shutil import shutil
from detection.utils import read_file_content from detection.utils import read_file_content
from .final_tests_util import * from .final_tests_util import clone_repo, Path, inject_random_backdoor
from detection.Regexdetection import find_dangerous_functions from detection.Regexdetection import find_dangerous_functions
from detection.GPTdetection import detectGPT from detection.GPTdetection import detectGPT
@ -27,7 +26,11 @@ class TestFinalTests(unittest.TestCase):
for file in self.all_python_files: for file in self.all_python_files:
content = read_file_content(str(file)) content = read_file_content(str(file))
results = find_dangerous_functions(content, ".py") results = find_dangerous_functions(content, ".py")
if len(results["high"]) > 0 or len(results["medium"]) > 0 or len(results["low"]) > 0: if (
len(results["high"]) > 0
or len(results["medium"]) > 0
or len(results["low"]) > 0
):
detectedNum += 1 detectedNum += 1
possibly_dangerous_file.append(file) possibly_dangerous_file.append(file)
print(detectedNum / self.filesNum) print(detectedNum / self.filesNum)
@ -39,14 +42,17 @@ class TestFinalTests(unittest.TestCase):
results = {} results = {}
try: try:
results = detectGPT(content) results = detectGPT(content)
except Exception as e: if (
print(e) len(results["high"]) > 0
if len(results["high"]) > 0 or len(results["medium"]) > 0 or len(results["low"]) > 0: or len(results["medium"]) > 0
or len(results["low"]) > 0
):
GPTdetectedNum += 1 GPTdetectedNum += 1
print(GPTdetectedNum) print(GPTdetectedNum)
self.assertGreaterEqual(GPTdetectedNum, detectedNum) self.assertGreaterEqual(GPTdetectedNum, detectedNum)
except Exception as e:
print(e)
if __name__ == "__main__": if __name__ == "__main__":