d2fbc0ceea
* fix dependencies uv.lock * refactor test markers for clarity * refactor: clean up imports and remove unused files * refactor: remove unused agent files and clean up market API instructions * refactor: enhance wrapper initialization with keyword arguments and clean up tests * refactor: remove PublicBinanceAgent * refactor: aggregator - simplified MarketDataAggregator and related models to functions * refactor: update README and .env.example to reflect the latest changes to the project * refactor: simplify product info and price creation in YFinanceWrapper * refactor: remove get_all_products method from market API wrappers and update documentation * fix: environment variable assertions * refactor: remove status attribute from ProductInfo and update related methods to use timestamp_ms * feat: implement aggregate_history_prices function to calculate hourly price averages * refactor: update docker-compose and app.py for improved environment variable handling and compatibility * feat: add detailed market instructions and improve error handling in price aggregation methods * feat: add aggregated news retrieval methods for top headlines and latest news * refactor: improve error messages in WrapperHandler for better clarity * fix: correct quote currency extraction in create_product_info and remove debug prints from tests
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
import os
|
|
import pytest
|
|
from app.markets import CoinBaseWrapper
|
|
|
|
@pytest.mark.market
|
|
@pytest.mark.api
|
|
@pytest.mark.skipif(not(os.getenv('COINBASE_API_KEY')) or not(os.getenv('COINBASE_API_SECRET')), reason="COINBASE_API_KEY or COINBASE_API_SECRET not set in environment variables")
|
|
class TestCoinBase:
|
|
|
|
def test_coinbase_init(self):
|
|
market = CoinBaseWrapper()
|
|
assert market is not None
|
|
assert hasattr(market, 'currency')
|
|
assert market.currency == "USD"
|
|
|
|
def test_coinbase_get_product(self):
|
|
market = CoinBaseWrapper()
|
|
product = market.get_product("BTC")
|
|
assert product is not None
|
|
assert hasattr(product, 'symbol')
|
|
assert product.symbol == "BTC"
|
|
assert hasattr(product, 'price')
|
|
assert product.price > 0
|
|
|
|
def test_coinbase_get_products(self):
|
|
market = CoinBaseWrapper()
|
|
products = market.get_products(["BTC", "ETH"])
|
|
assert products is not None
|
|
assert isinstance(products, list)
|
|
assert len(products) == 2
|
|
symbols = [p.symbol for p in products]
|
|
assert "BTC" in symbols
|
|
assert "ETH" in symbols
|
|
for product in products:
|
|
assert hasattr(product, 'price')
|
|
assert product.price > 0
|
|
|
|
def test_coinbase_invalid_product(self):
|
|
market = CoinBaseWrapper()
|
|
with pytest.raises(Exception):
|
|
_ = market.get_product("INVALID")
|
|
|
|
def test_coinbase_history(self):
|
|
market = CoinBaseWrapper()
|
|
history = market.get_historical_prices("BTC", limit=5)
|
|
assert history is not None
|
|
assert isinstance(history, list)
|
|
assert len(history) == 5
|
|
for entry in history:
|
|
assert hasattr(entry, 'timestamp_ms')
|
|
assert hasattr(entry, 'close')
|
|
assert hasattr(entry, 'high')
|
|
assert entry.close > 0
|
|
assert entry.high > 0
|
|
assert entry.timestamp_ms > 0
|