test_proto.py 2.0 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. from time import sleep
  2. from src.protocol import Protocol
  3. from src.mining import Miner
  4. from src.block import GENESIS_BLOCK
  5. from src.crypto import Signing
  6. from src.transaction import Transaction, TransactionInput, TransactionTarget
  7. def test_proto():
  8. reward_key = Signing.generate_private_key()
  9. proto1 = Protocol([], GENESIS_BLOCK, 1337)
  10. proto2 = Protocol([("127.0.0.1", 1337)], GENESIS_BLOCK, 1338)
  11. miner1 = Miner(proto1, reward_key)
  12. miner2 = Miner(proto2, reward_key)
  13. miner2.start_mining()
  14. miner1.start_mining()
  15. sleep(5)
  16. target_key = Signing.generate_private_key()
  17. reward_trans = miner2.chainbuilder.primary_block_chain.blocks[20].transactions[0]
  18. trans_in = TransactionInput(reward_trans.get_hash(), 0)
  19. trans_targ = TransactionTarget(target_key, reward_trans.targets[0].amount)
  20. trans = Transaction([trans_in], [trans_targ])
  21. trans.sign([reward_key])
  22. assert trans.verify(miner1.chainbuilder.primary_block_chain, set()), "transaction should be valid"
  23. proto2.received('transaction', trans.to_json_compatible(), None)
  24. sleep(5)
  25. chain_len1 = len(miner1.chainbuilder.primary_block_chain.blocks)
  26. chain_len2 = len(miner2.chainbuilder.primary_block_chain.blocks)
  27. print("Length of chain of miner 1: {}".format(chain_len1))
  28. print("Length of chain of miner 2: {}".format(chain_len2))
  29. miner1.stop_mining()
  30. miner2.stop_mining()
  31. assert max(chain_len1, chain_len2) * 90 // 100 < min(chain_len1, chain_len2), "chain lengths are VERY different"
  32. chain1 = miner1.chainbuilder.primary_block_chain
  33. hashes1 = [b.hash for b in chain1.blocks[:chain_len1 * 90 // 100]]
  34. hashes2 = [b.hash for b in miner2.chainbuilder.primary_block_chain.blocks[:chain_len1 * 90 // 100]]
  35. assert hashes1 == hashes2, "first 90% of chains should be identical"
  36. assert not trans.verify(miner1.chainbuilder.primary_block_chain, set()), "inserted transaction should be spent and therefore invalid"
  37. assert chain1.is_coin_still_valid(TransactionInput(trans.get_hash(), 0)), "someone spent our coins?"