fixed p2pool.test.test_p2p, broken by dab6c2e8

diff --git a/p2pool/test/test_p2p.py b/p2pool/test/test_p2p.py
index 35b91dd..376bd0d 100644 (file)
--- a/p2pool/test/test_p2p.py
+++ b/p2pool/test/test_p2p.py
@@ -13,16 +13,11 @@ class MyNode(p2p.Node):
         self.df = df
     
     def handle_share_hashes(self, hashes, peer):
-        peer.send_sharereq(
-            id=self.id_to_use,
+        peer.get_shares(
             hashes=[hashes[0]],
             parents=5,
             stops=[],
-        )
-    
-    def handle_share_reply(self, id, result, shares, peer):
-        if id == self.id_to_use:
-            self.df.callback(None)
+        ).chainDeferred(self.df)
 
 class Test(unittest.TestCase):
     @defer.inlineCallbacks