32 lines
954 B
Python
32 lines
954 B
Python
|
import pathlib
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
|
||
|
|
sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
|
||
|
|
|
||
|
|
from app.main import app, looks_like_shell_command # noqa: E402
|
||
|
|
|
||
|
|
|
||
|
|
def test_root_health():
|
||
|
|
client = TestClient(app)
|
||
|
|
response = client.get('/')
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.json() == {'status': 'gateway ok'}
|
||
|
|
|
||
|
|
|
||
|
|
def test_shell_command_heuristic_detects_commands():
|
||
|
|
assert looks_like_shell_command('ls -la /opt')
|
||
|
|
assert looks_like_shell_command('docker ps')
|
||
|
|
|
||
|
|
|
||
|
|
def test_shell_command_heuristic_rejects_natural_language():
|
||
|
|
assert not looks_like_shell_command('write a python function')
|
||
|
|
assert not looks_like_shell_command('Explain what this does')
|
||
|
|
|
||
|
|
|
||
|
|
def test_shell_command_heuristic_rejects_multiline_and_empty():
|
||
|
|
assert not looks_like_shell_command('')
|
||
|
|
assert not looks_like_shell_command(' ')
|
||
|
|
assert not looks_like_shell_command('ls -la\necho done')
|