from ganeti import daemon
from ganeti import errors
+from ganeti import constants
from ganeti import utils
import testutils
class _MyAsyncUDPSocket(daemon.AsyncUDPSocket):
- def __init__(self):
- daemon.AsyncUDPSocket.__init__(self)
+ def __init__(self, family):
+ daemon.AsyncUDPSocket.__init__(self, family)
self.received = []
self.error_count = 0
raise
-class TestAsyncUDPSocket(testutils.GanetiTestCase):
- """Test daemon.AsyncUDPSocket"""
+class _BaseAsyncUDPSocketTest:
+ """Base class for AsyncUDPSocket tests"""
+
+ family = None
+ address = None
def setUp(self):
- testutils.GanetiTestCase.setUp(self)
self.mainloop = daemon.Mainloop()
- self.server = _MyAsyncUDPSocket()
- self.client = _MyAsyncUDPSocket()
- self.server.bind(("127.0.0.1", 0))
+ self.server = _MyAsyncUDPSocket(self.family)
+ self.client = _MyAsyncUDPSocket(self.family)
+ self.server.bind((self.address, 0))
self.port = self.server.getsockname()[1]
# Save utils.IgnoreSignals so we can do evil things to it...
self.saved_utils_ignoresignals = utils.IgnoreSignals
testutils.GanetiTestCase.tearDown(self)
def testNoDoubleBind(self):
- self.assertRaises(socket.error, self.client.bind, ("127.0.0.1", self.port))
-
- def _ThreadedClient(self, payload):
- self.client.enqueue_send("127.0.0.1", self.port, payload)
- print "sending %s" % payload
- while self.client.writable():
- self.client.handle_write()
+ self.assertRaises(socket.error, self.client.bind, (self.address, self.port))
def testAsyncClientServer(self):
- self.client.enqueue_send("127.0.0.1", self.port, "p1")
- self.client.enqueue_send("127.0.0.1", self.port, "p2")
- self.client.enqueue_send("127.0.0.1", self.port, "terminate")
+ self.client.enqueue_send(self.address, self.port, "p1")
+ self.client.enqueue_send(self.address, self.port, "p2")
+ self.client.enqueue_send(self.address, self.port, "terminate")
self.mainloop.Run()
self.assertEquals(self.server.received, ["p1", "p2", "terminate"])
def testSyncClientServer(self):
- self.client.enqueue_send("127.0.0.1", self.port, "p1")
- self.client.enqueue_send("127.0.0.1", self.port, "p2")
+ self.client.handle_write()
+ self.client.enqueue_send(self.address, self.port, "p1")
+ self.client.enqueue_send(self.address, self.port, "p2")
while self.client.writable():
self.client.handle_write()
self.server.process_next_packet()
self.assertEquals(self.server.received, ["p1"])
self.server.process_next_packet()
self.assertEquals(self.server.received, ["p1", "p2"])
- self.client.enqueue_send("127.0.0.1", self.port, "p3")
+ self.client.enqueue_send(self.address, self.port, "p3")
while self.client.writable():
self.client.handle_write()
self.server.process_next_packet()
self.assertEquals(self.server.received, ["p1", "p2", "p3"])
def testErrorHandling(self):
- self.client.enqueue_send("127.0.0.1", self.port, "p1")
- self.client.enqueue_send("127.0.0.1", self.port, "p2")
- self.client.enqueue_send("127.0.0.1", self.port, "error")
- self.client.enqueue_send("127.0.0.1", self.port, "p3")
- self.client.enqueue_send("127.0.0.1", self.port, "error")
- self.client.enqueue_send("127.0.0.1", self.port, "terminate")
+ self.client.enqueue_send(self.address, self.port, "p1")
+ self.client.enqueue_send(self.address, self.port, "p2")
+ self.client.enqueue_send(self.address, self.port, "error")
+ self.client.enqueue_send(self.address, self.port, "p3")
+ self.client.enqueue_send(self.address, self.port, "error")
+ self.client.enqueue_send(self.address, self.port, "terminate")
self.assertRaises(errors.GenericError, self.mainloop.Run)
self.assertEquals(self.server.received,
["p1", "p2", "error"])
def testSignaledWhileReceiving(self):
utils.IgnoreSignals = lambda fn, *args, **kwargs: None
- self.client.enqueue_send("127.0.0.1", self.port, "p1")
- self.client.enqueue_send("127.0.0.1", self.port, "p2")
+ self.client.enqueue_send(self.address, self.port, "p1")
+ self.client.enqueue_send(self.address, self.port, "p2")
self.server.handle_read()
self.assertEquals(self.server.received, [])
- self.client.enqueue_send("127.0.0.1", self.port, "terminate")
+ self.client.enqueue_send(self.address, self.port, "terminate")
utils.IgnoreSignals = self.saved_utils_ignoresignals
self.mainloop.Run()
self.assertEquals(self.server.received, ["p1", "p2", "terminate"])
+ def testOversizedDatagram(self):
+ oversized_data = (constants.MAX_UDP_DATA_SIZE + 1) * "a"
+ self.assertRaises(errors.UdpDataSizeError, self.client.enqueue_send,
+ self.address, self.port, oversized_data)
+
+
+class TestAsyncIP4UDPSocket(testutils.GanetiTestCase, _BaseAsyncUDPSocketTest):
+ """Test IP4 daemon.AsyncUDPSocket"""
+
+ family = socket.AF_INET
+ address = "127.0.0.1"
+
+ def setUp(self):
+ testutils.GanetiTestCase.setUp(self)
+ _BaseAsyncUDPSocketTest.setUp(self)
+
+ def tearDown(self):
+ testutils.GanetiTestCase.tearDown(self)
+ _BaseAsyncUDPSocketTest.tearDown(self)
+
+
+class TestAsyncIP6UDPSocket(testutils.GanetiTestCase, _BaseAsyncUDPSocketTest):
+ """Test IP6 daemon.AsyncUDPSocket"""
+
+ family = socket.AF_INET6
+ address = "::1"
+
+ def setUp(self):
+ testutils.GanetiTestCase.setUp(self)
+ _BaseAsyncUDPSocketTest.setUp(self)
+
+ def tearDown(self):
+ testutils.GanetiTestCase.tearDown(self)
+ _BaseAsyncUDPSocketTest.tearDown(self)
+
class _MyAsyncStreamServer(daemon.AsyncStreamServer):
class _MyMessageStreamHandler(daemon.AsyncTerminatedMessageStream):
def __init__(self, connected_socket, client_address, terminator, family,
- message_fn, client_id):
+ message_fn, client_id, unhandled_limit):
daemon.AsyncTerminatedMessageStream.__init__(self, connected_socket,
client_address,
- terminator, family)
+ terminator, family,
+ unhandled_limit)
self.message_fn = message_fn
self.client_id = client_id
self.error_count = 0
self.server = _MyAsyncStreamServer(self.family, self.address,
self.handle_connection)
self.client_handler = _MyMessageStreamHandler
+ self.unhandled_limit = None
self.terminator = "\3"
self.address = self.server.getsockname()
self.clients = []
client_handler = self.client_handler(connected_socket, client_address,
self.terminator, self.family,
self.handle_message,
- client_id)
+ client_id, self.unhandled_limit)
self.connections.append(client_handler)
self.countTerminate("connect_terminate_count")
self.mainloop.Run()
self.assertEquals(len(self.connections), 1)
+ def testSendMessage(self):
+ self.connect_terminate_count = None
+ self.message_terminate_count = 3
+ client1 = self.getClient()
+ client2 = self.getClient()
+ client1.send("one\3composed\3message\3")
+ self.mainloop.Run()
+ self.assertEquals(self.messages[0], ["one", "composed", "message"])
+ self.assertFalse(self.connections[0].writable())
+ self.assertFalse(self.connections[1].writable())
+ self.connections[0].send_message("r0")
+ self.assert_(self.connections[0].writable())
+ self.assertFalse(self.connections[1].writable())
+ self.connections[0].send_message("r1")
+ self.connections[0].send_message("r2")
+ # We currently have no way to terminate the mainloop on write events, but
+ # let's assume handle_write will be called if writable() is True.
+ while self.connections[0].writable():
+ self.connections[0].handle_write()
+ client1.setblocking(0)
+ client2.setblocking(0)
+ self.assertEquals(client1.recv(4096), "r0\3r1\3r2\3")
+ self.assertRaises(socket.error, client2.recv, 4096)
+
+ def testLimitedUnhandledMessages(self):
+ self.connect_terminate_count = None
+ self.message_terminate_count = 3
+ self.unhandled_limit = 2
+ client1 = self.getClient()
+ client2 = self.getClient()
+ client1.send("one\3composed\3long\3message\3")
+ client2.send("c2one\3")
+ self.mainloop.Run()
+ self.assertEquals(self.messages[0], ["one", "composed"])
+ self.assertEquals(self.messages[1], ["c2one"])
+ self.assertFalse(self.connections[0].readable())
+ self.assert_(self.connections[1].readable())
+ self.connections[0].send_message("r0")
+ self.message_terminate_count = None
+ client1.send("another\3")
+ # when we write replies messages queued also get handled, but not the ones
+ # in the socket.
+ while self.connections[0].writable():
+ self.connections[0].handle_write()
+ self.assertFalse(self.connections[0].readable())
+ self.assertEquals(self.messages[0], ["one", "composed", "long"])
+ self.connections[0].send_message("r1")
+ self.connections[0].send_message("r2")
+ while self.connections[0].writable():
+ self.connections[0].handle_write()
+ self.assertEquals(self.messages[0], ["one", "composed", "long", "message"])
+ self.assert_(self.connections[0].readable())
+
+ def testLimitedUnhandledMessagesOne(self):
+ self.connect_terminate_count = None
+ self.message_terminate_count = 2
+ self.unhandled_limit = 1
+ client1 = self.getClient()
+ client2 = self.getClient()
+ client1.send("one\3composed\3message\3")
+ client2.send("c2one\3")
+ self.mainloop.Run()
+ self.assertEquals(self.messages[0], ["one"])
+ self.assertEquals(self.messages[1], ["c2one"])
+ self.assertFalse(self.connections[0].readable())
+ self.assertFalse(self.connections[1].readable())
+ self.connections[0].send_message("r0")
+ self.message_terminate_count = None
+ while self.connections[0].writable():
+ self.connections[0].handle_write()
+ self.assertFalse(self.connections[0].readable())
+ self.assertEquals(self.messages[0], ["one", "composed"])
+ self.connections[0].send_message("r2")
+ self.connections[0].send_message("r3")
+ while self.connections[0].writable():
+ self.connections[0].handle_write()
+ self.assertEquals(self.messages[0], ["one", "composed", "message"])
+ self.assert_(self.connections[0].readable())
+
class TestAsyncStreamServerUnixPath(TestAsyncStreamServerTCP):
"""Test daemon.AsyncStreamServer with a Unix path connection"""
TestAsyncStreamServerTCP.tearDown(self)
+class TestAsyncAwaker(testutils.GanetiTestCase):
+ """Test daemon.AsyncAwaker"""
+
+ family = socket.AF_INET
+
+ def setUp(self):
+ testutils.GanetiTestCase.setUp(self)
+ self.mainloop = daemon.Mainloop()
+ self.awaker = daemon.AsyncAwaker(signal_fn=self.handle_signal)
+ self.signal_count = 0
+ self.signal_terminate_count = 1
+
+ def tearDown(self):
+ self.awaker.close()
+
+ def handle_signal(self):
+ self.signal_count += 1
+ self.signal_terminate_count -= 1
+ if self.signal_terminate_count <= 0:
+ os.kill(os.getpid(), signal.SIGTERM)
+
+ def testBasicSignaling(self):
+ self.awaker.signal()
+ self.mainloop.Run()
+ self.assertEquals(self.signal_count, 1)
+
+ def testDoubleSignaling(self):
+ self.awaker.signal()
+ self.awaker.signal()
+ self.mainloop.Run()
+ # The second signal is never delivered
+ self.assertEquals(self.signal_count, 1)
+
+ def testReallyDoubleSignaling(self):
+ self.assert_(self.awaker.readable())
+ self.awaker.signal()
+ # Let's suppose two threads overlap, and both find need_signal True
+ self.awaker.need_signal = True
+ self.awaker.signal()
+ self.mainloop.Run()
+ # We still get only one signaling
+ self.assertEquals(self.signal_count, 1)
+
+ def testNoSignalFnArgument(self):
+ myawaker = daemon.AsyncAwaker()
+ self.assertRaises(socket.error, myawaker.handle_read)
+ myawaker.signal()
+ myawaker.handle_read()
+ self.assertRaises(socket.error, myawaker.handle_read)
+ myawaker.signal()
+ myawaker.signal()
+ myawaker.handle_read()
+ self.assertRaises(socket.error, myawaker.handle_read)
+ myawaker.close()
+
+ def testWrongSignalFnArgument(self):
+ self.assertRaises(AssertionError, daemon.AsyncAwaker, 1)
+ self.assertRaises(AssertionError, daemon.AsyncAwaker, "string")
+ self.assertRaises(AssertionError, daemon.AsyncAwaker, signal_fn=1)
+ self.assertRaises(AssertionError, daemon.AsyncAwaker, signal_fn="string")
+
+
if __name__ == "__main__":
testutils.GanetiTestProgram()