forked from LBRYCommunity/lbry-sdk
86 lines
3.3 KiB
Python
86 lines
3.3 KiB
Python
|
from lbrynet.metadata import Fee
|
||
|
|
from lbrynet.lbrynet_daemon import ExchangeRateManager
|
||
|
|
from lbrynet import conf
|
||
|
|
from lbrynet.core.Error import InvalidExchangeRateResponse
|
||
|
|
|
||
|
|
from twisted.trial import unittest
|
||
|
|
from twisted.internet import defer
|
||
|
|
from tests import util
|
||
|
|
|
||
|
|
|
||
|
|
class FeeFormatTest(unittest.TestCase):
|
||
|
|
def test_fee_created_with_correct_inputs(self):
|
||
|
|
fee_dict = {
|
||
|
|
'USD': {
|
||
|
|
'amount': 10.0,
|
||
|
|
'address': "bRcHraa8bYJZL7vkh5sNmGwPDERFUjGPP9"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
fee = Fee.FeeValidator(fee_dict)
|
||
|
|
self.assertEqual(10.0, fee['USD']['amount'])
|
||
|
|
|
||
|
|
|
||
|
|
class FeeTest(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
util.resetTime(self)
|
||
|
|
|
||
|
|
def test_fee_converts_to_lbc(self):
|
||
|
|
fee_dict = {
|
||
|
|
'USD': {
|
||
|
|
'amount': 10.0,
|
||
|
|
'address': "bRcHraa8bYJZL7vkh5sNmGwPDERFUjGPP9"
|
||
|
|
}
|
||
|
|
}
|
||
|
|
rates = {
|
||
|
|
'BTCLBC': {'spot': 3.0, 'ts': util.DEFAULT_ISO_TIME + 1},
|
||
|
|
'USDBTC': {'spot': 2.0, 'ts': util.DEFAULT_ISO_TIME + 2}
|
||
|
|
}
|
||
|
|
manager = ExchangeRateManager.DummyExchangeRateManager(rates)
|
||
|
|
result = manager.to_lbc(fee_dict).amount
|
||
|
|
self.assertEqual(60.0, result)
|
||
|
|
|
||
|
|
class GoogleBTCFeedTest(unittest.TestCase):
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_handle_response(self):
|
||
|
|
feed = ExchangeRateManager.GoogleBTCFeed()
|
||
|
|
|
||
|
|
response = '// [ { "id": "-2001" ,"t" : "USDBTC" ,"e" : "CURRENCY" ,"l" : "0.0008" ,"l_fix" : "" ,"l_cur" : "" ,"s": "0" ,"ltt":"" ,"lt" : "Feb 27, 10:21PM GMT" ,"lt_dts" : "2017-02-27T22:21:39Z" ,"c" : "-0.00001" ,"c_fix" : "" ,"cp" : "-0.917" ,"cp_fix" : "" ,"ccol" : "chr" ,"pcls_fix" : "" } ]'
|
||
|
|
out = yield feed._handle_response(response)
|
||
|
|
self.assertEqual(0.0008, out)
|
||
|
|
|
||
|
|
# check negative trade price throws exception
|
||
|
|
response = '// [ { "id": "-2001" ,"t" : "USDBTC" ,"e" : "CURRENCY" ,"l" : "-0.0008" ,"l_fix" : "" ,"l_cur" : "" ,"s": "0" ,"ltt":"" ,"lt" : "Feb 27, 10:21PM GMT" ,"lt_dts" : "2017-02-27T22:21:39Z" ,"c" : "-0.00001" ,"c_fix" : "" ,"cp" : "-0.917" ,"cp_fix" : "" ,"ccol" : "chr" ,"pcls_fix" : "" } ]'
|
||
|
|
with self.assertRaises(InvalidExchangeRateResponse):
|
||
|
|
out = yield feed._handle_response(response)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class BittrexFeedTest(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
conf.initialize_settings()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
conf.settings = None
|
||
|
|
|
||
|
|
@defer.inlineCallbacks
|
||
|
|
def test_handle_response(self):
|
||
|
|
feed = ExchangeRateManager.BittrexFeed()
|
||
|
|
|
||
|
|
response ='{"success":true,"message":"","result":[{"Id":6902471,"TimeStamp":"2017-02-27T23:41:52.213","Quantity":56.12611239,"Price":0.00001621,"Total":0.00090980,"FillType":"PARTIAL_FILL","OrderType":"SELL"},{"Id":6902403,"TimeStamp":"2017-02-27T23:31:40.463","Quantity":430.99988180,"Price":0.00001592,"Total":0.00686151,"FillType":"PARTIAL_FILL","OrderType":"SELL"}]}'
|
||
|
|
out = yield feed._handle_response(response)
|
||
|
|
expected= 1.0 / ((0.00090980+0.00686151) / (56.12611239+430.99988180))
|
||
|
|
self.assertEqual(expected, out)
|
||
|
|
|
||
|
|
response='{}'
|
||
|
|
with self.assertRaises(InvalidExchangeRateResponse):
|
||
|
|
out = yield feed._handle_response(response)
|
||
|
|
|
||
|
|
response='{"success":true,"result":[]}'
|
||
|
|
with self.assertRaises(InvalidExchangeRateResponse):
|
||
|
|
out = yield feed._handle_response(response)
|
||
|
|
|
||
|
|
|
||
|
|
|