mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-07 22:40:49 +08:00
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
"""Test the Models Settings module."""
|
|
|
|
from unittest.mock import mock_open, patch
|
|
|
|
from openbb_cli.models.settings import Settings
|
|
|
|
# pylint: disable=unused-argument
|
|
|
|
|
|
def test_default_values():
|
|
"""Test the default values of the settings model."""
|
|
fields = Settings.model_fields
|
|
assert fields["TEST_MODE"].default is False
|
|
assert fields["DEBUG_MODE"].default is False
|
|
assert fields["DEV_BACKEND"].default is False
|
|
assert fields["FILE_OVERWRITE"].default is False
|
|
assert fields["SHOW_VERSION"].default is True
|
|
assert fields["USE_INTERACTIVE_DF"].default is True
|
|
assert fields["USE_CLEAR_AFTER_CMD"].default is False
|
|
assert fields["USE_DATETIME"].default is True
|
|
assert fields["USE_PROMPT_TOOLKIT"].default is True
|
|
assert fields["ENABLE_EXIT_AUTO_HELP"].default is True
|
|
assert fields["ENABLE_RICH_PANEL"].default is True
|
|
assert fields["TOOLBAR_HINT"].default is True
|
|
assert fields["SHOW_MSG_OBBJECT_REGISTRY"].default is False
|
|
assert fields["TIMEZONE"].default == "America/New_York"
|
|
assert fields["FLAIR"].default == ":openbb"
|
|
assert fields["PREVIOUS_USE"].default is False
|
|
assert fields["N_TO_KEEP_OBBJECT_REGISTRY"].default == 10
|
|
assert fields["N_TO_DISPLAY_OBBJECT_REGISTRY"].default == 5
|
|
assert fields["RICH_STYLE"].default == "dark"
|
|
assert fields["ALLOWED_NUMBER_OF_ROWS"].default == 20
|
|
assert fields["ALLOWED_NUMBER_OF_COLUMNS"].default == 5
|
|
assert fields["HUB_URL"].default == "https://my.openbb.co"
|
|
assert fields["BASE_URL"].default == "https://payments.openbb.co"
|
|
|
|
|
|
# Test __repr__ output
|
|
def test_repr():
|
|
"""Test the __repr__ method of the settings model."""
|
|
settings = Settings()
|
|
repr_str = settings.__repr__() # pylint: disable=C2801
|
|
assert "Settings\n\n" in repr_str
|
|
|
|
|
|
# Test loading from environment variables
|
|
@patch(
|
|
"openbb_cli.models.settings.dotenv_values",
|
|
return_value={"OPENBB_TEST_MODE": "True", "OPENBB_VERSION": "2.0.0"},
|
|
)
|
|
def test_from_env(mock_dotenv_values):
|
|
"""Test loading settings from environment variables."""
|
|
settings = Settings.from_env({}) # type: ignore
|
|
assert settings["TEST_MODE"] == "True"
|
|
assert settings["VERSION"] == "2.0.0"
|
|
|
|
|
|
# Test setting an item and updating .env
|
|
@patch("openbb_cli.models.settings.set_key")
|
|
@patch(
|
|
"openbb_cli.models.settings.open",
|
|
new_callable=mock_open,
|
|
read_data="TEST_MODE=False\n",
|
|
)
|
|
def test_set_item(mock_file, mock_set_key):
|
|
"""Test setting an item and updating the .env file."""
|
|
settings = Settings()
|
|
settings.set_item("TEST_MODE", True)
|
|
assert settings.TEST_MODE is True
|