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)

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).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