Sleep after connect for router/dealer tests to ensure dealer is conne…

…cted prior to message.

cis_interface/communication/tests/test_CommBase.py

@@ -96,11 +96,14 @@ def test_msg(self):
     def setup(self, *args, **kwargs):
         r"""Initialize comm object pair."""
         assert(self.is_installed)
+        sleep_after_connect = kwargs.pop('sleep_after_connect', False)
         send_inst_kwargs = self.send_inst_kwargs
         kwargs.setdefault('nprev_comm', self.comm_count)
         kwargs.setdefault('nprev_fd', self.fd_count)
         self.send_instance = new_comm(self.name, **send_inst_kwargs)
         super(TestCommBase, self).setup(*args, **kwargs)
+        if sleep_after_connect:
+            self.send_instance.sleep()
         # CommBase is dummy class that never opens
         if self.comm in ['CommBase', 'AsyncComm']:
             assert(not self.send_instance.is_open)

cis_interface/communication/tests/test_ZMQComm.py

@@ -246,6 +246,12 @@ def __init__(self, *args, **kwargs):
         super(TestZMQCommROUTER, self).__init__(*args, **kwargs)
         self.socket_type = 'ROUTER'
 
+    def setup(self, *args, **kwargs):
+        r"""Initialize comm object pair with sleep after setup to ensure
+        dealer has connected."""
+        kwargs['sleep_after_connect'] = True
+        super(TestZMQCommROUTER, self).setup(*args, **kwargs)
+
     def test_router_recv(self):
         r"""Test router receipt of message from the dealer with an identity."""
         self.do_send_recv(reverse_comms=True, send_kwargs=dict(