Add tests

OpenBB-finance · Oct 31, 2023 · 0ce70af · 0ce70af
1 parent 968ad50
commit 0ce70af
Show file tree

Hide file tree

Showing 6 changed files with 57,014 additions and 6 deletions.
diff --git a/openbb_platform/extensions/stocks/integration/test_stocks_api.py b/openbb_platform/extensions/stocks/integration/test_stocks_api.py
@@ -959,7 +959,7 @@ def test_stocks_info(params, headers):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10, "provider": "wsj"})],
+    [({"sort": "desc", "provider": "yfinance"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_gainers(params, headers):
@@ -974,7 +974,7 @@ def test_stocks_disc_gainers(params, headers):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10, "provider": "wsj"})],
+    [({"sort": "desc", "provider": "yfinance"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_losers(params, headers):
@@ -989,7 +989,7 @@ def test_stocks_disc_losers(params, headers):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10, "provider": "wsj"})],
+    [({"sort": "desc", "provider": "yfinance"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_active(params, headers):
@@ -1000,3 +1000,61 @@ def test_stocks_disc_active(params, headers):
     result = requests.get(url, headers=headers, timeout=10)
     assert isinstance(result, requests.Response)
     assert result.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "params",
+    [
+        ({"symbol": "AAPL", "provider": "sec"}),
+        ({"limit": 24, "provider": "sec", "symbol": "AAPL", "skip_reports": 1}),
+    ],
+)
+@pytest.mark.integration
+def test_stocks_ftd(params, headers):
+    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/stocks/ftd?{query_str}"
+    result = requests.get(url, headers=headers, timeout=10)
+    assert isinstance(result, requests.Response)
+    assert result.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "params",
+    [({"symbol": "AAPL", "provider": "fmp"})],
+)
+@pytest.mark.integration
+def test_stocks_price_performance(params, headers):
+    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/stocks/price_performance?{query_str}"
+    result = requests.get(url, headers=headers, timeout=10)
+    assert isinstance(result, requests.Response)
+    assert result.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "params",
+    [
+        (
+            {
+                "symbol": "AAPL",
+                "start_date": "2023-01-01",
+                "end_date": "2023-06-06",
+                "limit": 300,
+                "provider": "intrinio",
+            }
+        )
+    ],
+)
+@pytest.mark.integration
+def test_stocks_calendar_ipo(params, headers):
+    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/stocks/calendar_ipo?{query_str}"
+    result = requests.get(url, headers=headers, timeout=10)
+    assert isinstance(result, requests.Response)
+    assert result.status_code == 200
diff --git a/openbb_platform/extensions/stocks/integration/test_stocks_python.py b/openbb_platform/extensions/stocks/integration/test_stocks_python.py
@@ -909,7 +909,7 @@ def test_stocks_info(params, obb):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10})],
+    [({"sort": "desc"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_gainers(params, obb):
@@ -923,7 +923,7 @@ def test_stocks_disc_gainers(params, obb):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10})],
+    [({"sort": "desc"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_losers(params, obb):
@@ -937,7 +937,7 @@ def test_stocks_disc_losers(params, obb):
 
 @pytest.mark.parametrize(
     "params",
-    [({"sort": "desc", "limit": 10})],
+    [({"sort": "desc"})],
 )
 @pytest.mark.integration
 def test_stocks_disc_active(params, obb):
@@ -947,3 +947,57 @@ def test_stocks_disc_active(params, obb):
     assert result
     assert isinstance(result, OBBject)
     assert len(result.results) > 0
+
+
+@pytest.mark.parametrize(
+    "params",
+    [
+        ({"symbol": "AAPL"}),
+        ({"limit": 24, "provider": "sec", "symbol": "AAPL", "skip_reports": 1}),
+    ],
+)
+@pytest.mark.integration
+def test_stocks_ftd(params, obb):
+    params = {p: v for p, v in params.items() if v}
+
+    result = obb.stocks.ftd(**params)
+    assert result
+    assert isinstance(result, OBBject)
+    assert len(result.results) > 0
+
+
+@pytest.mark.parametrize(
+    "params",
+    [({"symbol": "AAPL"})],
+)
+@pytest.mark.integration
+def test_stocks_price_performance(params, obb):
+    params = {p: v for p, v in params.items() if v}
+
+    result = obb.stocks.price_performance(**params)
+    assert result
+    assert isinstance(result, OBBject)
+    assert len(result.results) > 0
+
+
+@pytest.mark.parametrize(
+    "params",
+    [
+        (
+            {
+                "symbol": "AAPL",
+                "start_date": "2023-01-01",
+                "end_date": "2023-06-06",
+                "limit": 300,
+            }
+        )
+    ],
+)
+@pytest.mark.integration
+def test_stocks_calendar_ipo(params, obb):
+    params = {p: v for p, v in params.items() if v}
+
+    result = obb.stocks.calendar_ipo(**params)
+    assert result
+    assert isinstance(result, OBBject)
+    assert len(result.results) > 0