From b3f7e7f0422ebc19ce16afd3725500d1b0b5efc1 Mon Sep 17 00:00:00 2001
From: hjoaquim <h.joaquim@campus.fct.unl.pt>
Date: Tue, 1 Aug 2023 06:48:08 +0100
Subject: [PATCH] more tests for models

---
 .../openbb_core/app/model/error/__init__.py   |  0
 .../tests/app/models/test_features_keys.py    |  8 +++++
 .../core/tests/app/models/test_hub_session.py | 30 +++++++++++++++++++
 .../app/models/test_hub_user_settings.py      |  7 +++++
 .../sdk/core/tests/app/results/test_empty.py  | 10 +++++++
 5 files changed, 55 insertions(+)
 delete mode 100644 openbb_sdk/sdk/core/openbb_core/app/model/error/__init__.py
 create mode 100644 openbb_sdk/sdk/core/tests/app/models/test_features_keys.py
 create mode 100644 openbb_sdk/sdk/core/tests/app/models/test_hub_session.py
 create mode 100644 openbb_sdk/sdk/core/tests/app/models/test_hub_user_settings.py
 create mode 100644 openbb_sdk/sdk/core/tests/app/results/test_empty.py

diff --git a/openbb_sdk/sdk/core/openbb_core/app/model/error/__init__.py b/openbb_sdk/sdk/core/openbb_core/app/model/error/__init__.py
deleted file mode 100644
index e69de29bb2d..00000000000
diff --git a/openbb_sdk/sdk/core/tests/app/models/test_features_keys.py b/openbb_sdk/sdk/core/tests/app/models/test_features_keys.py
new file mode 100644
index 00000000000..19eda989243
--- /dev/null
+++ b/openbb_sdk/sdk/core/tests/app/models/test_features_keys.py
@@ -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"
diff --git a/openbb_sdk/sdk/core/tests/app/models/test_hub_session.py b/openbb_sdk/sdk/core/tests/app/models/test_hub_session.py
new file mode 100644
index 00000000000..1f55189599c
--- /dev/null
+++ b/openbb_sdk/sdk/core/tests/app/models/test_hub_session.py
@@ -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
diff --git a/openbb_sdk/sdk/core/tests/app/models/test_hub_user_settings.py b/openbb_sdk/sdk/core/tests/app/models/test_hub_user_settings.py
new file mode 100644
index 00000000000..ca2f78ff9a8
--- /dev/null
+++ b/openbb_sdk/sdk/core/tests/app/models/test_hub_user_settings.py
@@ -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)
diff --git a/openbb_sdk/sdk/core/tests/app/results/test_empty.py b/openbb_sdk/sdk/core/tests/app/results/test_empty.py
new file mode 100644
index 00000000000..79d9796505a
--- /dev/null
+++ b/openbb_sdk/sdk/core/tests/app/results/test_empty.py
@@ -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)