File size: 2,701 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 81 82 83 84 85 86 87 88 89 90 |
"""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 TestController(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 = TestController()
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}"
|