mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-07 06:23:26 +08:00
9f0d592839
* remove python 3.9 support and code * black * more cli lint * more linting * more lint * fix for tests * docstring grammar police * add lock to to build function to avoid async import race conditions * grammar police * lots more linting * relock
90 lines
2.6 KiB
Python
90 lines
2.6 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
|
|
|
|
# pylint: disable=unused-variable, redefined-outer-name
|
|
|
|
|
|
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" # Missing leading '/'
|
|
base_controller.check_path()
|
|
|
|
with pytest.raises(ValueError):
|
|
base_controller.PATH = "/invalid_path" # Missing trailing '/'
|
|
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 = ["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}"
|