mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-15 12:56:57 +08:00
a116d9bbf2
* Unit test batch 1 * CLI controller * Test batch 3 * Test batch 4 * Test batch 5 * clean some workflows and setup actions * test * rename wfs * rename * update action * Skip * fix cli tests --------- Co-authored-by: Henrique Joaquim <henriquecjoaquim@gmail.com> Co-authored-by: Diogo Sousa <montezdesousa@gmail.com> Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com>
80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
"""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
|
|
)
|