forked from LBRYCommunity/lbry-sdk
128 lines
5 KiB
Python
128 lines
5 KiB
Python
import mock
|
|
import json
|
|
import unittest
|
|
|
|
from twisted.internet import defer
|
|
from twisted import trial
|
|
|
|
from lbryschema.decode import smart_decode
|
|
from lbryum.wallet import NewWallet
|
|
from lbrynet import conf
|
|
from lbrynet.core import Session, PaymentRateManager, Wallet
|
|
from lbrynet.database.storage import SQLiteStorage
|
|
from lbrynet.daemon.Daemon import Daemon as LBRYDaemon
|
|
|
|
from lbrynet.tests import util
|
|
from lbrynet.tests.mocks import mock_conf_settings, FakeNetwork
|
|
from lbrynet.tests.mocks import BlobAvailabilityTracker as DummyBlobAvailabilityTracker
|
|
from lbrynet.tests.mocks import ExchangeRateManager as DummyExchangeRateManager
|
|
from lbrynet.tests.mocks import BTCLBCFeed, USDBTCFeed
|
|
from lbrynet.tests.util import is_android
|
|
|
|
|
|
import logging
|
|
logging.getLogger("lbryum").setLevel(logging.WARNING)
|
|
|
|
|
|
def get_test_daemon(data_rate=None, generous=True, with_fee=False):
|
|
if data_rate is None:
|
|
data_rate = conf.ADJUSTABLE_SETTINGS['data_rate'][1]
|
|
|
|
rates = {
|
|
'BTCLBC': {'spot': 3.0, 'ts': util.DEFAULT_ISO_TIME + 1},
|
|
'USDBTC': {'spot': 2.0, 'ts': util.DEFAULT_ISO_TIME + 2}
|
|
}
|
|
daemon = LBRYDaemon(None)
|
|
daemon.session = mock.Mock(spec=Session.Session)
|
|
daemon.session.wallet = mock.Mock(spec=Wallet.LBRYumWallet)
|
|
daemon.session.wallet.wallet = mock.Mock(spec=NewWallet)
|
|
daemon.session.wallet.wallet.use_encryption = False
|
|
daemon.session.wallet.network = FakeNetwork()
|
|
daemon.session.storage = mock.Mock(spec=SQLiteStorage)
|
|
market_feeds = [BTCLBCFeed(), USDBTCFeed()]
|
|
daemon.exchange_rate_manager = DummyExchangeRateManager(market_feeds, rates)
|
|
base_prm = PaymentRateManager.BasePaymentRateManager(rate=data_rate)
|
|
prm = PaymentRateManager.NegotiatedPaymentRateManager(base_prm, DummyBlobAvailabilityTracker(),
|
|
generous=generous)
|
|
daemon.session.payment_rate_manager = prm
|
|
|
|
metadata = {
|
|
"author": "fake author",
|
|
"language": "en",
|
|
"content_type": "fake/format",
|
|
"description": "fake description",
|
|
"license": "fake license",
|
|
"license_url": "fake license url",
|
|
"nsfw": False,
|
|
"sources": {
|
|
"lbry_sd_hash": 'd2b8b6e907dde95245fe6d144d16c2fdd60c4e0c6463ec98'
|
|
'b85642d06d8e9414e8fcfdcb7cb13532ec5454fb8fe7f280'
|
|
},
|
|
"thumbnail": "fake thumbnail",
|
|
"title": "fake title",
|
|
"ver": "0.0.3"
|
|
}
|
|
if with_fee:
|
|
metadata.update(
|
|
{"fee": {"USD": {"address": "bQ6BGboPV2SpTMEP7wLNiAcnsZiH8ye6eA", "amount": 0.75}}})
|
|
daemon._resolve_name = lambda _: defer.succeed(metadata)
|
|
migrated = smart_decode(json.dumps(metadata))
|
|
daemon.session.wallet.resolve = lambda *_: defer.succeed(
|
|
{"test": {'claim': {'value': migrated.claim_dict}}})
|
|
return daemon
|
|
|
|
|
|
class TestCostEst(trial.unittest.TestCase):
|
|
def setUp(self):
|
|
mock_conf_settings(self)
|
|
util.resetTime(self)
|
|
|
|
def test_fee_and_generous_data(self):
|
|
size = 10000000
|
|
correct_result = 4.5
|
|
daemon = get_test_daemon(generous=True, with_fee=True)
|
|
self.assertEquals(daemon.get_est_cost("test", size).result, correct_result)
|
|
|
|
def test_fee_and_ungenerous_data(self):
|
|
size = 10000000
|
|
fake_fee_amount = 4.5
|
|
data_rate = conf.ADJUSTABLE_SETTINGS['data_rate'][1]
|
|
correct_result = size / 10 ** 6 * data_rate + fake_fee_amount
|
|
daemon = get_test_daemon(generous=False, with_fee=True)
|
|
self.assertEquals(daemon.get_est_cost("test", size).result, correct_result)
|
|
|
|
def test_generous_data_and_no_fee(self):
|
|
size = 10000000
|
|
correct_result = 0.0
|
|
daemon = get_test_daemon(generous=True)
|
|
self.assertEquals(daemon.get_est_cost("test", size).result, correct_result)
|
|
|
|
def test_ungenerous_data_and_no_fee(self):
|
|
size = 10000000
|
|
data_rate = conf.ADJUSTABLE_SETTINGS['data_rate'][1]
|
|
correct_result = size / 10 ** 6 * data_rate
|
|
daemon = get_test_daemon(generous=False)
|
|
self.assertEquals(daemon.get_est_cost("test", size).result, correct_result)
|
|
|
|
|
|
class TestJsonRpc(trial.unittest.TestCase):
|
|
def setUp(self):
|
|
def noop():
|
|
return None
|
|
|
|
mock_conf_settings(self)
|
|
util.resetTime(self)
|
|
self.test_daemon = get_test_daemon()
|
|
self.test_daemon.session.wallet.is_first_run = False
|
|
self.test_daemon.session.wallet.get_best_blockhash = noop
|
|
|
|
def test_status(self):
|
|
d = defer.maybeDeferred(self.test_daemon.jsonrpc_status)
|
|
d.addCallback(lambda status: self.assertDictContainsSubset({'is_running': False}, status))
|
|
|
|
@unittest.skipIf(is_android(),
|
|
'Test cannot pass on Android because PYTHONOPTIMIZE removes the docstrings.')
|
|
def test_help(self):
|
|
d = defer.maybeDeferred(self.test_daemon.jsonrpc_help, command='status')
|
|
d.addCallback(lambda result: self.assertSubstring('daemon status', result['help']))
|
|
# self.assertSubstring('daemon status', d.result)
|