refactored integration folder as well

This commit is contained in:
StrikerRUS 2019-10-08 18:01:38 +03:00 committed by Lex Berezhny
parent 62e3258dda
commit eb8a88dbd6
2 changed files with 4 additions and 4 deletions

View file

@ -18,7 +18,7 @@ class NetworkTests(IntegrationTestCase):
self.assertEqual(self.ledger.network.remote_height, initial_height + 1) self.assertEqual(self.ledger.network.remote_height, initial_height + 1)
async def test_server_features(self): async def test_server_features(self):
self.assertEqual({ self.assertDictEqual({
'genesis_hash': self.conductor.spv_node.coin_class.GENESIS_HASH, 'genesis_hash': self.conductor.spv_node.coin_class.GENESIS_HASH,
'hash_function': 'sha256', 'hash_function': 'sha256',
'hosts': {}, 'hosts': {},
@ -37,7 +37,7 @@ class NetworkTests(IntegrationTestCase):
'DAILY_FEE': '42'}) 'DAILY_FEE': '42'})
await self.conductor.spv_node.start(self.conductor.blockchain_node) await self.conductor.spv_node.start(self.conductor.blockchain_node)
await self.ledger.network.on_connected.first await self.ledger.network.on_connected.first
self.assertEqual({ self.assertDictEqual({
'genesis_hash': self.conductor.spv_node.coin_class.GENESIS_HASH, 'genesis_hash': self.conductor.spv_node.coin_class.GENESIS_HASH,
'hash_function': 'sha256', 'hash_function': 'sha256',
'hosts': {}, 'hosts': {},
@ -162,7 +162,7 @@ class ServerPickingTestCase(AsyncioTestCase):
asyncio.ensure_future(network.start()) asyncio.ensure_future(network.start())
await asyncio.wait_for(network.on_connected.first, timeout=1) await asyncio.wait_for(network.on_connected.first, timeout=1)
self.assertTrue(network.is_connected) self.assertTrue(network.is_connected)
self.assertEqual(network.client.server, ('127.0.0.1', 1337)) self.assertTupleEqual(network.client.server, ('127.0.0.1', 1337))
self.assertTrue(all([not session.is_closing() for session in network.session_pool.available_sessions])) self.assertTrue(all([not session.is_closing() for session in network.session_pool.available_sessions]))
# ensure we are connected to all of them after a while # ensure we are connected to all of them after a while
await asyncio.sleep(1) await asyncio.sleep(1)

View file

@ -148,7 +148,7 @@ class BasicTransactionTests(IntegrationTestCase):
return summary return summary
self.conductor.spv_node.server.mempool.transaction_summaries = random_summary self.conductor.spv_node.server.mempool.transaction_summaries = random_summary
# 10 unconfirmed txs, all from blockchain wallet # 10 unconfirmed txs, all from blockchain wallet
sends = list(self.blockchain.send_to_address(address, 10) for _ in range(10)) sends = [self.blockchain.send_to_address(address, 10) for _ in range(10)]
# use batching to reduce issues with send_to_address on cli # use batching to reduce issues with send_to_address on cli
for batch in range(0, len(sends), 10): for batch in range(0, len(sends), 10):
txids = await asyncio.gather(*sends[batch:batch + 10]) txids = await asyncio.gather(*sends[batch:batch + 10])