def test_disconnect_notconnected(self): """ Attempting to disconnect when not connected should raise exception. """ conn = Connection('1.2.3.4', 61613) self.assertRaises(NotConnectedError, conn.disconnect) conn.connect() conn.disconnect() self.assertRaises(NotConnectedError, conn.disconnect)
def test_connected(self): """ Test that 'connected' property reflects connection status. """ conn = Connection('1.2.3.4', 61613) conn.connect() self.assertTrue(conn.connected) conn.disconnect() self.assertFalse(conn.connected) conn.connect() self.assertTrue(conn.connected)
def test_connect(self): """ Test basic connection functionality. """ conn = Connection('1.2.3.4', 61613) conn.connect() print self.mocksocket.method_calls self.assertTrue(self.mocksocket.connect.called) self.assertEquals((('1.2.3.4', 61613),), self.mocksocket.connect.call_args[0]) self.mocksocket.reset_mock() conn.connect() self.assertFalse(self.mocksocket.connect.called)
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)