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)
