34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
from contextlib import redirect_stderr, redirect_stdout
|
|
from io import StringIO
|
|
import json
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
|
|
from aaisa.cli import main
|
|
|
|
|
|
class CliTests(unittest.TestCase):
|
|
def test_clean_scan_returns_zero(self) -> None:
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
Path(directory, "safe.txt").write_text("ordinary content\n", encoding="utf-8")
|
|
output = StringIO()
|
|
with redirect_stdout(output):
|
|
result = main(["scan", directory])
|
|
self.assertEqual(result, 0)
|
|
self.assertEqual(json.loads(output.getvalue())["summary"]["finding_count"], 0)
|
|
|
|
def test_finding_returns_one(self) -> None:
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
Path(directory, "requirements.txt").write_text("requests>=2\n", encoding="utf-8")
|
|
with redirect_stdout(StringIO()):
|
|
self.assertEqual(main(["scan", directory]), 1)
|
|
|
|
def test_missing_path_returns_two(self) -> None:
|
|
with redirect_stderr(StringIO()):
|
|
self.assertEqual(main(["scan", "/definitely/missing/aaisa-path"]), 2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|