From b4c528d3aa4e205164d93e6436475c10bd5d9c2c Mon Sep 17 00:00:00 2001
From: Danglewood <85772166+deeleeramone@users.noreply.github.com>
Date: Thu, 12 Dec 2024 23:15:07 -0800
Subject: [PATCH] some fails

---
 .../integration/test_websockets_api.py        | 34 +++++++++++++++++++
 .../integration/test_websockets_python.py     | 33 ++++++++++++++++++
 2 files changed, 67 insertions(+)

diff --git a/openbb_platform/extensions/websockets/integration/test_websockets_api.py b/openbb_platform/extensions/websockets/integration/test_websockets_api.py
index bc8849c7a99..6d2c15c4b9d 100644
--- a/openbb_platform/extensions/websockets/integration/test_websockets_api.py
+++ b/openbb_platform/extensions/websockets/integration/test_websockets_api.py
@@ -88,6 +88,7 @@ def headers():
                 "symbol": "spy,qqq,iwm,tsla,nvda",
                 "asset_type": "stock",
                 "feed": "realtime",
+                "trades_only": True,
                 "auth_token": None,
                 "results_file": None,
                 "save_results": False,
@@ -304,6 +305,39 @@ def test_websockets_unsubscribe(params, headers):
     assert result.status_code == 200
 
 
+@parametrize(
+    "params",
+    [
+        {
+            "name": "test_fmp",
+            "auth_token": None,
+        },
+        {
+            "name": "test_tiingo",
+            "auth_token": None,
+        },
+        {
+            "name": "test_polygon",
+            "auth_token": None,
+        },
+        {
+            "name": "test_intrinio",
+            "auth_token": None,
+        },
+    ],
+)
+@pytest.mark.skip(reason="Python interface only.")
+def test_websockets_get_client(params, headers):
+    """Test the websockets_get_client endpoint."""
+    params = {p: v for p, v in params.items() if v}
+
+    query_str = get_querystring(params, [])
+    url = f"http://0.0.0.0:8000/api/v1/websockets/get_client?{query_str}"
+    result = requests.get(url, headers=headers, timeout=10)
+    assert isinstance(result, requests.Response)
+    assert result.status_code == 200
+
+
 @parametrize(
     "params",
     [
diff --git a/openbb_platform/extensions/websockets/integration/test_websockets_python.py b/openbb_platform/extensions/websockets/integration/test_websockets_python.py
index 063857ebf57..9521070b061 100644
--- a/openbb_platform/extensions/websockets/integration/test_websockets_python.py
+++ b/openbb_platform/extensions/websockets/integration/test_websockets_python.py
@@ -84,6 +84,7 @@ def obb(pytestconfig):
                 "symbol": "spy,qqq,iwm,tsla,nvda",
                 "asset_type": "stock",
                 "feed": "realtime",
+                "trades_only": True,
                 "auth_token": None,
                 "results_file": None,
                 "save_results": False,
@@ -320,6 +321,38 @@ def test_websockets_get_client(params, obb):
     assert result.results is not None
 
 
+@parametrize(
+    "params",
+    [
+        {
+            "name": "test_fmp",
+            "auth_token": None,
+        },
+        {
+            "name": "test_tiingo",
+            "auth_token": None,
+        },
+        {
+            "name": "test_polygon",
+            "auth_token": None,
+        },
+        {
+            "name": "test_intrinio",
+            "auth_token": None,
+        },
+    ],
+)
+@pytest.mark.integration
+def test_websockets_get_client_status(params, obb):
+    """Test the websockets_get_client endpoint."""
+    params = {p: v for p, v in params.items() if v}
+
+    result = obb.websockets.get_client_status(**params)
+    assert result
+    assert isinstance(result, OBBject)
+    assert result.results is not None
+
+
 @parametrize(
     "params",
     [