3 from twisted.internet import defer, endpoints, protocol, reactor
4 from twisted.trial import unittest
6 from p2pool import networks, p2p
7 from p2pool.bitcoin import data as bitcoin_data
8 from p2pool.util import deferral
11 class Test(unittest.TestCase):
12 @defer.inlineCallbacks
13 def test_sharereq(self):
14 class MyNode(p2p.Node):
15 def __init__(self, df):
16 p2p.Node.__init__(self, lambda: None, 29333, networks.nets['bitcoin'], {}, set([('127.0.0.1', 9333)]), 0, 0, 0, 0)
20 def handle_share_hashes(self, hashes, peer):
25 ).chainDeferred(self.df)
35 @defer.inlineCallbacks
36 def test_tx_limit(self):
37 class MyNode(p2p.Node):
38 def __init__(self, df):
39 p2p.Node.__init__(self, lambda: None, 29333, networks.nets['bitcoin'], {}, set([('127.0.0.1', 9333)]), 0, 0, 0, 0)
44 @defer.inlineCallbacks
45 def got_conn(self, conn):
46 p2p.Node.got_conn(self, conn)
48 yield deferral.sleep(.5)
50 new_mining_txs = dict(self.mining_txs_var.value)
61 new_mining_txs[bitcoin_data.hash256(bitcoin_data.tx_type.pack(huge_tx))] = huge_tx
62 self.mining_txs_var.set(new_mining_txs)
64 self.sent_time = reactor.seconds()
66 def lost_conn(self, conn, reason):
67 self.df.callback(None)
69 p2p.Protocol.max_remembered_txs_size *= 10
75 if not (n.sent_time <= reactor.seconds() <= n.sent_time + 1):
76 raise ValueError('node did not disconnect within 1 seconds of receiving too much tx data')
79 p2p.Protocol.max_remembered_txs_size //= 10