def test_socket_close(get_by_name): ("SocketManager().close() should unregister, pop it " "from the list of connected sockets and then close") socket = get_by_name.return_value # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager manager = SocketManager(zmq, context) manager.poller = Mock(name='poller') manager.addresses = { 'foobar': 'wooot' } # When I call close manager.close("foobar") # Then it should have unregistered the socket with the poller manager.poller.unregister.assert_called_once_with(socket) # And the socket should have been removed from the internal addresses manager.addresses.should.be.empty # And socket.close should have been called socket.close.assert_called_once_with()
def test_socket_close(get_by_name): ("SocketManager().close() should unregister, pop it " "from the list of connected sockets and then close") socket = get_by_name.return_value # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager manager = SocketManager(zmq, context) manager.poller = Mock(name='poller') manager.addresses = {'foobar': 'wooot'} # When I call close manager.close("foobar") # Then it should have unregistered the socket with the poller manager.poller.unregister.assert_called_once_with(socket) # And the socket should have been removed from the internal addresses manager.addresses.should.be.empty # And socket.close should have been called socket.close.assert_called_once_with()
def test_socket_close_no_socket(get_by_name): ("SocketManager().close() should do nothing if a socket " "with that given name does not exist") # Background: no sockets will be returned by get_by_name get_by_name.return_value = None # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager manager = SocketManager(zmq, context) manager.poller = Mock(name='poller') # When I call close manager.close("foobar") # Then it should not have called unregister manager.poller.unregister.called.should.be.false