def test_socket_manager_get_or_create_preexisting(create, get_by_name, register_socket): ("SocketManager().get_or_create() should just return " "and register a pre-existing socket with the same name") # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager with no sockets manager = SocketManager(zmq, context) manager.sockets = {} # When I call get_or_create result = manager.get_or_create("foobar", "SOME_TYPE", "fake-polling-mechanism") # Then it should have returned a socket result.should.equal(get_by_name.return_value) # And the socket should have been created create.assert_called_once_with("foobar", "SOME_TYPE") # And should have been retrieved by name get_by_name.assert_called_once_with("foobar") # And should have been registered register_socket.assert_called_once_with( result, "fake-polling-mechanism" )
def test_socket_manager_get_or_create_preexisting(create, get_by_name, register_socket): ("SocketManager().get_or_create() should just return " "and register a pre-existing socket with the same name") # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager with no sockets manager = SocketManager(zmq, context) manager.sockets = {} # When I call get_or_create result = manager.get_or_create("foobar", "SOME_TYPE", "fake-polling-mechanism") # Then it should have returned a socket result.should.equal(get_by_name.return_value) # And the socket should have been created create.assert_called_once_with("foobar", "SOME_TYPE") # And should have been retrieved by name get_by_name.assert_called_once_with("foobar") # And should have been registered register_socket.assert_called_once_with(result, "fake-polling-mechanism")
def test_socket_manager_get_or_create(create, get_by_name, register_socket): ("SocketManager().get_or_create() should create a socket and register it") # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager with no sockets manager = SocketManager(zmq, context) manager.sockets = { "foobar": "SOME_TYPE" } # When I call get_or_create result = manager.get_or_create("foobar", "SOME_TYPE", "fake-polling-mechanism") # Then it should have returned a socket result.should.equal(get_by_name.return_value) # And .create should not have been called create.called.should.be.false # And should have been retrieved by name get_by_name.assert_called_once_with("foobar") # And should have been registered register_socket.assert_called_once_with( result, "fake-polling-mechanism", )
def test_socket_manager_get_or_create(create, get_by_name, register_socket): ("SocketManager().get_or_create() should create a socket and register it") # Given a zmq mock zmq = Mock() # And a context context = Mock() # And a socket manager with no sockets manager = SocketManager(zmq, context) manager.sockets = {"foobar": "SOME_TYPE"} # When I call get_or_create result = manager.get_or_create("foobar", "SOME_TYPE", "fake-polling-mechanism") # Then it should have returned a socket result.should.equal(get_by_name.return_value) # And .create should not have been called create.called.should.be.false # And should have been retrieved by name get_by_name.assert_called_once_with("foobar") # And should have been registered register_socket.assert_called_once_with( result, "fake-polling-mechanism", )