A further test for market.protocol.

Author: Tom Galloway

market/tests/test_protocol.py

@@ -1,16 +1,28 @@
 from twisted.trial import unittest
 
+from dht.node import Node
+from dht.utils import digest
+from dht.routing import RoutingTable
 from market.protocol import MarketProtocol
+from dht.tests.utils import mknode
 
 class MarketProtocolTest(unittest.TestCase):
-    def test_connect_multiplexer(self):
+    def setUp(self):
+        self.node = Node(digest("test"), "127.0.0.1", 1234)
+        self.router = RoutingTable(self, 20, self.node.id)
+
+    def test_connect_multiplexer_correctly(self):
         mp = MarketProtocol(0, 0, 0, 0)
         self.assertEqual(mp.multiplexer, None)
-	mp.connect_multiplexer("3")
-	self.assertEqual(mp.multiplexer, "3")
+        mp.connect_multiplexer("3")
+        self.assertEqual(mp.multiplexer, "3")
 
-    def test_add_listener(self):
+    def test_add_listener_correctly(self):
         mp = MarketProtocol(0, 0, 0, 0)
         self.assertEqual(len(mp.listeners), 0)
-	mp.add_listener(3)
-	self.assertEqual(len(mp.listeners),1)
+        mp.add_listener(3)
+        self.assertEqual(len(mp.listeners), 1)
+
+    def test_rpc_get_image_invalid_image_hash(self):
+        mp = MarketProtocol(self.node, self.router, 0, 0)
+        self.assertRaises(Exception, mp.rpc_get_image(mknode(), "invalid_hash"));