# SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0

import pytest

import tomllib
import garak._plugins


def test_requirements_txt_pyproject_toml():
    with open("requirements.txt", "r", encoding="utf-8") as req_file:
        reqtxt_reqs = req_file.readlines()
        reqtxt_reqs = list(
            filter(lambda x: not x.startswith("#"), map(str.strip, reqtxt_reqs))
        )
        reqtxt_reqs.sort()
    with open("pyproject.toml", "rb") as pyproject_file:
        pyproject_toml = tomllib.load(pyproject_file)
        pyproject_reqs = pyproject_toml["project"]["dependencies"]
        for test_deps in pyproject_toml["project"]["optional-dependencies"].values():
            for dep in test_deps:
                pyproject_reqs.append(dep)
        pyproject_reqs.sort()  # for easier output debugging
    # assert len(reqtxt_reqs) == len(pyproject_reqs) # same number of requirements
    spurious_req = set(reqtxt_reqs) - set(pyproject_reqs)
    assert (
        spurious_req == set()
    ), f"spurious items in requirements.txt, {spurious_req}"  # things in reqtxt but not in pyproject
    spurious_pyproject = set(pyproject_reqs) - set(reqtxt_reqs)
    assert (
        spurious_pyproject == set()
    ), f"spurious items in pyproject.toml, {spurious_pyproject}"  # things in pyproject but not in reqtxt


PLUGIN_TYPES = garak._plugins.PLUGIN_TYPES