def test_listen(self): # make a listen loop that will exit right away chmock = Mock(spec=ListenChannel) chmock.accept.side_effect = ChannelClosedError nodemock = Mock(spec=NodeB) nodemock.channel.return_value = chmock ep = ListeningBaseEndpoint(node=nodemock, from_name=NameTrio(sentinel.ex, sentinel.queue)) ep.listen() chmock.setup_listener.assert_called_once_with(ep._recv_name, binding=sentinel.queue)
def test_listen(self): # make a listen loop that will exit right away chmock = Mock(spec=ListenChannel) chmock.accept.side_effect = ChannelClosedError nodemock = Mock(spec=NodeB) nodemock.channel.return_value = chmock ep = ListeningBaseEndpoint(node=nodemock, from_name=NameTrio(sentinel.ex, sentinel.queue)) ep.listen() chmock.setup_listener.assert_called_once_with(ep._recv_name, binding=sentinel.queue)
def test_listen_with_base_transport_for_name(self): # make a listen loop that will exit right away chmock = Mock(spec=ListenChannel) chmock.accept.side_effect = ChannelClosedError nodemock = Mock(spec=NodeB) nodemock.channel.return_value = chmock class FakeRecvName(BaseTransport, NameTrio): pass recv_name = FakeRecvName() recv_name.setup_listener = Mock() ep = ListeningBaseEndpoint(node=nodemock, from_name=recv_name) ep.listen(binding=sentinel.binding) self.assertTrue(ep.get_ready_event().is_set()) self.assertIn('transport', nodemock.channel.call_args[1]) self.assertIn(recv_name, nodemock.channel.call_args[1].itervalues())
def test_listen_with_base_transport_for_name(self): # make a listen loop that will exit right away chmock = Mock(spec=ListenChannel) chmock.accept.side_effect = ChannelClosedError nodemock = Mock(spec=NodeB) nodemock.channel.return_value = chmock class FakeRecvName(BaseTransport, NameTrio): pass recv_name = FakeRecvName() recv_name.setup_listener = Mock() ep = ListeningBaseEndpoint(node=nodemock, from_name=recv_name) ep.listen(binding=sentinel.binding) self.assertTrue(ep.get_ready_event().is_set()) self.assertIn('transport', nodemock.channel.call_args[1]) self.assertIn(recv_name, nodemock.channel.call_args[1].itervalues())