def test_merged_chain_without_transaction_conflicts(self): dag = Dag(0) block_hash, block_reward = ChainGenerator.insert_dummy_with_payments( dag, [dag.genesis_hash()], [], 1) payment1 = TransactionFactory.create_payment(block_reward, 0, [os.urandom(32)], [15]) block2_hash, block2_reward = ChainGenerator.insert_dummy_with_payments( dag, [block_hash], [payment1], 2) payment2 = TransactionFactory.create_payment(block2_reward, 0, [os.urandom(32)], [15]) block3_hash, block3_reward = ChainGenerator.insert_dummy_with_payments( dag, [block_hash], [payment2], 3) block4_hash, block4_reward = ChainGenerator.insert_dummy_with_payments( dag, [block2_hash, block3_hash], [], 4) iterator = MergingIter(dag, block4_hash) payments = [ block.block.payment_txs for block in iterator if block != None ] payments = list(reversed(payments)) spent, unspent = Resolver.resolve(payments) self.assertEqual(len(spent), 0) self.assertEqual(len(unspent), 4) self.assertIn(Entry(block3_reward, 0), unspent) self.assertIn(Entry(block4_reward, 0), unspent) self.assertIn(Entry(payment1.get_hash(), 0), unspent) self.assertIn(Entry(payment2.get_hash(), 0), unspent)
def test_merged_chain_with_transaction_conflicts(self): dag = Dag(0) block_hash, block_reward = ChainGenerator.insert_dummy_with_payments( dag, [dag.genesis_hash()], [], 1) #second block spends first block reward payment1 = TransactionFactory.create_payment(block_reward, 0, [os.urandom(32)], [15]) block2_hash, block2_reward = ChainGenerator.insert_dummy_with_payments( dag, [block_hash], [payment1], 2) #third block also spends first block reward payment2 = TransactionFactory.create_payment(block_reward, 0, [os.urandom(32)], [15]) block3_hash, block3_reward = ChainGenerator.insert_dummy_with_payments( dag, [block_hash], [payment2], 3) #making block2_hash go first, so its transactions will have a priority block4_hash, block4_reward = ChainGenerator.insert_dummy_with_payments( dag, [block2_hash, block3_hash], [], 4) iterator = MergingIter(dag, block4_hash) payments = [ block.block.payment_txs for block in iterator if block != None ] payments = list(reversed(payments)) spent, unspent = Resolver.resolve(payments) self.assertEqual(len(spent), 0) self.assertEqual(len(unspent), 4) self.assertIn(Entry(block2_reward, 0), unspent) self.assertIn(Entry(block3_reward, 0), unspent) self.assertIn(Entry(block4_reward, 0), unspent) self.assertIn(Entry(payment1.get_hash(), 0), unspent) self.assertNotIn( Entry(payment2.get_hash(), 0), unspent ) #payment 2 is consedered conflicting as it goes later in merged chain
def test_simple(self): dag = Dag(0) block_hash, block_reward = ChainGenerator.insert_dummy_with_payments( dag, [dag.genesis_hash()], [], 1) payment = TransactionFactory.create_payment(block_reward, 0, [os.urandom(32)], [12]) block2_hash, block2_reward = ChainGenerator.insert_dummy_with_payments( dag, [block_hash], [payment], 2) iterator = MergingIter(dag, block2_hash) payments = [block.block.payment_txs for block in iterator] payments = list(reversed(payments)) spent, unspent = Resolver.resolve(payments) self.assertEqual(len(spent), 0) self.assertIn(Entry(block2_reward, 0), unspent) self.assertIn(Entry(payment.get_hash(), 0), unspent)
def test_merged_chain_with_block_conflict(self): dag = Dag(0) watcher = ConflictWatcher(dag) actor1 = Private.publickey(Private.generate()) actor2 = Private.publickey(Private.generate()) actor3 = Private.publickey(Private.generate()) block1_hash, block1_reward = ChainGenerator.insert_dummy_with_payments( dag, [dag.genesis_hash()], [], 1) watcher.on_new_block_by_validator(block1_hash, 1, actor1) payment1 = TransactionFactory.create_payment(block1_reward, 0, [os.urandom(32)], [15]) block2_hash, block2_reward = ChainGenerator.insert_dummy_with_payments( dag, [block1_hash], [payment1], 2) watcher.on_new_block_by_validator(block2_hash, 1, actor2) #another block by the same validator spending the same output payment1c = TransactionFactory.create_payment(block1_reward, 0, [os.urandom(32)], [15]) block2c_hash, block2c_reward = ChainGenerator.insert_dummy_with_payments( dag, [block1_hash], [payment1c], 2) watcher.on_new_block_by_validator(block2c_hash, 1, actor2) block3_hash, block3_reward = ChainGenerator.insert_dummy_with_payments( dag, [block2_hash, block2c_hash], [], 3) watcher.on_new_block_by_validator(block3_hash, 1, actor3) # DagVisualizer.visualize(dag, True) iterator = MergingIter(dag, block3_hash, watcher) payments = [ block.block.payment_txs for block in iterator if block != None ] payments = list(reversed(payments)) spent, unspent = Resolver.resolve(payments) self.assertEqual(len(spent), 0) self.assertEqual(len(unspent), 3) self.assertIn(Entry(block2_reward, 0), unspent) self.assertIn(Entry(block3_reward, 0), unspent) self.assertIn(Entry(payment1.get_hash(), 0), unspent) self.assertNotIn(Entry(payment1c.get_hash(), 0), unspent) total_block_rewards = 0 iterator = MergingIter(dag, block3_hash, watcher) for block in iterator: if block: if not block.block.payment_txs: continue #it might be genesis, or just some silly validator who decided not to earn a reward block_reward = block.block.payment_txs[0] total_block_rewards += block_reward.amounts[0] unspent_total = 0 unspent_list = unspent #HACKY rename :) for unspent in unspent_list: unspent_tx = dag.payments_by_hash[unspent.tx] unspent_output = unspent.number unspent_amount = unspent_tx.amounts[unspent_output] unspent_total += unspent_amount self.assertEqual(total_block_rewards, unspent_total)