cleanup integration test

This commit is contained in:
Lex Berezhny 2018-06-26 20:22:35 -04:00
parent 1dfc18683d
commit da265198f1

View file

@ -1,5 +1,5 @@
import asyncio
from orchstr8.testcase import IntegrationTestCase
from orchstr8.testcase import IntegrationTestCase, d2f
from torba.constants import COIN
@ -10,45 +10,30 @@ class BasicTransactionTests(IntegrationTestCase):
async def test_sending_and_receiving(self):
account1, account2 = self.account, self.wallet.generate_account(self.ledger)
await account1.ensure_address_gap().asFuture(asyncio.get_event_loop())
self.assertEqual(await self.get_balance(account1), 0)
self.assertEqual(await self.get_balance(account2), 0)
address = await account1.receiving.get_or_create_usable_address().asFuture(asyncio.get_event_loop())
sendtxid = await self.blockchain.send_to_address(address.decode(), 5.5)
await self.on_transaction_id(sendtxid) #mempool
address1 = await d2f(account1.receiving.get_or_create_usable_address())
sendtxid = await self.blockchain.send_to_address(address1.decode(), 5.5)
await self.on_transaction_id(sendtxid) # mempool
await self.blockchain.generate(1)
await self.on_transaction_id(sendtxid) #confirmed
await self.on_transaction_id(sendtxid) # confirmed
self.assertEqual(await self.get_balance(account1), int(5.5*COIN))
self.assertEqual(await self.get_balance(account2), 0)
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 5.5)
self.assertEqual(round(await self.get_balance(account2)/COIN, 1), 0)
address = await account2.receiving.get_or_create_usable_address().asFuture(asyncio.get_event_loop())
hash1 = self.ledger.address_to_hash160(address)
tx = await self.ledger.transaction_class.pay(
[self.ledger.transaction_class.output_class.pay_pubkey_hash(2*COIN, hash1)],
address2 = await d2f(account2.receiving.get_or_create_usable_address())
hash2 = self.ledger.address_to_hash160(address2)
tx = await d2f(self.ledger.transaction_class.pay(
[self.ledger.transaction_class.output_class.pay_pubkey_hash(2*COIN, hash2)],
[account1], account1
).asFuture(asyncio.get_event_loop())
))
await self.broadcast(tx)
await self.on_transaction(tx) #mempool
tx2 = await self.ledger.transaction_class.pay(
[self.ledger.transaction_class.output_class.pay_pubkey_hash(1*COIN, hash1)],
[account1], account1
).asFuture(asyncio.get_event_loop())
await self.broadcast(tx2)
await self.on_transaction(tx2) #mempool
await self.on_transaction(tx) # mempool
await self.blockchain.generate(1)
await asyncio.wait([
self.on_header(202),
self.on_transaction(tx),
self.on_transaction(tx2),
])
await self.on_transaction(tx) # confirmed
#self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 3.5)
#self.assertEqual(round(await self.get_balance(account2)/COIN, 1), 2.0)
self.assertTrue(await d2f(self.ledger.is_valid_transaction(tx, 202)))
self.assertTrue(await self.ledger.is_valid_transaction(tx, 202).asFuture(asyncio.get_event_loop()))
self.assertTrue(await self.ledger.is_valid_transaction(tx2, 202).asFuture(asyncio.get_event_loop()))
self.assertEqual(round(await self.get_balance(account1)/COIN, 1), 3.5)
self.assertEqual(round(await self.get_balance(account2)/COIN, 1), 2.0)