45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
import asyncio
|
|
import logging
|
|
from decimal import Decimal
|
|
from lbry.testcase import AsyncioTestCase
|
|
from lbry.extras.daemon.exchange_rate_manager import ExchangeRate, ExchangeRateManager, FEEDS, MarketFeed
|
|
|
|
|
|
class TestExchangeRateManager(AsyncioTestCase):
|
|
async def test_exchange_rate_manager(self):
|
|
manager = ExchangeRateManager(FEEDS)
|
|
manager.start()
|
|
self.addCleanup(manager.stop)
|
|
for feed in manager.market_feeds:
|
|
self.assertFalse(feed.is_online)
|
|
self.assertIsNone(feed.rate)
|
|
await manager.wait()
|
|
failures = set()
|
|
for feed in manager.market_feeds:
|
|
if feed.is_online:
|
|
self.assertIsInstance(feed.rate, ExchangeRate)
|
|
else:
|
|
failures.add(feed.name)
|
|
self.assertFalse(feed.has_rate)
|
|
self.assertLessEqual(len(failures), 1, f"feed failures: {failures}. Please check exchange rate feeds!")
|
|
lbc = manager.convert_currency('USD', 'LBC', Decimal('1.0'))
|
|
self.assertGreaterEqual(lbc, 2.0)
|
|
self.assertLessEqual(lbc, 80.0)
|
|
lbc = manager.convert_currency('BTC', 'LBC', Decimal('0.01'))
|
|
self.assertGreaterEqual(lbc, 1_000)
|
|
self.assertLessEqual(lbc, 20_000)
|
|
|
|
async def test_it_handles_feed_being_offline(self):
|
|
class FakeFeed(MarketFeed):
|
|
name = "fake"
|
|
url = "http://impossi.bru"
|
|
manager = ExchangeRateManager((FakeFeed,))
|
|
manager.start()
|
|
self.addCleanup(manager.stop)
|
|
for feed in manager.market_feeds:
|
|
self.assertFalse(feed.is_online)
|
|
self.assertIsNone(feed.rate)
|
|
await asyncio.wait_for(manager.wait(), 2)
|
|
for feed in manager.market_feeds:
|
|
self.assertFalse(feed.is_online)
|
|
self.assertFalse(feed.has_rate)
|