mirror of
https://github.com/NCBM/plyngent.git
synced 2026-07-23 14:06:27 +08:00
24 lines
628 B
Python
24 lines
628 B
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from click.testing import CliRunner
|
||
|
|
|
||
|
|
from plyngent.cli.app import main
|
||
|
|
|
||
|
|
|
||
|
|
def test_providers_list() -> None:
|
||
|
|
config = Path(__file__).resolve().parents[1] / "test_config" / "plyngent-valid.toml"
|
||
|
|
runner = CliRunner()
|
||
|
|
result = runner.invoke(main, ["providers", "--config", str(config)])
|
||
|
|
assert result.exit_code == 0
|
||
|
|
assert "test1" in result.output
|
||
|
|
assert "openai" in result.output
|
||
|
|
|
||
|
|
|
||
|
|
def test_help() -> None:
|
||
|
|
runner = CliRunner()
|
||
|
|
result = runner.invoke(main, ["--help"])
|
||
|
|
assert result.exit_code == 0
|
||
|
|
assert "chat" in result.output
|