20 lines
526 B
Python
20 lines
526 B
Python
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
|