remove MockSocket in test_session

subclassing zmq.Socket doesn't behave well on Python 3,
and there's no point anyway.

IPython/zmq/tests/test_session.py

@@ -27,22 +27,6 b' class SessionTestCase(BaseZMQTestCase):'
         self.session = ss.Session()
 
 
-class MockSocket(zmq.Socket):
-    data = None
-
-    def __init__(self, *args, **kwargs):
-        super(MockSocket,self).__init__(*args,**kwargs)
-        self.data = []
-
-    def send_multipart(self, msgparts, *args, **kwargs):
-        self.data.extend(msgparts)
-
-    def send(self, part, *args, **kwargs):
-        self.data.append(part)
-
-    def recv_multipart(self, *args, **kwargs):
-        return self.data
-
 class TestSession(SessionTestCase):
 
     def test_msg(self):
@@ -76,11 +60,16 b' class TestSession(SessionTestCase):'
         self.assertEqual(type(new_msg['content']['b']),type(new_msg['content']['b']))
 
     def test_send(self):
-        socket = MockSocket(zmq.Context.instance(),zmq.PAIR)
+        ctx = zmq.Context.instance()
+        A = ctx.socket(zmq.PAIR)
+        B = ctx.socket(zmq.PAIR)
+        A.bind("inproc://test")
+        B.connect("inproc://test")
 
         msg = self.session.msg('execute', content=dict(a=10))
-        self.session.send(socket, msg, ident=b'foo', buffers=[b'bar'])
-        ident, msg_list = self.session.feed_identities(socket.data)
+        self.session.send(A, msg, ident=b'foo', buffers=[b'bar'])
+        
+        ident, msg_list = self.session.feed_identities(B.recv_multipart())
         new_msg = self.session.unserialize(msg_list)
         self.assertEqual(ident[0], b'foo')
         self.assertEqual(new_msg['msg_id'],msg['msg_id'])
@@ -91,16 +80,14 b' class TestSession(SessionTestCase):'
         self.assertEqual(new_msg['metadata'],msg['metadata'])
         self.assertEqual(new_msg['buffers'],[b'bar'])
 
-        socket.data = []
-
         content = msg['content']
         header = msg['header']
         parent = msg['parent_header']
         metadata = msg['metadata']
         msg_type = header['msg_type']
-        self.session.send(socket, None, content=content, parent=parent,
+        self.session.send(A, None, content=content, parent=parent,
             header=header, metadata=metadata, ident=b'foo', buffers=[b'bar'])
-        ident, msg_list = self.session.feed_identities(socket.data)
+        ident, msg_list = self.session.feed_identities(B.recv_multipart())
         new_msg = self.session.unserialize(msg_list)
         self.assertEqual(ident[0], b'foo')
         self.assertEqual(new_msg['msg_id'],msg['msg_id'])
@@ -111,10 +98,8 b' class TestSession(SessionTestCase):'
         self.assertEqual(new_msg['parent_header'],msg['parent_header'])
         self.assertEqual(new_msg['buffers'],[b'bar'])
 
-        socket.data = []
-
-        self.session.send(socket, msg, ident=b'foo', buffers=[b'bar'])
-        ident, new_msg = self.session.recv(socket)
+        self.session.send(A, msg, ident=b'foo', buffers=[b'bar'])
+        ident, new_msg = self.session.recv(B)
         self.assertEqual(ident[0], b'foo')
         self.assertEqual(new_msg['msg_id'],msg['msg_id'])
         self.assertEqual(new_msg['msg_type'],msg['msg_type'])
@@ -124,7 +109,9 b' class TestSession(SessionTestCase):'
         self.assertEqual(new_msg['parent_header'],msg['parent_header'])
         self.assertEqual(new_msg['buffers'],[b'bar'])
 
-        socket.close()
+        A.close()
+        B.close()
+        ctx.term()
 
     def test_args(self):
         """initialization arguments for Session"""