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())