add http basic header to integration tests

.github/workflows/platform-api-integration-test.yml

@@ -15,6 +15,9 @@ jobs:
     runs-on: ubuntu-latest
     env:
       SLACK_WEBHOOK_URL: ${{ secrets.SLACK_WEBHOOK_URL }}
+      OPENBB_API_AUTH: true
+      OPENBB_API_USERNAME: 'USER'
+      OPENBB_API_PASSWORD: 'PASS'
 
     steps:
       - name: Slack Notification - Starting

openbb_platform/extensions/crypto/integration/test_crypto_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/econometrics/integration/test_econometrics_api.py

@@ -1,37 +1,22 @@
 import json
 import random
+from extensions.tests.utils.helpers import get_auth
 
 import pytest
 import requests
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
-def auth_header():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
-
-
 @pytest.fixture(scope="session")
 def headers():
-    h = {}
-    auth = auth_header()
-    h.update(auth)
-    return h
+    return {"Authorization": get_auth()}
 
 
-def get_data(menu: str, symbol: str, provider: str):
+def get_data(menu: str, symbol: str, provider: str, headers):
     """Randomly pick a symbol and a provider and get data from the selected menu."""
 
     url = f"http://0.0.0.0:8000/api/v1/{menu}/load?symbol={symbol}&provider={provider}"
-    result = requests.get(url, headers=auth_header(), timeout=10)
+    result = requests.get(url, headers=headers, timeout=10)
     return result.json()["results"]
 
 
@@ -45,7 +30,9 @@ def get_stocks_data():
     symbol = random.choice(["AAPL", "NVDA", "MSFT", "TSLA", "AMZN", "V"])  # noqa: S311
     provider = random.choice(["fmp", "intrinio", "polygon", "yfinance"])  # noqa: S311
 
-    data["stocks_data"] = get_data("stocks", symbol=symbol, provider=provider)
+    data["stocks_data"] = get_data(
+        "stocks", symbol=symbol, provider=provider, headers=headers
+    )
     return data["stocks_data"]
 
 
@@ -57,7 +44,9 @@ def get_crypto_data():
     symbol = random.choice(["BTC"])  # noqa: S311
     provider = random.choice(["fmp"])  # noqa: S311
 
-    data["crypto_data"] = get_data(menu="crypto", symbol=symbol, provider=provider)
+    data["crypto_data"] = get_data(
+        menu="crypto", symbol=symbol, provider=provider, headers=headers
+    )
     return data["crypto_data"]
 
 

openbb_platform/extensions/economy/integration/test_economy_api.py

@@ -1,20 +1,13 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/fixedincome/integration/test_fixedincome_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/forex/integration/test_forex_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/futures/integration/test_futures_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/news/integration/test_news_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/qa/integration/test_qa_api.py

@@ -3,35 +3,20 @@ import random
 
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
-def auth_header():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
-
-
 @pytest.fixture(scope="session")
 def headers():
-    h = {}
-    auth = auth_header()
-    h.update(auth)
-    return h
+    return {"Authorization": get_auth()}
 
 
-def get_data(menu: str, symbol: str, provider: str):
+def get_data(menu: str, symbol: str, provider: str, headers):
     """Randomly pick a symbol and a provider and get data from the selected menu."""
 
     url = f"http://0.0.0.0:8000/api/v1/{menu}/load?symbol={symbol}&provider={provider}"
-    result = requests.get(url, headers=auth_header(), timeout=10)
+    result = requests.get(url, headers=headers, timeout=10)
     return result.json()["results"]
 
 
@@ -45,7 +30,9 @@ def get_stocks_data():
     symbol = random.choice(["AAPL", "NVDA", "MSFT", "TSLA", "AMZN", "V"])  # noqa: S311
     provider = random.choice(["fmp", "intrinio", "polygon", "yfinance"])  # noqa: S311
 
-    data["stocks_data"] = get_data("stocks", symbol=symbol, provider=provider)
+    data["stocks_data"] = get_data(
+        "stocks", symbol=symbol, provider=provider, headers=headers
+    )
     return data["stocks_data"]
 
 
