mock_protocol.py 487 B

1234567891011121314151617
  1. from src.mining import Miner
  2. class MockProtocol:
  3. def __init__(self):
  4. self.block_receive_handlers = []
  5. self.trans_receive_handlers = []
  6. def fake_block_received(self, block):
  7. for handler in self.block_receive_handlers:
  8. handler(block)
  9. def fake_trans_received(self, trans):
  10. for handler in self.trans_receive_handlers:
  11. handler(trans)
  12. def broadcast_primary_block(self, block):
  13. self.fake_block_received(block)