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",
)
