limit max amount of tx data stored to 1.5 MB

[p2pool.git] / p2pool / test / test_p2p.py

diff --git a/p2pool/test/test_p2p.py b/p2pool/test/test_p2p.py
index a771e8c..7707ce7 100644 (file)
--- a/p2pool/test/test_p2p.py
+++ b/p2pool/test/test_p2p.py
@@ -1,9 +1,11 @@
 import random
 
-from twisted.internet import defer
+from twisted.internet import defer, endpoints, protocol, reactor
 from twisted.trial import unittest
 
 from p2pool import networks, p2p
+from p2pool.bitcoin import data as bitcoin_data
+from p2pool.util import deferral
 
 
 class Test(unittest.TestCase):
@@ -29,3 +31,56 @@ class Test(unittest.TestCase):
             yield df
         finally:
             yield n.stop()
+    
+    @defer.inlineCallbacks
+    def test_tx_limit(self):
+        class MyNode(p2p.Node):
+            def __init__(self, df):
+                p2p.Node.__init__(self, lambda: None, 29333, networks.nets['litecoin_testnet'], {}, set([('127.0.0.1', 19338)]), 0, 0, 0, 0)
+                
+                self.df = df
+                self.sent_time = 0
+            
+            @defer.inlineCallbacks
+            def got_conn(self, conn):
+                p2p.Node.got_conn(self, conn)
+                huge_tx = dict(
+                    version=0,
+                    tx_ins=[],
+                    tx_outs=[dict(
+                        value=0,
+                        script='x'*900000,
+                    )],
+                    lock_time=0,
+                )
+                new_mining_txs = dict(self.mining_txs_var.value)
+                new_mining_txs[bitcoin_data.hash256(bitcoin_data.tx_type.pack(huge_tx))] = huge_tx
+                self.mining_txs_var.set(new_mining_txs)
+                
+                yield deferral.sleep(1)
+                
+                huge_tx = dict(
+                    version=0,
+                    tx_ins=[],
+                    tx_outs=[dict(
+                        value=0,
+                        script='x'*900000,
+                    )],
+                    lock_time=1,
+                )
+                new_mining_txs = dict(self.mining_txs_var.value)
+                new_mining_txs[bitcoin_data.hash256(bitcoin_data.tx_type.pack(huge_tx))] = huge_tx
+                self.mining_txs_var.set(new_mining_txs)
+                self.sent_time = reactor.seconds()
+                
+            
+            def lost_conn(self, conn, reason):
+                self.df.callback(None)
+        
+        df = defer.Deferred()
+        n = MyNode(df)
+        n.start()
+        yield df
+        if not (n.sent_time <= reactor.seconds() <= n.sent_time + 1):
+            raise ValueError('node did not disconnect within 1 seconds of receiving too much tx data')
+        yield n.stop()