From bd13106fa7819e53c1605c21a7558717646f8df8 Mon Sep 17 00:00:00 2001
From: DavidKorczynski <david@adalogics.com>
Date: Mon, 21 Oct 2024 16:27:44 +0100
Subject: [PATCH] webapp: api: add more blockers to test-case paths (#1790)

Signed-off-by: David Korczynski <david@adalogics.com>
---
 .../assets/db/must_include_small.config       |  3 ++
 .../app/webapp/routes.py                      | 29 ++++++++++++++-----
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/tools/web-fuzzing-introspection/app/static/assets/db/must_include_small.config b/tools/web-fuzzing-introspection/app/static/assets/db/must_include_small.config
index 76b510d1c..6c482ad01 100644
--- a/tools/web-fuzzing-introspection/app/static/assets/db/must_include_small.config
+++ b/tools/web-fuzzing-introspection/app/static/assets/db/must_include_small.config
@@ -10,3 +10,6 @@ json-java
 java-diff-utils
 idna
 configparser
+tinyxml2
+libheif
+
diff --git a/tools/web-fuzzing-introspection/app/webapp/routes.py b/tools/web-fuzzing-introspection/app/webapp/routes.py
index d84e37b4e..e0f28ca03 100644
--- a/tools/web-fuzzing-introspection/app/webapp/routes.py
+++ b/tools/web-fuzzing-introspection/app/webapp/routes.py
@@ -2140,8 +2140,22 @@ def all_project_header_files():
     return {'result': 'failed', 'msg': 'did not find project'}
 
 
+def should_ignore_testpath(test_path: str) -> bool:
+    """Returns true if the test path should be ignored"""
+    banned_paths = {
+        '/src/fuzztest/', '/src/libfuzzer/', '/src/aflplusplus/', '/LPM/',
+        '/googletest/', '/third_party/', '/thirdparty/', 'fuzz', '/depends/',
+        '/external/', '/build/', '/src/inspector/', '/source-code/',
+        '/workspace/'
+    }
+    if [bp for bp in banned_paths if bp in test_path]:
+        return True
+    return False
+
+
 def extract_project_tests(project_name,
                           refine: bool = True) -> Optional[List[str]]:
+    """Extracts the tests in terms of file paths of a given project"""
     tests_file = os.path.join(
         os.path.dirname(__file__),
         f"../static/assets/db/db-projects/{project_name}/test_files.json")
@@ -2151,15 +2165,10 @@ def extract_project_tests(project_name,
     with open(tests_file, 'r') as f:
         tests_file_list = json.load(f)
 
-    banned_paths = {
-        '/src/fuzztest/', '/src/libfuzzer/', '/src/aflplusplus/', '/LPM/',
-        '/googletest/', '/third_party/', '/thirdparty/', 'fuzz', '/depends/',
-        '/external/', '/build/', '/src/inspector/', '/source-code/'
-    }
     if refine:
         refined_list = []
         for test_file in tests_file_list:
-            if [bp for bp in banned_paths if bp in test_file]:
+            if should_ignore_testpath(test_file):
                 continue
             refined_list.append(test_file)
         tests_file_list = refined_list
@@ -2167,6 +2176,7 @@ def extract_project_tests(project_name,
 
 
 def _light_project_tests(project_name):
+    """Returns the list of test files in a project based on FI light."""
     target_project = get_project_with_name(project_name)
     if not target_project:
         return []
@@ -2175,7 +2185,12 @@ def _light_project_tests(project_name):
         return []
 
     light_test_files = target_project.light_analysis.get('test-files', [])
-    return light_test_files
+    returner_list = []
+    for test_file in light_test_files:
+        if should_ignore_testpath(test_file):
+            continue
+        returner_list.append(test_file)
+    return returner_list
 
 
 @blueprint.route('/api/project-tests')