@@ -57,7 +44,9 @@ def get_crypto_data():
     symbol = random.choice(["BTC"])  # noqa: S311
     provider = random.choice(["fmp"])  # noqa: S311
 
-    data["crypto_data"] = get_data(menu="crypto", symbol=symbol, provider=provider)
+    data["crypto_data"] = get_data(
+        menu="crypto", symbol=symbol, provider=provider, headers=headers
+    )
     return data["crypto_data"]
 
 

openbb_platform/extensions/stocks/integration/test_stocks_api.py

@@ -1,20 +1,12 @@
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 
 
 @pytest.mark.parametrize(

openbb_platform/extensions/ta/integration/test_ta_api.py

@@ -3,35 +3,20 @@ import random
 
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
 
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
-def auth_header():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
-
-
 @pytest.fixture(scope="session")
 def headers():
-    h = {}
-    auth = auth_header()
-    h.update(auth)
-    return h
+    return {"Authorization": get_auth()}
 
 
-def get_data(menu: str, symbol: str, provider: str):
+def get_data(menu: str, symbol: str, provider: str, headers):
     """Randomly pick a symbol and a provider and get data from the selected menu."""
 
     url = f"http://0.0.0.0:8000/api/v1/{menu}/load?symbol={symbol}&provider={provider}"
-    result = requests.get(url, headers=auth_header(), timeout=10)
+    result = requests.get(url, headers=headers, timeout=10)
     return result.json()["results"]
 
 
@@ -45,7 +30,9 @@ def get_stocks_data():
     symbol = random.choice(["AAPL", "NVDA", "MSFT", "TSLA", "AMZN", "V"])  # noqa: S311
     provider = random.choice(["fmp", "intrinio", "polygon", "yfinance"])  # noqa: S311
 
-    data["stocks_data"] = get_data("stocks", symbol=symbol, provider=provider)
+    data["stocks_data"] = get_data(
+        "stocks", symbol=symbol, provider=provider, headers=headers
+    )
     return data["stocks_data"]
 
 
@@ -57,7 +44,9 @@ def get_crypto_data():
     symbol = random.choice(["BTC"])  # noqa: S311
     provider = random.choice(["fmp"])  # noqa: S311
 
-    data["crypto_data"] = get_data(menu="crypto", symbol=symbol, provider=provider)
+    data["crypto_data"] = get_data(
+        menu="crypto", symbol=symbol, provider=provider, headers=headers
+    )
     return data["crypto_data"]
 
 

openbb_platform/extensions/tests/utils/helpers.py

@@ -0,0 +1,13 @@
+import base64
+
+from openbb_core.env import Env
+
+
+def get_auth() -> str:
+    """Returns the basic auth header."""
+
+    userpass = f"{Env().API_USERNAME}:{Env().API_PASSWORD}"
+    userpass_bytes = userpass.encode("ascii")
+    base64_bytes = base64.b64encode(userpass_bytes)
+
+    return f"Basic {base64_bytes.decode('ascii')}"

openbb_platform/extensions/tests/utils/integration_tests_api_generator.py

@@ -24,21 +24,12 @@ def write_init_test_template(http_method: str, path: str):
     template += """
 import pytest
 import requests
+from extensions.tests.utils.helpers import get_auth
 from openbb_provider.utils.helpers import get_querystring
 
-
-def get_token():
-    return requests.post(
-        "http://0.0.0.0:8000/api/v1/account/token",
-        data={"username": "openbb", "password": "openbb"},
-        timeout=10,
-    )
-
-
 @pytest.fixture(scope="session")
 def headers():
-    access_token = get_token().json()["access_token"]
-    return {"Authorization": f"Bearer {access_token}"}
+    return {"Authorization": get_auth()}
 """
 
     with open(path, "w") as f:

openbb_platform/tests/conftest.py

@@ -1,2 +1,4 @@
 # TODO: See if we can dynamically import the test modules under the openbb_platform
-# file structure.
+# f
+
+# TODO: Add global fixture for headers here