def test_disconnect_reconnect(self): """ Test disconnect and reconnect behavior. """ conn = Connection('1.2.3.4', 61613) conn.connect() conn.disconnect() self.assertFalse(conn.connected) conn.send("MESSAGE") self.assertTrue(self.mocksocket.connect.called) self.assertTrue(self.mocksocket.sendall.called)
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)