From 6d13807047ed81cf0dcc791f5ec1a91ec9feb1b5 Mon Sep 17 00:00:00 2001 From: James Maslek Date: Wed, 29 Nov 2023 15:49:34 -0500 Subject: [PATCH] debuggint int tests (#5823) * debug * debug2 * extra debug step * Another hail mary * add debug * MMMMMbop * didnt import base64 * Other tests with headers * increase timeout to 20 --------- Co-authored-by: Pratyush Shukla --- .../workflows/platform-api-integration-test.yml | 1 - .../charting/integration/test_charting_api.py | 2 +- .../integration/test_derivatives_api.py | 9 ++++++++- .../equity/integration/test_equity_api.py | 14 ++++++++++---- .../extensions/news/integration/test_news_api.py | 9 ++++++++- 5 files changed, 27 insertions(+), 8 deletions(-) diff --git a/.github/workflows/platform-api-integration-test.yml b/.github/workflows/platform-api-integration-test.yml index 1bc4ad96bb3..961241328e6 100644 --- a/.github/workflows/platform-api-integration-test.yml +++ b/.github/workflows/platform-api-integration-test.yml @@ -82,7 +82,6 @@ jobs: - name: Launch the Uvicorn Process run: | source runner_env/bin/activate - pip list uvicorn openbb_core.api.rest_api:app --host 0.0.0.0 --port 8000 --reload & - name: Wait for 42 seconds diff --git a/openbb_platform/extensions/charting/integration/test_charting_api.py b/openbb_platform/extensions/charting/integration/test_charting_api.py index a6f1f8d8ddd..f0001c9be55 100644 --- a/openbb_platform/extensions/charting/integration/test_charting_api.py +++ b/openbb_platform/extensions/charting/integration/test_charting_api.py @@ -37,7 +37,7 @@ def get_equity_data(): return data["stocks_data"] url = "http://0.0.0.0:8000/api/v1/equity/price/historical?symbol=AAPL&provider=fmp" - result = requests.get(url, headers=get_headers(), timeout=10) + result = requests.get(url, headers=get_headers(), timeout=20) data["stocks_data"] = result.json()["results"] return data["stocks_data"] diff --git a/openbb_platform/extensions/derivatives/integration/test_derivatives_api.py b/openbb_platform/extensions/derivatives/integration/test_derivatives_api.py index 45093aed6a5..53ccec665da 100644 --- a/openbb_platform/extensions/derivatives/integration/test_derivatives_api.py +++ b/openbb_platform/extensions/derivatives/integration/test_derivatives_api.py @@ -1,7 +1,10 @@ """API integration tests for the derivatives extension.""" +import base64 + import pytest import requests +from openbb_core.env import Env from openbb_core.provider.utils.helpers import get_querystring # pylint: disable=too-many-lines,redefined-outer-name @@ -9,7 +12,11 @@ from openbb_core.provider.utils.helpers import get_querystring @pytest.fixture(scope="session") def headers(): - return {} + userpass = f"{Env().API_USERNAME}:{Env().API_PASSWORD}" + userpass_bytes = userpass.encode("ascii") + base64_bytes = base64.b64encode(userpass_bytes) + + return {"Authorization": f"Basic {base64_bytes.decode('ascii')}"} @pytest.mark.parametrize( diff --git a/openbb_platform/extensions/equity/integration/test_equity_api.py b/openbb_platform/extensions/equity/integration/test_equity_api.py index 1a1fb621f84..04b25cb00f9 100644 --- a/openbb_platform/extensions/equity/integration/test_equity_api.py +++ b/openbb_platform/extensions/equity/integration/test_equity_api.py @@ -1,8 +1,10 @@ """API integration tests for equity extension.""" +import base64 from datetime import time import pytest import requests +from openbb_core.env import Env from openbb_core.provider.utils.helpers import get_querystring # pylint: disable=too-many-lines,redefined-outer-name @@ -10,7 +12,11 @@ from openbb_core.provider.utils.helpers import get_querystring @pytest.fixture(scope="session") def headers(): - return {} + userpass = f"{Env().API_USERNAME}:{Env().API_PASSWORD}" + userpass_bytes = userpass.encode("ascii") + base64_bytes = base64.b64encode(userpass_bytes) + + return {"Authorization": f"Basic {base64_bytes.decode('ascii')}"} @pytest.mark.parametrize( @@ -1329,7 +1335,7 @@ def test_equity_shorts_short_interest(params, headers): query_str = get_querystring(params, []) url = f"http://0.0.0.0:8000/api/v1/equity/shorts/short_interest?{query_str}" - result = requests.get(url, headers=headers, timeout=10) + result = requests.get(url, headers=headers, timeout=20) assert isinstance(result, requests.Response) assert result.status_code == 200 @@ -1369,7 +1375,7 @@ def test_equity_price_nbbo(params, headers): query_str = get_querystring(params, []) url = f"http://0.0.0.0:8000/api/v1/equity/price/nbbo?{query_str}" - result = requests.get(url, headers=headers, timeout=10) + result = requests.get(url, headers=headers, timeout=20) assert isinstance(result, requests.Response) assert result.status_code == 200 @@ -1410,7 +1416,7 @@ def test_equity_market_snapshots(params, headers): query_str = get_querystring(params, []) url = f"http://0.0.0.0:8000/api/v1/equity/market_snapshots?{query_str}" - result = requests.get(url, headers=headers, timeout=10) + result = requests.get(url, headers=headers, timeout=20) assert isinstance(result, requests.Response) assert result.status_code == 200 diff --git a/openbb_platform/extensions/news/integration/test_news_api.py b/openbb_platform/extensions/news/integration/test_news_api.py index 3aff0a40c0a..9b94ac1796b 100644 --- a/openbb_platform/extensions/news/integration/test_news_api.py +++ b/openbb_platform/extensions/news/integration/test_news_api.py @@ -1,11 +1,18 @@ +import base64 + import pytest import requests +from openbb_core.env import Env from openbb_core.provider.utils.helpers import get_querystring @pytest.fixture(scope="session") def headers(): - return {} + userpass = f"{Env().API_USERNAME}:{Env().API_PASSWORD}" + userpass_bytes = userpass.encode("ascii") + base64_bytes = base64.b64encode(userpass_bytes) + + return {"Authorization": f"Basic {base64_bytes.decode('ascii')}"} # pylint: disable=redefined-outer-name