stress test

This commit is contained in:
Lex Berezhny 2018-12-05 23:45:35 -05:00
parent 367ffdea1f
commit b2544139e6
3 changed files with 27 additions and 27 deletions

View file

@ -6,7 +6,7 @@ from torba.client.constants import CENT
class SyncTests(IntegrationTestCase): class SyncTests(IntegrationTestCase):
VERBOSITY = logging.INFO VERBOSITY = logging.WARN
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)

View file

@ -1,13 +1,10 @@
import logging import logging
import asyncio import asyncio
from itertools import chain
from torba.testcase import IntegrationTestCase from torba.testcase import IntegrationTestCase
from torba.client.constants import COIN from torba.client.constants import COIN
log = logging.getLogger(__name__)
log.setLevel(logging.INFO)
class BasicTransactionTests(IntegrationTestCase): class BasicTransactionTests(IntegrationTestCase):
VERBOSITY = logging.WARN VERBOSITY = logging.WARN
@ -15,34 +12,37 @@ class BasicTransactionTests(IntegrationTestCase):
async def test_stressing(self): async def test_stressing(self):
await self.blockchain.generate(1000) await self.blockchain.generate(1000)
await self.assertBalance(self.account, '0.0') await self.assertBalance(self.account, '0.0')
address1 = await self.account.receiving.get_or_create_usable_address() addresses = await self.account.receiving.get_addresses()
hash1 = self.ledger.address_to_hash160(address1)
txids = await asyncio.gather(*( sends = list(chain(
self.blockchain.send_to_address(address1, 100) (self.blockchain.send_to_address(address, 10) for address in addresses),
for _ in range(10) (self.blockchain.send_to_address(addresses[-1], 10) for _ in range(10))
)) ))
for batch in range(0, len(sends), 10):
txids = await asyncio.gather(*sends[batch:batch+10])
await asyncio.wait([ await asyncio.wait([
self.on_transaction_id(txid) self.on_transaction_id(txid) for txid in txids
for txid in txids
]) ])
await self.assertBalance(self.account, '1000.0') await self.assertBalance(self.account, '300.0')
addresses = await self.account.receiving.get_addresses()
self.assertEqual(40, len(addresses))
tasks = [] await self.blockchain.generate(1)
for _ in range(10):
self.assertEqual(30, await self.account.get_utxo_count())
hash1 = self.ledger.address_to_hash160(addresses[-1])
tx = await self.ledger.transaction_class.create( tx = await self.ledger.transaction_class.create(
[], [],
[self.ledger.transaction_class.output_class.pay_pubkey_hash(1*COIN, hash1)], [self.ledger.transaction_class.output_class.pay_pubkey_hash(299*COIN, hash1)],
[self.account], self.account [self.account], self.account
) )
await self.broadcast(tx) await self.broadcast(tx)
tasks.append(asyncio.create_task(self.ledger.wait(tx))) await self.ledger.wait(tx)
await asyncio.wait(tasks) self.assertEqual(2, await self.account.get_utxo_count()) # 299 + change
await self.assertBalance(self.account, '999.99876')
async def test_sending_and_receiving(self): async def test_sending_and_receiving(self):
account1, account2 = self.account, self.wallet.generate_account(self.ledger) account1, account2 = self.account, self.wallet.generate_account(self.ledger)

View file

@ -49,7 +49,7 @@ def get_blockchain_node_from_ledger(ledger_module):
def set_logging(ledger_module, level): def set_logging(ledger_module, level):
logging.getLogger('torba').setLevel(level) logging.getLogger('torba').setLevel(level)
logging.getLogger('torba.client').setLevel(logging.INFO) logging.getLogger('torba.client').setLevel(level)
logging.getLogger('torba.server').setLevel(level) logging.getLogger('torba.server').setLevel(level)
#logging.getLogger('asyncio').setLevel(level) #logging.getLogger('asyncio').setLevel(level)
logging.getLogger('blockchain').setLevel(level) logging.getLogger('blockchain').setLevel(level)