mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 04:31:13 +00:00
009699928f
* Relink sdk/sdk -> platform/platform * Relink sdk -> platform * Rename folders * Fix dev_install script * Fix test generators * Fix test generators * Fix test generators * Organize imports
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
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=5,
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def headers():
|
|
access_token = get_token().json()["access_token"]
|
|
return {"Authorization": f"Bearer {access_token}"}
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"params",
|
|
[
|
|
(
|
|
{
|
|
"symbol": "ES",
|
|
"start_date": "2023-01-01",
|
|
"end_date": "2023-06-06",
|
|
"expiration": "2024-06",
|
|
}
|
|
),
|
|
(
|
|
{
|
|
"interval": "1d",
|
|
"period": "1d",
|
|
"prepost": True,
|
|
"adjust": True,
|
|
"back_adjust": True,
|
|
"provider": "yfinance",
|
|
"symbol": "ES",
|
|
"start_date": "2023-01-01",
|
|
"end_date": "2023-06-06",
|
|
"expiration": "2024-06",
|
|
}
|
|
),
|
|
],
|
|
)
|
|
@pytest.mark.integration
|
|
def test_futures_load(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/futures/load?{query_str}"
|
|
result = requests.get(url, headers=headers, timeout=5)
|
|
assert isinstance(result, requests.Response)
|
|
assert result.status_code == 200
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"params",
|
|
[({"symbol": "AAPL", "date": "2023-01-01"})],
|
|
)
|
|
@pytest.mark.integration
|
|
def test_futures_curve(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/futures/curve?{query_str}"
|
|
result = requests.get(url, headers=headers, timeout=5)
|
|
assert isinstance(result, requests.Response)
|
|
assert result.status_code == 200
|