Tacit Lab
536425e8ea
- Add with_spinner context manager with cyan Braille animation for human mode. - Wrap all query/execution commands in cli.py with loading spinners. - Integrate shtab: auto-install shell completions during init for zsh/bash. - Add `completion` subcommand for manual script generation. - Fix stale output_format default in DEFAULT_CONFIG (json → tui). - Add help descriptions to all second-level subcommands. - Version 2.0.4. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
51 lines
2.0 KiB
Python
51 lines
2.0 KiB
Python
"""CLI tests for CoinHunter V2."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import io
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from coinhunter import cli
|
|
|
|
|
|
class CLITestCase(unittest.TestCase):
|
|
def test_help_includes_v2_commands(self):
|
|
parser = cli.build_parser()
|
|
help_text = parser.format_help()
|
|
self.assertIn("init", help_text)
|
|
self.assertIn("account", help_text)
|
|
self.assertIn("opportunity", help_text)
|
|
|
|
def test_init_dispatches(self):
|
|
captured = {}
|
|
with patch.object(cli, "ensure_init_files", return_value={"force": True, "root": "/tmp/ch"}), patch.object(
|
|
cli, "install_shell_completion", return_value={"shell": "zsh", "installed": True, "path": "/tmp/ch/_coinhunter"}
|
|
), patch.object(
|
|
cli, "print_output", side_effect=lambda payload, **kwargs: captured.setdefault("payload", payload)
|
|
):
|
|
result = cli.main(["init", "--force"])
|
|
self.assertEqual(result, 0)
|
|
self.assertTrue(captured["payload"]["force"])
|
|
self.assertIn("completion", captured["payload"])
|
|
|
|
def test_old_command_is_rejected(self):
|
|
with self.assertRaises(SystemExit):
|
|
cli.main(["exec", "bal"])
|
|
|
|
def test_runtime_error_is_rendered_cleanly(self):
|
|
stderr = io.StringIO()
|
|
with patch.object(cli, "load_config", side_effect=RuntimeError("boom")), patch("sys.stderr", stderr):
|
|
result = cli.main(["market", "tickers", "BTCUSDT"])
|
|
self.assertEqual(result, 1)
|
|
self.assertIn("error: boom", stderr.getvalue())
|
|
|
|
def test_upgrade_dispatches(self):
|
|
captured = {}
|
|
with patch.object(cli, "self_upgrade", return_value={"command": "pipx upgrade coinhunter", "returncode": 0}), patch.object(
|
|
cli, "print_output", side_effect=lambda payload, **kwargs: captured.setdefault("payload", payload)
|
|
):
|
|
result = cli.main(["upgrade"])
|
|
self.assertEqual(result, 0)
|
|
self.assertEqual(captured["payload"]["returncode"], 0)
|