Add tests module

This commit is contained in:
2026-01-29 21:27:54 +00:00
parent 9fc7224918
commit dc65a6ef71

19
tests/test_cli.py Normal file
View File

@@ -0,0 +1,19 @@
import pytest
from click.testing import CliRunner
from src.cli import main, check, validate
class TestCLI:
def setup_method(self):
self.runner = CliRunner()
def test_main_help(self):
r = self.runner.invoke(main, ["--help"])
assert r.exit_code == 0
def test_check_safe(self):
r = self.runner.invoke(check, ["echo test"])
assert r.exit_code == 0
def test_check_dangerous(self):
r = self.runner.invoke(check, ["rm -rf $TARGET"])
assert r.exit_code == 1