File size: 2,760 Bytes
b93364a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
"""Test the CLI controller."""
from unittest.mock import MagicMock, patch
import pytest
from openbb_cli.controllers.cli_controller import (
CLIController,
handle_job_cmds,
parse_and_split_input,
run_cli,
)
# pylint: disable=redefined-outer-name, unused-argument
def test_parse_and_split_input_custom_filters():
"""Test the parse_and_split_input function with custom filters."""
input_cmd = "query -q AAPL/P"
result = parse_and_split_input(
input_cmd, custom_filters=[r"((\ -q |\ --question|\ ).*?(/))"]
)
assert (
"AAPL/P" not in result
), "Should filter out terms that look like a sorting parameter"
@patch("openbb_cli.controllers.cli_controller.CLIController.print_help")
def test_cli_controller_print_help(mock_print_help):
"""Test the CLIController print_help method."""
controller = CLIController()
controller.print_help()
mock_print_help.assert_called_once()
@pytest.mark.parametrize(
"controller_input, expected_output",
[
("settings", True),
("random_command", False),
],
)
def test_CLIController_has_command(controller_input, expected_output):
"""Test the CLIController has_command method."""
controller = CLIController()
assert hasattr(controller, f"call_{controller_input}") == expected_output
def test_handle_job_cmds_with_export_path():
"""Test the handle_job_cmds function with an export path."""
jobs_cmds = ["export /path/to/export some_command"]
result = handle_job_cmds(jobs_cmds)
expected = "some_command"
assert expected in result[0] # type: ignore
@patch("openbb_cli.controllers.cli_controller.CLIController.switch", return_value=[])
@patch("openbb_cli.controllers.cli_controller.print_goodbye")
def test_run_cli_quit_command(mock_print_goodbye, mock_switch):
"""Test the run_cli function with the quit command."""
run_cli(["quit"], test_mode=True)
mock_print_goodbye.assert_called_once()
@pytest.mark.skip("This test is not working as expected")
def test_execute_openbb_routine_with_mocked_requests():
"""Test the call_exe function with mocked requests."""
with patch("requests.get") as mock_get:
response = MagicMock()
response.status_code = 200
response.json.return_value = {"script": "print('Hello World')"}
mock_get.return_value = response
# Here we need to call the correct function, assuming it's something like `call_exe` for URL-based scripts
controller = CLIController()
controller.call_exe(
["--url", "https://my.openbb.co/u/test/routine/test.openbb"]
)
mock_get.assert_called_with(
"https://my.openbb.co/u/test/routine/test.openbb?raw=true", timeout=10
)
|