cleaned up test.test_p2p.Test.test_sharereq
authorForrest Voight <forrest@forre.st>
Sat, 13 Oct 2012 16:29:40 +0000 (12:29 -0400)
committerForrest Voight <forrest@forre.st>
Sun, 14 Oct 2012 17:22:58 +0000 (13:22 -0400)
p2pool/test/test_p2p.py

index 376bd0d..a771e8c 100644 (file)
@@ -5,23 +5,23 @@ from twisted.trial import unittest
 
 from p2pool import networks, p2p
 
-class MyNode(p2p.Node):
-    def __init__(self, df):
-        p2p.Node.__init__(self, lambda: None, 29333, networks.nets['bitcoin'], {}, set([('127.0.0.1', 9333)]), 0, 0, 0, 0)
-        
-        self.id_to_use = random.randrange(2**256)
-        self.df = df
-    
-    def handle_share_hashes(self, hashes, peer):
-        peer.get_shares(
-            hashes=[hashes[0]],
-            parents=5,
-            stops=[],
-        ).chainDeferred(self.df)
 
 class Test(unittest.TestCase):
     @defer.inlineCallbacks
     def test_sharereq(self):
+        class MyNode(p2p.Node):
+            def __init__(self, df):
+                p2p.Node.__init__(self, lambda: None, 29333, networks.nets['bitcoin'], {}, set([('127.0.0.1', 9333)]), 0, 0, 0, 0)
+                
+                self.df = df
+            
+            def handle_share_hashes(self, hashes, peer):
+                peer.get_shares(
+                    hashes=[hashes[0]],
+                    parents=5,
+                    stops=[],
+                ).chainDeferred(self.df)
+        
         df = defer.Deferred()
         n = MyNode(df)
         n.start()