| """5.2: o `opentars` do terminal como processo de verdade (nos três sistemas): --version, --help, -p, --chat com pedido, o |
| chat interativo por stdin e o diagnóstico, contra um Ollama de mentira. Nenhum pode cair com Traceback nem sair com código |
| errado. Pegaria o terminal quebrado no Windows e no macOS.""" |
| import os |
| import subprocess |
| import sys |
| import tempfile |
| from pathlib import Path |
|
|
| sys.path.insert(0, str(Path(__file__).resolve().parent)) |
| from _util import RAIZ |
|
|
| import fake_ollama |
|
|
| servidor, url = fake_ollama.subir(modelos=["ProjectOpenTARS/Gargantua-2:latest"], resposta="Olá, tudo certo por aqui.") |
| casa = tempfile.mkdtemp(prefix="opentars-cli-") |
| env = dict(os.environ, HOME=casa, USERPROFILE=casa, APPDATA=casa, LOCALAPPDATA=casa, XDG_CONFIG_HOME=casa + "/.config", |
| OLLAMA_HOST=url, TARS_IDIOMA="en", TARS_SEM_AVISO_VERSAO="1", TARS_SEM_ATALHO="1", TARS_DIR_LOG=casa + "/log", |
| TARS_ARQUIVO_CONFIG=casa + "/config.json", TARS_ARQUIVO_SESSOES=casa + "/sessoes.json", TARS_GARGANTUA="completo", |
| TARS_ARQUIVO_AUTOTESTE=casa + "/autoteste.txt", PYTHONUTF8="1", PYTHONIOENCODING="utf-8") |
| env.pop("QT_QPA_PLATFORM", None) |
|
|
|
|
| def rodar(argumentos, entrada=None, ambiente=None, espera=120): |
| r = subprocess.run([sys.executable, str(Path(RAIZ) / "tars.py"), *argumentos], input=entrada, capture_output=True, |
| text=True, encoding="utf-8", errors="replace", timeout=espera, env=ambiente or env, cwd=casa) |
| return r.returncode, r.stdout + r.stderr |
|
|
|
|
| def sem_queda(nome, saida): |
| assert "Traceback" not in saida and "Error:" not in saida.split("\n\n")[0][:0], (nome, saida[-1500:]) |
|
|
|
|
| codigo, saida = rodar(["--version"]) |
| assert codigo == 0 and "5.2" in saida and "Traceback" not in saida, (codigo, saida) |
| codigo, saida = rodar(["--help"]) |
| assert codigo == 0 and "--chat" in saida and "Traceback" not in saida, (codigo, saida[-800:]) |
| print("OK: --version e --help") |
|
|
| codigo, saida = rodar(["-p", "say hello"]) |
| assert codigo == 0 and "Olá, tudo certo por aqui." in saida and "Traceback" not in saida, (codigo, saida[-1500:]) |
| print("OK: -p responde e sai com 0") |
|
|
| codigo, saida = rodar(["--chat", "say hello"]) |
| assert codigo == 0 and "Olá, tudo certo por aqui." in saida and "Traceback" not in saida, (codigo, saida[-1500:]) |
| print("OK: --chat com pedido responde e sai com 0") |
|
|
| codigo, saida = rodar(["--chat"], entrada="say hello\n/sair\n") |
| assert codigo == 0 and "Olá, tudo certo por aqui." in saida and "Traceback" not in saida, (codigo, saida[-1500:]) |
| assert "\x01" not in saida and "\x02" not in saida, "marcas do readline vazaram para a tela (sem readline, no Windows)" |
| print("OK: o chat interativo por stdin responde, sai com /sair e não deixa lixo de readline") |
|
|
| codigo, saida = rodar(["--chat"], entrada="say hello\n/sair\n", ambiente=dict(env, PYTHONUTF8="0", PYTHONIOENCODING="cp1252")) |
| assert codigo == 0 and "Traceback" not in saida, (codigo, saida[-1500:]) |
| print("OK: o chat não cai com a saída em cp1252 (sem UTF-8)") |
|
|
| |
| servidor.roteiro = [{"ferramenta": ("execute_terminal", {"command": "mkdir Projeto"})}, |
| {"ferramenta": ("execute_terminal", {"command": "echo ola > Projeto/nota.txt"})}, |
| {"ferramenta": ("execute_terminal", {"command": "ls Projeto"})}, |
| {"texto": "Pronto: a pasta Projeto está criada."}] |
| codigo, saida = rodar(["-p", "create a folder called Projeto with a note inside"]) |
| assert codigo == 0 and "Traceback" not in saida, (codigo, saida[-2000:]) |
| assert (Path(casa) / "Projeto").is_dir() and (Path(casa) / "Projeto" / "nota.txt").exists(), (os.listdir(casa), saida[-2000:]) |
| servidor.roteiro = None |
| print("OK: o terminal executa ferramentas de verdade (mkdir, echo > arquivo, ls)") |
|
|
| |
| servidor.roteiro = [{"ferramenta": ("execute_terminal", {"command": "mkdir Outra"})}, {"texto": "Feito."}] |
| codigo, saida = rodar(["--chat"], entrada="create Outra\n/sair\n") |
| assert codigo == 0 and "Traceback" not in saida and (Path(casa) / "Outra").is_dir(), (codigo, os.listdir(casa), saida[-2000:]) |
| servidor.roteiro = None |
| print("OK: o chat interativo executa ferramentas") |
|
|
| codigo, saida = rodar(["--diagnostico"]) |
| assert "Traceback" not in saida, (codigo, saida[-1500:]) |
| print("OK: --diagnostico não cai") |
| servidor.shutdown() |
| print("test_cli_portatil: TUDO OK.") |
|
|
