mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
889ff94f12
* add remember attribute to user * don't save hist if not remember * remove routines on exit if not remember * fix unittests * remove hub routines on exit * check if user is local first * remove hist file on exit if not remember * comma * Fix `exe --file` choices on login (#4799) * download and save routines in sync, to have choices immediatly available * fix: docs min --------- Co-authored-by: jose-donato <zmcdonato@gmail.com> Co-authored-by: James Maslek <jmaslek11@gmail.com> --------- Co-authored-by: jose-donato <zmcdonato@gmail.com> Co-authored-by: James Maslek <jmaslek11@gmail.com>
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
# IMPORTATION STANDARD
|
|
|
|
# IMPORTATION THIRDPARTY
|
|
import pytest
|
|
|
|
# IMPORTATION INTERNAL
|
|
from openbb_terminal.core.models.user_model import (
|
|
CredentialsModel,
|
|
PreferencesModel,
|
|
ProfileModel,
|
|
SourcesModel,
|
|
UserModel,
|
|
)
|
|
|
|
# pylint: disable=protected-access, redefined-outer-name
|
|
|
|
TEST_SESSION = {
|
|
"access_token": "test_token",
|
|
"token_type": "bearer",
|
|
"uuid": "test_uuid",
|
|
}
|
|
|
|
|
|
@pytest.fixture(name="test_user")
|
|
def fixture_test_user():
|
|
test_user = UserModel(
|
|
credentials=CredentialsModel(),
|
|
preferences=PreferencesModel(),
|
|
profile=ProfileModel(),
|
|
sources=SourcesModel(),
|
|
)
|
|
test_user.profile.load_user_info(TEST_SESSION, "test@email.com", False)
|
|
return test_user
|
|
|
|
|
|
def test_load_user_info(test_user):
|
|
"""Test load user info."""
|
|
assert test_user.profile.token == "test_token"
|
|
assert test_user.profile.token_type == "bearer"
|
|
assert test_user.profile.uuid == "test_uuid"
|
|
assert test_user.profile.email == "test@email.com"
|
|
assert test_user.profile.remember is False
|
|
|
|
|
|
def test_get_session(test_user):
|
|
"""Test get session."""
|
|
assert test_user.profile.get_session() == TEST_SESSION
|
|
|
|
|
|
def test_get_uuid(test_user):
|
|
"""Test get uuid."""
|
|
assert test_user.profile.get_uuid() == "test_uuid"
|
|
|
|
|
|
def test_get_auth_header(test_user):
|
|
assert test_user.profile.get_auth_header() == "Bearer test_token"
|
|
|
|
|
|
def test_get_token(test_user):
|
|
assert test_user.profile.get_token() == "test_token"
|