File size: 2,617 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 |
"""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
|