Example #1
0
 def test_auto_connect(self):
     """ Test the fact that send and read automatically attempt to connect. """
     conn = Connection('1.2.3.4', 61613)
     conn.send("MESSAGE")
     self.assertTrue(self.mocksocket.connect.called)
     conn.disconnect()
     
     f = frame.ConnectedFrame('my-session-id')
     self.mocksocket.reset_mock()
     self.mocksocket.recv.side_effect = lambda len: str(f)
     
     result = conn.read()
     self.assertEquals(str(f), str(result))
     self.assertTrue(self.mocksocket.connect.called)