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)
