"""Integration tests for charting API."""
import base64
import json
import pytest
import requests
from extensions.tests.conftest import parametrize
from openbb_core.env import Env
from openbb_core.provider.utils.helpers import get_querystring
@pytest.fixture(scope="session")
def headers():
"""Headers fixture."""
return get_headers()
# pylint:disable=redefined-outer-name
data: dict = {}
def get_headers():
"""Get headers for requests."""
if "headers" in data:
return data["headers"]
userpass = f"{Env().API_USERNAME}:{Env().API_PASSWORD}"
userpass_bytes = userpass.encode("ascii")
base64_bytes = base64.b64encode(userpass_bytes)
data["headers"] = {"Authorization": f"Basic {base64_bytes.decode('ascii')}"}
return data["headers"]
def get_equity_data():
"""Get equity data."""
if "stocks_data" in 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)
data["stocks_data"] = result.json()["results"]
return data["stocks_data"]
@parametrize(
"params",
[
(
{
"provider": "yfinance",
"symbol": "AAPL",
"chart": True,
}
),
],
)
@pytest.mark.integration
def test_charting_equity_price_historical(params, headers):
"""Test chart equity price historical.."""
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/equity/price/historical?{query_str}"
result = requests.get(url, headers=headers, timeout=40)
assert isinstance(result, requests.Response)
assert result.status_code == 200
chart = result.json()["chart"]
fig = chart.pop("fig", {})
assert chart
assert not fig
assert list(chart.keys()) == ["content", "format"]
@parametrize(
"params",
[
(
{
"provider": "yfinance",
"symbol": "USDGBP",
"chart": True,
}
),
],
)
@pytest.mark.integration
def test_charting_currency_price_historical(params, headers):
"""Test chart currency price historical."""
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/currency/price/historical?{query_str}"
result = requests.get(url, headers=headers, timeout=40)
assert isinstance(result, requests.Response)
assert result.status_code == 200
chart = result.json()["chart"]
fig = chart.pop("fig", {})
assert chart
assert not fig
assert list(chart.keys()) == ["content", "format"]
@parametrize(
"params",
[
(
{
"provider": "yfinance",
"symbol": "QQQ",
"chart": True,
}
),
],
)
@pytest.mark.integration
def test_charting_etf_historical(params, headers):
"""Test chart etf historical."""
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/etf/historical?{query_str}"
result = requests.get(url, headers=headers, timeout=40)
assert isinstance(result, requests.Respon