def setUp(self):
self.btcd = Mock()
self.sut = BitcoindFactory(self.btcd)
class TestBitcoindFactory(TestCase):
def setUp(self):
self.btcd = Mock()
self.sut = BitcoindFactory(self.btcd)
def tearDown(self):
self.btcd.reset_mock()
def test_get_mempool_transactions(self):
response = {"result": ["txid1", "txid2", "txid3"]}
self.btcd.get_raw_mempool.side_effect = [response]
i, limit = 0, 2
r = self.sut.get_mempool_transactions(limit=limit)
self.assertIsInstance(r, GeneratorType)
self.btcd.get_raw_mempool.assert_called_once_with()
for i, x in enumerate(r):
self.assertIsInstance(x, BTCDTransaction)
self.assertEqual(i+1, limit)
def test_get_transactions(self):
request = ["txid1", "txid2", "txid3"]
get_raw_transactions_response = [{"result": "rawtx1"},
{"result": "rawtx2"},
{"result": "rawtx3"}]
decode_raw_transactions_response = [{"result": {"txid": "hash1"}},
{"result": {"txid": "hash2"}},
{"result": {"txid": "hash3"}}]
i = 0
self.btcd.get_raw_transaction.side_effect = get_raw_transactions_response
self.btcd.decode_raw_transaction.side_effect = decode_raw_transactions_response
r = self.sut.get_transactions(request)
self.assertIsInstance(r, GeneratorType)
for i, x in enumerate(r):
self.assertIsInstance(x, BTCDTransaction)
self.assertEqual(x.txid, decode_raw_transactions_response[i]['result']['txid'])
self.assertEqual(i+1, len(request))
def test_generate_blocks_from_height_explicit_stop_with_limit(self):
i, limit = 0, 3
get_block_hash_response = [{"result": "block_hash_1"},
{"result": "block_hash_2"},
{"result": "block_hash_3"}]
get_block_response = [{"result": {'nextblockhash': 'block_hash_2',
'hash': 'block_hash_1'}},
{"result":{'nextblockhash': 'block_hash_3',
'hash': 'block_hash_2'}},
{"result":{'nextblockhash': 'block_hash_4',
'hash': 'block_hash_3'}}]
self.btcd.get_block_hash.side_effect = get_block_hash_response
self.btcd.get_block.side_effect = get_block_response
blocks = self.sut.generate_blocks(blockheight=1, max_iterations=limit)
for i, block in enumerate(blocks):
self.assertIsInstance(block, BTCDBlock)
self.assertEqual(block.hash, get_block_response[i]['result']['hash'])
self.assertEqual(i+1, limit)
def test_generate_blocks_from_height_0_explicit_stop_with_limit(self):
i, limit = 0, 3
get_block_hash_response = [{"result": "block_hash_1"},
{"result":"block_hash_2"},
{"result": "block_hash_3"}]
get_block_response = [{"result":{'nextblockhash': 'block_hash_2',
'hash': 'block_hash_1'}},
{"result":{'nextblockhash': 'block_hash_3',
'hash': 'block_hash_2'}},
{"result":{'nextblockhash': 'block_hash_4',
'hash': 'block_hash_3'}}]
self.btcd.get_block_hash.side_effect = get_block_hash_response
self.btcd.get_block.side_effect = get_block_response
blocks = self.sut.generate_blocks(blockheight=0, max_iterations=limit)
for i, block in enumerate(blocks):
self.assertIsInstance(block, BTCDBlock)
self.assertEqual(block.hash, get_block_response[i]['result']['hash'])
self.assertEqual(i+1, limit)
def test_generate_blocks_from_hash_explicit_stop_with_limit(self):
i, limit = 0, 3
get_block_response = [{"result":{'nextblockhash': 'block_hash_2',
'hash': 'block_hash_1'}},
{"result":{'nextblockhash': 'block_hash_3',
'hash': 'block_hash_2'}},
{"result":{'nextblockhash': 'block_hash_4',
'hash': 'block_hash_3'}}]
self.btcd.get_block.side_effect = get_block_response
blocks = self.sut.generate_blocks(blockhash='block_hash_1', max_iterations=limit)
for i, block in enumerate(blocks):
self.assertIsInstance(block, BTCDBlock)
self.assertEqual(block.hash, get_block_response[i]['result']['hash'])
self.assertEqual(i+1, limit)
def test_generate_blocks_from_height_explicit_natural_stop(self):
get_block_hash_response = [{"result": "block_hash_1"},
{"result": "block_hash_2"},
{"result": "block_hash_3"}]
get_block_response = [{"result": {'nextblockhash': 'block_hash_2',
'hash': 'block_hash_1'}},
{"result": {'nextblockhash': 'block_hash_3',
'hash': 'block_hash_2'}},
{"result": {'nextblockhash': None,
'hash': 'block_hash_3'}}]
self.btcd.get_block_hash.side_effect = get_block_hash_response
self.btcd.get_block.side_effect = get_block_response
blocks = self.sut.generate_blocks(blockheight=1)
i = 0
for i, block in enumerate(blocks):
self.assertIsInstance(block, BTCDBlock)
self.assertEqual(block.hash, get_block_response[i]['result']['hash'])
self.assertEqual(i+1, 3)
def test_generate_blocks_from_height_explicit_stop_height(self):
get_block_hash_response = [{"result": "block_hash_1"},
{"result": "block_hash_2"},
{"result": "block_hash_3"}]
get_block_response = [{"result": {'nextblockhash': 'block_hash_2', 'height': 1,
'hash': 'block_hash_1'}},
{"result":{'nextblockhash': 'block_hash_3', 'height': 2,
'hash': 'block_hash_2'}},
{"result":{'nextblockhash': 'block_hash_4', 'height': 3,
'hash': 'block_hash_3'}}]
self.btcd.get_block_hash.side_effect = get_block_hash_response
self.btcd.get_block.side_effect = get_block_response
blocks = self.sut.generate_blocks(blockheight=1, stop_blockheight=3)
i = 0
for i, block in enumerate(blocks):
self.assertIsInstance(block, BTCDBlock)
self.assertEqual(block.hash, get_block_response[i]['result']['hash'])
self.assertEqual(i+1, 3)
def test_blocks_generator_multiple_starts(self):
with self.assertRaises(ValueError):
self.sut.generate_blocks(blockhash='cafe', blockheight=1)
def test_blocks_generator_no_starts(self):
with self.assertRaises(ValueError):
self.sut.generate_blocks()
def test_blocks_generator_multiple_stops_height_and_iterations(self):
with self.assertRaises(ValueError):
self.sut.generate_blocks(max_iterations=1, stop_blockheight=10)
def test_blocks_generator_multiple_stops_hash_and_iterations(self):
with self.assertRaises(ValueError):
self.sut.generate_blocks(max_iterations=1, stop_blockhash='cafe')
def test_blocks_generator_multiple_stops_hash_and_height(self):
with self.assertRaises(ValueError):
self.sut.generate_blocks(stop_blockhash=1, stop_blockheight=10)
