Roland Thomas
8db7a9e6dc
- extend OptionsManager to support multi-namespace option resolution and toggling - integrate OptionsManager more deeply across Action, ChainedAction, and ActionGroup - propagate shared runtime configuration through execution layers - refine action composition model (sequential + parallel execution semantics) - improve lifecycle consistency across BaseAction, Action, ChainedAction, and ActionGroup - begin aligning execution flow with centralized context and options handling wip: routing and root option parsing behavior still in progress
256 lines
6.3 KiB
Python
256 lines
6.3 KiB
Python
import asyncio
|
|
import sys
|
|
|
|
import pytest
|
|
from rich.text import Text
|
|
|
|
from falyx import Falyx
|
|
from falyx.console import console as falyx_console
|
|
from falyx.exceptions import FalyxError
|
|
from falyx.parser import ParseResult
|
|
from falyx.signals import BackSignal, CancelSignal, FlowSignal, HelpSignal, QuitSignal
|
|
|
|
|
|
async def throw_error_action(error: str):
|
|
if error == "QuitSignal":
|
|
raise QuitSignal("Quit signal triggered.")
|
|
elif error == "BackSignal":
|
|
raise BackSignal("Back signal triggered.")
|
|
elif error == "CancelSignal":
|
|
raise CancelSignal("Cancel signal triggered.")
|
|
elif error == "ValueError":
|
|
raise ValueError("This is a ValueError.")
|
|
elif error == "HelpSignal":
|
|
raise HelpSignal("Help signal triggered.")
|
|
elif error == "FalyxError":
|
|
raise FalyxError("This is a FalyxError.")
|
|
elif error == "FlowSignal":
|
|
raise FlowSignal("Flow signal triggered.")
|
|
else:
|
|
raise asyncio.CancelledError("An error occurred in the action.")
|
|
|
|
|
|
@pytest.fixture
|
|
def flx() -> Falyx:
|
|
sys.argv = ["falyx", "T"]
|
|
flx = Falyx()
|
|
flx.add_command(
|
|
"T",
|
|
"Test",
|
|
action=lambda: "hello",
|
|
)
|
|
flx.add_tldr_example(
|
|
entry_key="T",
|
|
usage="",
|
|
description="This is a TLDR example for the T command.",
|
|
)
|
|
return flx
|
|
|
|
|
|
@pytest.fixture
|
|
def flx_with_submenu() -> Falyx:
|
|
flx = Falyx()
|
|
submenu = Falyx("Submenu")
|
|
submenu.add_command(
|
|
"T",
|
|
"Test",
|
|
action=lambda: "hello from submenu",
|
|
)
|
|
submenu.add_tldr_example(
|
|
entry_key="T",
|
|
usage="",
|
|
description="This is a TLDR example for the T command in the submenu.",
|
|
)
|
|
flx.add_submenu(
|
|
"S",
|
|
"Submenu",
|
|
submenu=submenu,
|
|
)
|
|
return flx
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_basic(capsys):
|
|
sys.argv = ["falyx", "-h"]
|
|
flx = Falyx()
|
|
with pytest.raises(SystemExit):
|
|
await flx.run()
|
|
|
|
captured = capsys.readouterr()
|
|
assert "Show this help menu." in captured.out
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_default_to_menu(flx):
|
|
sys.argv = ["falyx", "T"]
|
|
flx.default_to_menu = False
|
|
|
|
with pytest.raises(SystemExit):
|
|
await flx.run()
|
|
|
|
await flx.run(always_start_menu=True)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_default_to_menu_help(flx):
|
|
sys.argv = ["falyx"]
|
|
flx.default_to_menu = False
|
|
with pytest.raises(SystemExit, match="0"):
|
|
with falyx_console.capture() as capture:
|
|
await flx.run()
|
|
|
|
captured = Text.from_ansi(capture.get()).plain
|
|
assert "Show this help menu." in captured
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_debug_hooks(flx):
|
|
sys.argv = ["falyx", "--debug-hooks", "T"]
|
|
|
|
assert flx.options.get("debug_hooks") is False
|
|
|
|
with pytest.raises(SystemExit):
|
|
await flx.run()
|
|
|
|
assert flx.options.get("debug_hooks") is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_never_prompt(flx):
|
|
sys.argv = ["falyx", "--never-prompt", "T"]
|
|
|
|
assert flx.options.get("never_prompt") is False
|
|
|
|
with pytest.raises(SystemExit):
|
|
await flx.run()
|
|
|
|
falyx_console.print(flx.options.get_namespace_dict("default"))
|
|
|
|
assert flx.options.get("debug_hooks") is False
|
|
assert flx.options.get("never_prompt") is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_bad_args(flx):
|
|
sys.argv = ["falyx", "T", "--unknown-arg"]
|
|
|
|
with pytest.raises(SystemExit, match="2"):
|
|
await flx.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_help(flx):
|
|
sys.argv = ["falyx", "T", "--help"]
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "--help"]
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "-h"]
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "--tldr"]
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "-T"]
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_entry_not_found(flx):
|
|
sys.argv = ["falyx", "UNKNOWN_COMMAND"]
|
|
|
|
with pytest.raises(SystemExit, match="2"):
|
|
await flx.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_test_exceptions(flx):
|
|
flx.add_command(
|
|
"E",
|
|
"Throw Error",
|
|
action=throw_error_action,
|
|
)
|
|
|
|
sys.argv = ["falyx", "E", "ValueError"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "QuitSignal"]
|
|
with pytest.raises(SystemExit, match="130"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "BackSignal"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "CancelSignal"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "HelpSignal"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "FlowSignal"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "--verbose", "E", "FalyxError"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
sys.argv = ["falyx", "E", "UnknownError"]
|
|
with pytest.raises(SystemExit, match="1"):
|
|
await flx.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_no_args(flx):
|
|
sys.argv = ["falyx"]
|
|
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_submenu(flx_with_submenu):
|
|
sys.argv = ["falyx", "S", "T"]
|
|
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx_with_submenu.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_submenu_help(flx_with_submenu):
|
|
sys.argv = ["falyx", "S", "--help"]
|
|
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx_with_submenu.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_submenu_tldr(flx_with_submenu):
|
|
sys.argv = ["falyx", "S", "--tldr"]
|
|
|
|
with pytest.raises(SystemExit, match="0"):
|
|
await flx_with_submenu.run()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_run_preview(flx):
|
|
sys.argv = ["falyx", "preview", "T"]
|
|
|
|
with pytest.raises(SystemExit, match="0"):
|
|
with falyx_console.capture() as capture:
|
|
await flx.run()
|
|
|
|
captured = Text.from_ansi(capture.get()).plain
|
|
assert "Command: 'T'" in captured
|
|
assert "Would call: <lambda>(args=(), kwargs={})" in captured
|