mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 12:40:35 +00:00
9a30186174
* stash some changes * add more robust testing * mypy * point PR at V5 * introduce spec file * codespell * test fix * fix workflow environment setup * fix workflow environment setup * fix workflow environment setup * add pyyaml to dependencies * split lint jobs * fix workflow environment setup * fix workflow environment setup * workflow env setup * workflow env setup * clean up code comments * add auth hook entrypoints * codespell * add codegen feature * codespell * move _unpack into dispatchers for consistency with codegen packages * surface nested models in the response * fix missing coverage in CI * socrata updates * test fix * detect plotly output * add --include and --exclude flags from generate-extension command * cap test matrix at python 3.14 * no useless comments * platform controller command description split * merge URL overloads from path params * exclude none and unset from model dump --------- Co-authored-by: deeleeramone <> Co-authored-by: Copilot <copilot@github.com>
89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
"""Integration tests for the base_controller module."""
|
|
|
|
from unittest.mock import Mock, patch
|
|
|
|
import pytest
|
|
|
|
from openbb_cli.controllers.base_controller import BaseController
|
|
from openbb_cli.session import Session
|
|
|
|
|
|
class DummyController(BaseController):
|
|
"""Test controller for the BaseController."""
|
|
|
|
PATH = "/test/"
|
|
|
|
def print_help(self):
|
|
"""Print help message."""
|
|
|
|
|
|
@pytest.fixture
|
|
def base_controller():
|
|
"""Set up the environment for each test function."""
|
|
session = Session() # noqa: F841
|
|
controller = DummyController()
|
|
return controller
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_check_path_valid(base_controller):
|
|
"""Test that check_path does not raise an error for a valid path."""
|
|
base_controller.PATH = "/equity/"
|
|
try:
|
|
base_controller.check_path()
|
|
except ValueError:
|
|
pytest.fail("check_path raised ValueError unexpectedly!")
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_check_path_invalid(base_controller):
|
|
"""Test that check_path raises an error for an invalid path."""
|
|
with pytest.raises(ValueError):
|
|
base_controller.PATH = "invalid_path"
|
|
base_controller.check_path()
|
|
|
|
with pytest.raises(ValueError):
|
|
base_controller.PATH = "/invalid_path"
|
|
base_controller.check_path()
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_parse_input(base_controller):
|
|
"""Test the parse_input method."""
|
|
input_str = "/equity/price/help"
|
|
expected_output = ["home", "equity", "price", "help"]
|
|
assert base_controller.parse_input(input_str) == expected_output, (
|
|
"Input parsing failed"
|
|
)
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_switch_command_execution(base_controller):
|
|
"""Test the switch method."""
|
|
base_controller.queue = []
|
|
base_controller.switch("/home/../reset/")
|
|
assert base_controller.queue == [
|
|
"home",
|
|
"..",
|
|
"reset",
|
|
], "Switch did not update the queue correctly"
|
|
|
|
|
|
@patch("openbb_cli.controllers.base_controller.BaseController.call_help")
|
|
@pytest.mark.integration
|
|
def test_command_routing(mock_call_help, base_controller):
|
|
"""Test the command routing."""
|
|
base_controller.switch("help")
|
|
mock_call_help.assert_called_once()
|
|
|
|
|
|
@pytest.mark.integration
|
|
def test_custom_reset(base_controller):
|
|
"""Test the custom reset method."""
|
|
base_controller.custom_reset = Mock(return_value=["custom", "reset"])
|
|
base_controller.call_reset(None)
|
|
expected_queue = ["quit", "reset", "custom", "reset"]
|
|
assert base_controller.queue == expected_queue, (
|
|
f"Expected queue to be {expected_queue}, but was {base_controller.queue}"
|
|
)
|