lbry-sdk/tests/client_tests/integration/test_transactions.py

102 lines
3.6 KiB
Python
Raw Normal View History

import logging
2018-07-01 23:20:17 +02:00
import asyncio
2018-11-04 06:55:50 +01:00
from torba.testcase import IntegrationTestCase
from torba.client.constants import COIN
2018-06-08 05:47:46 +02:00
2018-12-05 07:13:31 +01:00
log = logging.getLogger(__name__)
log.setLevel(logging.INFO)
2018-06-08 05:47:46 +02:00
class BasicTransactionTests(IntegrationTestCase):
2018-11-04 06:55:50 +01:00
VERBOSITY = logging.WARN
2018-06-08 05:47:46 +02:00
2018-12-05 07:13:31 +01:00
async def test_stressing(self):
await self.blockchain.generate(1000)
await self.assertBalance(self.account, '0.0')
address1 = await self.account.receiving.get_or_create_usable_address()
hash1 = self.ledger.address_to_hash160(address1)
txids = await asyncio.gather(*(
self.blockchain.send_to_address(address1, 100)
for _ in range(10)
))
await asyncio.wait([
self.on_transaction_id(txid)
for txid in txids
])
2018-12-05 07:13:31 +01:00
await self.assertBalance(self.account, '1000.0')
tasks = []
for _ in range(10):
tx = await self.ledger.transaction_class.create(
[],
[self.ledger.transaction_class.output_class.pay_pubkey_hash(1*COIN, hash1)],
[self.account], self.account
)
await self.broadcast(tx)
tasks.append(asyncio.create_task(self.ledger.wait(tx)))
await asyncio.wait(tasks)
await self.assertBalance(self.account, '999.99876')
2018-12-05 07:13:31 +01:00
2018-06-14 03:37:02 +02:00
async def test_sending_and_receiving(self):
2018-06-08 05:47:46 +02:00
account1, account2 = self.account, self.wallet.generate_account(self.ledger)
2018-11-19 04:54:00 +01:00
await self.ledger.subscribe_account(account2)
2018-06-08 05:47:46 +02:00
2018-11-19 04:54:00 +01:00
await self.assertBalance(account1, '0.0')
await self.assertBalance(account2, '0.0')
2018-06-08 05:47:46 +02:00
2018-07-01 23:20:17 +02:00
sendtxids = []
for i in range(5):
2018-10-15 04:16:51 +02:00
address1 = await account1.receiving.get_or_create_usable_address()
sendtxid = await self.blockchain.send_to_address(address1, 1.1)
2018-07-01 23:20:17 +02:00
sendtxids.append(sendtxid)
await self.on_transaction_id(sendtxid) # mempool
2018-06-12 16:02:04 +02:00
await self.blockchain.generate(1)
2018-07-01 23:20:17 +02:00
await asyncio.wait([ # confirmed
self.on_transaction_id(txid) for txid in sendtxids
])
2018-06-08 05:47:46 +02:00
2018-11-19 04:54:00 +01:00
await self.assertBalance(account1, '5.5')
await self.assertBalance(account2, '0.0')
2018-06-08 05:47:46 +02:00
2018-10-15 04:16:51 +02:00
address2 = await account2.receiving.get_or_create_usable_address()
2018-06-27 02:22:35 +02:00
hash2 = self.ledger.address_to_hash160(address2)
2018-10-15 04:16:51 +02:00
tx = await self.ledger.transaction_class.create(
[],
2018-06-27 02:22:35 +02:00
[self.ledger.transaction_class.output_class.pay_pubkey_hash(2*COIN, hash2)],
2018-06-12 16:02:04 +02:00
[account1], account1
2018-10-15 04:16:51 +02:00
)
2018-06-08 05:47:46 +02:00
await self.broadcast(tx)
await self.ledger.wait(tx) # mempool
2018-06-14 02:57:57 +02:00
await self.blockchain.generate(1)
await self.ledger.wait(tx) # confirmed
2018-06-08 05:47:46 +02:00
2018-11-19 04:54:00 +01:00
await self.assertBalance(account1, '3.499802')
await self.assertBalance(account2, '2.0')
2018-07-05 03:30:38 +02:00
2018-10-15 04:16:51 +02:00
utxos = await self.account.get_utxos()
tx = await self.ledger.transaction_class.create(
[self.ledger.transaction_class.input_class.spend(utxos[0])],
[],
[account1], account1
2018-10-15 04:16:51 +02:00
)
2018-07-10 04:02:18 +02:00
await self.broadcast(tx)
await self.ledger.wait(tx) # mempool
2018-07-10 04:02:18 +02:00
await self.blockchain.generate(1)
await self.ledger.wait(tx) # confirmed
2018-09-22 04:18:30 +02:00
2018-10-15 04:16:51 +02:00
txs = await account1.get_transactions()
2018-09-22 04:18:30 +02:00
tx = txs[1]
self.assertEqual(round(tx.inputs[0].txo_ref.txo.amount/COIN, 1), 1.1)
self.assertEqual(round(tx.inputs[1].txo_ref.txo.amount/COIN, 1), 1.1)
self.assertEqual(round(tx.outputs[0].amount/COIN, 1), 2.0)
self.assertEqual(tx.outputs[0].get_address(self.ledger), address2)
self.assertEqual(tx.outputs[0].is_change, False)
self.assertEqual(tx.outputs[1].is_change, True)