mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
869cfed125
* merge branch develop with main * bump desktop version * fix the merge * sync Cargo.lock to desktop 2.0.0 * fix cli conftest.py --------- Co-authored-by: deeleeramone <>
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}"
|
|
)
|