Improve query suite test based on feedback

misc/pytest/lib/query-suite-test.py

@@ -1,23 +0,0 @@
-
-import os
-
-def test(codeql, cwd, expected_files, semmle_code_dir, query_suite):
-    actual = codeql.resolve.queries(query_suite, _capture=True).strip()
-    actual = sorted(actual.splitlines())
-    actual = [os.path.relpath(q, semmle_code_dir) for q in actual]
-    actual_file_name = query_suite + '.actual'
-    expected_files.add(actual_file_name)
-    (cwd / actual_file_name).write_text('\n'.join(actual)+'\n')
-
-def test_not_included_queries(codeql, lang_folder_name, cwd, expected_files, semmle_code_dir, query_suites):
-    all_queries = codeql.resolve.queries(semmle_code_dir / 'ql' / lang_folder_name / 'ql' / 'src', _capture=True).strip().splitlines()
-
-    included_in_qls = set()
-    for query_suite in query_suites:
-        included_in_qls |= set(codeql.resolve.queries(query_suite, _capture=True).strip().splitlines())
-
-    not_included = sorted(set(all_queries) - included_in_qls)
-    not_included = [os.path.relpath(q, semmle_code_dir) for q in not_included]
-    not_included_file_name = 'not_included_in_qls.actual'
-    expected_files.add(not_included_file_name)
-    (cwd / not_included_file_name).write_text('\n'.join(not_included)+'\n')

misc/pytest/lib/query_suites.py

@@ -0,0 +1,30 @@
+
+import os
+import pytest
+
+@pytest.fixture
+def check_query_suite(codeql, cwd, expected_files, semmle_code_dir):
+    def ret(query_suite):
+        actual = codeql.resolve.queries(query_suite, _capture=True).strip()
+        actual = sorted(actual.splitlines())
+        actual = [os.path.relpath(q, semmle_code_dir) for q in actual]
+        actual_file_name = query_suite + '.actual'
+        expected_files.add(actual_file_name)
+        (cwd / actual_file_name).write_text('\n'.join(actual)+'\n')
+    return ret
+
+@pytest.fixture
+def check_queries_not_included(codeql, cwd, expected_files, semmle_code_dir):
+    def ret(lang_folder_name, query_suites):
+        all_queries = codeql.resolve.queries(semmle_code_dir / 'ql' / lang_folder_name / 'ql' / 'src', _capture=True).strip().splitlines()
+
+        included_in_qls = set()
+        for query_suite in query_suites:
+            included_in_qls |= set(codeql.resolve.queries(query_suite, _capture=True).strip().splitlines())
+
+        not_included = sorted(set(all_queries) - included_in_qls)
+        not_included = [os.path.relpath(q, semmle_code_dir) for q in not_included]
+        not_included_file_name = 'not_included_in_qls.actual'
+        expected_files.add(not_included_file_name)
+        (cwd / not_included_file_name).write_text('\n'.join(not_included)+'\n')
+    return ret