Add tests module
This commit is contained in:
19
tests/test_cli.py
Normal file
19
tests/test_cli.py
Normal 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
|
||||
Reference in New Issue
Block a user