p2pool/test/test_p2p.py

   1 import random
   2
   3 from twisted.internet import defer
   4 from twisted.trial import unittest
   5
   6 from p2pool import networks, p2p
   7
   8 class MyNode(p2p.Node):
   9     def __init__(self, df):
  10         p2p.Node.__init__(self, lambda: None, 29333, networks.nets['bitcoin'], {}, set([('127.0.0.1', 9333)]), 0, 0, 0, 0)
  11
  12         self.id_to_use = random.randrange(2**256)
  13         self.df = df
  14
  15     def handle_share_hashes(self, hashes, peer):
  16         peer.get_shares(
  17             hashes=[hashes[0]],
  18             parents=5,
  19             stops=[],
  20         ).chainDeferred(self.df)
  21
  22 class Test(unittest.TestCase):
  23     @defer.inlineCallbacks
  24     def test_sharereq(self):
  25         df = defer.Deferred()
  26         n = MyNode(df)
  27         n.start()
  28         try:
  29             yield df
  30         finally:
  31             yield n.stop()