mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-07 22:40:49 +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>
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
"""Test the Controller Factory."""
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
from openbb_cli.controllers.platform_controller_factory import (
|
|
PlatformControllerFactory,
|
|
)
|
|
|
|
# pylint: disable=redefined-outer-name, unused-argument
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_processor():
|
|
"""Fixture to mock ArgparseClassProcessor."""
|
|
with patch(
|
|
"openbb_cli.controllers.platform_controller_factory.ArgparseClassProcessor"
|
|
) as mock:
|
|
instance = mock.return_value
|
|
instance.paths = {"settings": "menu"}
|
|
instance.translators = {"test_router_settings": MagicMock()}
|
|
yield instance
|
|
|
|
|
|
@pytest.fixture
|
|
def platform_router():
|
|
"""Fixture to provide a mock platform_router class."""
|
|
|
|
class MockRouter:
|
|
pass
|
|
|
|
return MockRouter
|
|
|
|
|
|
@pytest.fixture
|
|
def factory(platform_router, mock_processor):
|
|
"""Fixture to create a PlatformControllerFactory with mocked dependencies."""
|
|
return PlatformControllerFactory(
|
|
platform_router=platform_router, reference={"test": "ref"}
|
|
)
|
|
|
|
|
|
def test_init(mock_processor):
|
|
"""Test the initialization of the PlatformControllerFactory."""
|
|
factory = PlatformControllerFactory(
|
|
platform_router=MagicMock(), reference={"test": "ref"}
|
|
)
|
|
assert factory.router_name.lower() == "magicmock"
|
|
assert factory.controller_name == "MagicmockController"
|
|
|
|
|
|
def test_create_controller(factory):
|
|
"""Test the creation of a controller class."""
|
|
ControllerClass = factory.create()
|
|
|
|
assert "PlatformController" in [base.__name__ for base in ControllerClass.__bases__]
|
|
assert ControllerClass.CHOICES_GENERATION
|
|
assert "settings" in ControllerClass.CHOICES_MENUS
|
|
assert "test_router_settings" not in [
|
|
cmd.replace("test_router_", "") for cmd in ControllerClass.CHOICES_COMMANDS
|
|
]
|