more tests for models

This commit is contained in:
hjoaquim
2023-08-01 06:48:08 +01:00
parent d9b45e22ee
commit b3f7e7f042
5 changed files with 55 additions and 0 deletions

View File

@@ -0,0 +1,8 @@
from openbb_core.app.model.hub.features_keys import FeaturesKeys
def test_features_keys():
fk = FeaturesKeys(API_BITQUERY_KEY="test", API_BIZTOC_TOKEN="test2")
assert fk.API_BITQUERY_KEY == "test"
assert fk.API_BIZTOC_TOKEN == "test2"

View File

@@ -0,0 +1,30 @@
from openbb_core.app.model.hub.hub_session import HubSession
def test_hub_session():
session = HubSession(
access_token="mock_access_token",
token_type="mock_token_type",
email="mock_email",
user_uuid="mock_user_uuid",
username="mock_username",
primary_usage="mock_primary_usage",
)
assert session.access_token == "mock_access_token"
assert session.token_type == "mock_token_type"
assert session.email == "mock_email"
assert session.user_uuid == "mock_user_uuid"
assert session.username == "mock_username"
assert session.primary_usage == "mock_primary_usage"
def test_fields():
fields = HubSession.__fields__
fields_keys = fields.keys()
assert "access_token" in fields_keys
assert "token_type" in fields_keys
assert "email" in fields_keys
assert "user_uuid" in fields_keys
assert "username" in fields_keys
assert "primary_usage" in fields_keys

View File

@@ -0,0 +1,7 @@
from openbb_core.app.model.hub.hub_user_settings import HubUserSettings
def test_hub_user_settings():
hub_settings = HubUserSettings()
assert isinstance(hub_settings, HubUserSettings)

View File

@@ -0,0 +1,10 @@
from openbb_core.app.model.results.empty import Empty
from pydantic import BaseModel
def test_empty_model():
empty = Empty()
assert isinstance(empty, Empty)
assert isinstance(empty, BaseModel)