test: read dependencies from pyproject.toml in test_requirements.py

Author: Molkree

test/test_requirements.py

@@ -5,13 +5,20 @@
 import json
 import re
 import subprocess
+import sys
 import tempfile
 from importlib.metadata import version
 from pathlib import Path
 
+if sys.version_info >= (3, 11):
+    import tomllib
+else:
+    import tomli as tomllib
+
 ROOT_PATH = Path(__file__).parent.parent
 TEMP_DIR = Path(tempfile.mkdtemp(prefix="requirements_scan-"))
 
+PYPROJECT_TOML = str(ROOT_PATH / "pyproject.toml")
 REQ_TXT = str(ROOT_PATH / "requirements.txt")
 REQ_CSV = str(ROOT_PATH / "requirements.csv")
 TRIAGE_JSON = str(ROOT_PATH / "triage.json")
@@ -48,6 +55,15 @@ def get_packages_from_txt(txt_name: str) -> set[str]:
     return txt_package_names
 
 
+def get_packages_from_pyproject_toml(toml_name: str) -> set[str]:
+    toml_package_names: set[str] = set()
+    with open(toml_name, "rb") as toml_file:
+        toml_data = tomllib.load(toml_file)
+    for package in toml_data["project"]["dependencies"]:
+        toml_package_names.add(re.split(">|<|\\[|;|=|\n", package)[0])
+    return toml_package_names
+
+
 def get_out_of_sync_packages(
     csv_packages: set[str], dependency_packages: set[str]
 ) -> tuple[set[str], set[str]]:
@@ -61,7 +77,7 @@ def test_txt_csv_sync() -> None:
     errors: set[str] = set()
 
     csv_packages = get_packages_from_csv(REQ_CSV)
-    dependency_packages = get_packages_from_txt(REQ_TXT)
+    dependency_packages = get_packages_from_pyproject_toml(PYPROJECT_TOML)
     req_new_packages, req_removed_packages = get_out_of_sync_packages(
         csv_packages, dependency_packages
     )