mojo: Add RunLoop::Current() and refactor unit tests.

mojo/python/tests/messaging_unittest.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import unittest
 
+import mojo_unittest
+
 # pylint: disable=F0401
-import mojo.embedder
 from mojo.bindings import messaging
 from mojo import system
 
 
 class _ForwardingConnectionErrorHandler(messaging.ConnectionErrorHandler):
 
   def __init__(self, callback):
     self._callback = callback
 
   def OnError(self, result):
     self._callback(result)
 
 
-class ConnectorTest(unittest.TestCase):
+class ConnectorTest(mojo_unittest.MojoTestCase):
 
   def setUp(self):
-    mojo.embedder.Init()
-    self.loop = system.RunLoop()
+    super(ConnectorTest, self).setUp()
     self.received_messages = []
     self.received_errors = []
     def _OnMessage(message):
       self.received_messages.append(message)
       return True
     def _OnError(result):
       self.received_errors.append(result)
     handles = system.MessagePipe()
     self.connector = messaging.Connector(handles.handle1)
     self.connector.SetIncomingMessageReceiver(
         messaging.ForwardingMessageReceiver(_OnMessage))
     self.connector.SetErrorHandler(
         _ForwardingConnectionErrorHandler(_OnError))
     self.connector.Start()
     self.handle = handles.handle0
 
 
   def tearDown(self):
     self.connector = None
     self.handle = None
-    self.loop = None
+    super(ConnectorTest, self).tearDown()
 
   def testConnectorRead(self):
     self.handle.WriteMessage()
     self.loop.RunUntilIdle()
     self.assertTrue(self.received_messages)
     self.assertFalse(self.received_errors)
 
   def testConnectorWrite(self):
     self.connector.Accept(messaging.Message())
     (result, _, _) = self.handle.ReadMessage()
@@ skipping 69 matching lines @@
     data = header.Serialize()
     other_header = messaging.MessageHeader.Deserialize(data)
 
     self.assertEqual(other_header.message_type, 0xdeadbeaf)
     self.assertTrue(other_header.has_request_id)
     self.assertFalse(other_header.expects_response)
     self.assertTrue(other_header.is_response)
     self.assertEqual(other_header.request_id, 0xdeadbeafdeadbeaf)
 
 
-class RouterTest(unittest.TestCase):
+class RouterTest(mojo_unittest.MojoTestCase):
 
   def setUp(self):
-    mojo.embedder.Init()
-    self.loop = system.RunLoop()
+    super(RouterTest, self).setUp()
     self.received_messages = []
     self.received_errors = []
     def _OnMessage(message):
       self.received_messages.append(message)
       return True
     def _OnError(result):
       self.received_errors.append(result)
     handles = system.MessagePipe()
     self.router = messaging.Router(handles.handle1)
     self.router.SetIncomingMessageReceiver(
         messaging.ForwardingMessageReceiver(_OnMessage))
     self.router.SetErrorHandler(
         _ForwardingConnectionErrorHandler(_OnError))
     self.router.Start()
     self.handle = handles.handle0
 
   def tearDown(self):
     self.router = None
     self.handle = None
-    self.loop = None
+    super(RouterTest, self).tearDown()
 
   def testSimpleMessage(self):
     header_data = messaging.MessageHeader(0, messaging.NO_FLAG).Serialize()
     message = messaging.Message(header_data)
     self.router.Accept(message)
     self.loop.RunUntilIdle()
     self.assertFalse(self.received_errors)
     self.assertFalse(self.received_messages)
     (res, data, _) = self.handle.ReadMessage(bytearray(len(header_data)))
     self.assertEquals(system.RESULT_OK, res)
@@ skipping 26 matching lines @@
     self.assertNotEquals(message_header.request_id, 0)
     response_header_data = messaging.MessageHeader(
         0,
         messaging.MESSAGE_IS_RESPONSE_FLAG,
         message_header.request_id).Serialize()
     self.handle.WriteMessage(response_header_data)
     self.loop.RunUntilIdle()
     self.assertFalse(self.received_errors)
     self.assertEquals(len(back_messages), 1)
     self.assertEquals(back_messages[0].data, response_header_data)