lbry-sdk/lbry/tests/integration/test_purchase_command.py

185 lines
8.5 KiB
Python
Raw Normal View History

from typing import Optional
from lbry.testcase import CommandTestCase
from lbry.schema.purchase import Purchase
from lbry.wallet.transaction import Transaction
from lbry.wallet.dewies import lbc_to_dewies, dewies_to_lbc
class PurchaseCommandTests(CommandTestCase):
async def asyncSetUp(self):
await super().asyncSetUp()
self.merchant_address = await self.blockchain.get_raw_change_address()
async def priced_stream(self, name='stream', price: Optional[str] = '2.0', currency='LBC') -> Transaction:
kwargs = {}
if price and currency:
kwargs = {
'fee_amount': price,
'fee_currency': currency,
'fee_address': self.merchant_address
}
file_path = self.create_upload_file(data=b'high value content')
tx = await self.daemon.jsonrpc_stream_create(
name, '0.01', file_path=file_path, **kwargs
)
await self.ledger.wait(tx)
await self.generate(1)
await self.ledger.wait(tx)
await self.daemon.jsonrpc_file_delete(claim_name=name)
return tx
async def create_purchase(self, name, price):
stream = await self.priced_stream(name, price)
claim_id = stream.outputs[0].claim_id
purchase = await self.daemon.jsonrpc_purchase_create(claim_id)
await self.ledger.wait(purchase)
return claim_id
2019-12-20 02:34:10 +01:00
async def assertStreamPurchased(self, stream: Transaction, operation):
await self.account.release_all_outputs()
buyer_balance = await self.account.get_balance()
merchant_balance = lbc_to_dewies(str(await self.blockchain.get_balance()))
pre_purchase_count = (await self.daemon.jsonrpc_purchase_list())['total_items']
2019-12-20 02:34:10 +01:00
purchase = await operation()
stream_txo, purchase_txo = stream.outputs[0], purchase.outputs[0]
stream_fee = stream_txo.claim.stream.fee
self.assertEqual(stream_fee.dewies, purchase_txo.amount)
self.assertEqual(stream_fee.address, purchase_txo.get_address(self.ledger))
await self.ledger.wait(purchase)
await self.generate(1)
merchant_balance += lbc_to_dewies('1.0') # block reward
await self.ledger.wait(purchase)
self.assertEqual(
await self.account.get_balance(), buyer_balance - (purchase.input_sum-purchase.outputs[2].amount))
self.assertEqual(
str(await self.blockchain.get_balance()), dewies_to_lbc(merchant_balance + purchase_txo.amount))
purchases = await self.daemon.jsonrpc_purchase_list()
self.assertEqual(purchases['total_items'], pre_purchase_count+1)
tx = purchases['items'][0].tx_ref.tx
self.assertEqual(len(tx.outputs), 3) # purchase txo, purchase data, change
txo0 = tx.outputs[0]
txo1 = tx.outputs[1]
self.assertEqual(txo0.purchase, txo1) # purchase txo has reference to purchase data
self.assertTrue(txo1.is_purchase_data)
self.assertTrue(txo1.can_decode_purchase_data)
self.assertIsInstance(txo1.purchase_data, Purchase)
self.assertEqual(txo1.purchase_data.claim_id, stream_txo.claim_id)
async def test_purchasing(self):
stream = await self.priced_stream()
claim_id = stream.outputs[0].claim_id
# explicit purchase of claim
2019-12-20 02:58:36 +01:00
await self.assertStreamPurchased(stream, lambda: self.daemon.jsonrpc_purchase_create(claim_id))
# check that `get` doesn't purchase it again
balance = await self.account.get_balance()
response = await self.daemon.jsonrpc_get('lbry://stream')
self.assertIsNone(response.content_fee)
self.assertEqual(await self.account.get_balance(), balance)
self.assertItemCount(await self.daemon.jsonrpc_purchase_list(), 1)
# `get` does purchase a stream we don't have yet
another_stream = await self.priced_stream('another')
2019-12-20 02:34:10 +01:00
async def imagine_its_a_lambda():
response = await self.daemon.jsonrpc_get('lbry://another')
return response.content_fee
await self.assertStreamPurchased(another_stream, imagine_its_a_lambda)
# purchase non-existent claim fails
with self.assertRaisesRegex(Exception, "Could not find claim with claim_id"):
await self.daemon.jsonrpc_purchase_create('abc123')
# purchase stream with no price fails
no_price_stream = await self.priced_stream('no_price_stream', price=None)
with self.assertRaisesRegex(Exception, "does not have a purchase price"):
await self.daemon.jsonrpc_purchase_create(no_price_stream.outputs[0].claim_id)
# purchase claim you already own fails
with self.assertRaisesRegex(Exception, "You already have a purchase for claim_id"):
await self.daemon.jsonrpc_purchase_create(claim_id)
# force purchasing claim you already own
2019-12-20 02:58:36 +01:00
await self.assertStreamPurchased(
stream, lambda: self.daemon.jsonrpc_purchase_create(claim_id, allow_duplicate_purchase=True)
)
# purchase by uri
abc_stream = await self.priced_stream('abc')
2019-12-20 02:58:36 +01:00
await self.assertStreamPurchased(abc_stream, lambda: self.daemon.jsonrpc_purchase_create(url='lbry://abc'))
async def test_purchase_and_transaction_list(self):
self.assertItemCount(await self.daemon.jsonrpc_purchase_list(), 0)
self.assertItemCount(await self.daemon.jsonrpc_transaction_list(), 1)
claim_id1 = await self.create_purchase('a', '1.0')
claim_id2 = await self.create_purchase('b', '1.0')
result = await self.out(self.daemon.jsonrpc_purchase_list())
self.assertItemCount(await self.daemon.jsonrpc_transaction_list(), 5)
self.assertItemCount(result, 2)
self.assertEqual(result['items'][0]['type'], 'purchase')
self.assertEqual(result['items'][0]['claim_id'], claim_id2)
self.assertNotIn('claim', result['items'][0])
self.assertEqual(result['items'][1]['type'], 'purchase')
self.assertEqual(result['items'][1]['claim_id'], claim_id1)
self.assertNotIn('claim', result['items'][1])
result = await self.out(self.daemon.jsonrpc_purchase_list(resolve=True))
self.assertEqual(result['items'][0]['claim']['name'], 'b')
self.assertEqual(result['items'][1]['claim']['name'], 'a')
result = await self.daemon.jsonrpc_transaction_list()
self.assertEqual(result['items'][0]['purchase_info'][0]['claim_id'], claim_id2)
self.assertEqual(result['items'][2]['purchase_info'][0]['claim_id'], claim_id1)
result = await self.claim_search()
self.assertEqual(result[0]['claim_id'], result[0]['purchase_receipt']['claim_id'])
self.assertEqual(result[1]['claim_id'], result[1]['purchase_receipt']['claim_id'])
url = result[0]['canonical_url']
resolve = await self.resolve(url)
self.assertEqual(result[0]['claim_id'], resolve[url]['purchase_receipt']['claim_id'])
self.assertItemCount(await self.daemon.jsonrpc_file_list(), 0)
await self.daemon.jsonrpc_get('lbry://a')
await self.daemon.jsonrpc_get('lbry://b')
files = await self.file_list()
self.assertEqual(files[0]['claim_id'], files[0]['purchase_receipt']['claim_id'])
self.assertEqual(files[1]['claim_id'], files[1]['purchase_receipt']['claim_id'])
async def test_seller_can_spend_received_purchase_funds(self):
self.merchant_address = await self.account.receiving.get_or_create_usable_address()
daemon2 = await self.add_daemon()
address2 = await daemon2.wallet_manager.default_account.receiving.get_or_create_usable_address()
sendtxid = await self.blockchain.send_to_address(address2, 2)
await self.confirm_tx(sendtxid, daemon2.ledger)
stream = await self.priced_stream('a', '1.0')
await self.assertBalance(self.account, '9.987893')
self.assertItemCount(await self.daemon.jsonrpc_utxo_list(), 1)
purchase = await daemon2.jsonrpc_purchase_create(stream.outputs[0].claim_id)
await self.ledger.wait(purchase)
await self.generate(1)
await self.ledger.wait(purchase)
await self.assertBalance(self.account, '10.987893')
self.assertItemCount(await self.daemon.jsonrpc_utxo_list(), 2)
spend = await self.daemon.jsonrpc_wallet_send('10.5', address2)
await self.ledger.wait(spend)
await self.generate(1)
await self.ledger.wait(spend)
await self.assertBalance(self.account, '0.487695')
self.assertItemCount(await self.daemon.jsonrpc_utxo_list(), 1)