Python SocketManager.sockets方法代码示例

# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import sockets [as 别名]
def test_socket_manager_not_ready(get_by_name, engage):
    ("SocketManager().ready() should return None when the socket is not available")

    # Given a zmq mock
    zmq = Mock()

    # And a context
    context = Mock()

    # And a socket manager with a fake socket and an *EMPTY* cache
    manager = SocketManager(zmq, context)
    manager.sockets = {
        "foobar": "SOME_TYPE"
    }
    manager.cache = {}

    # And that get_by_name will return "socket_sentinel"
    get_by_name.return_value = 'socket_sentinel'

    # When I call ready
    socket = manager.ready('foobar', 'some-polling-mechanism', timeout=1000)

    # Then the result should be None
    socket.should.be.none

    # And engage should have been called
    engage.assert_called_once_with(1000)

# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import sockets [as 别名]
def test_socket_manager_ready(get_by_name, engage):
    ("SocketManager().ready() should return the socket "
     "if it's ready and engaged with the given polling mechanism")

    # Given a zmq mock
    zmq = Mock()

    # And a context
    context = Mock()

    # And a socket manager with a fake socket and a fake cache
    manager = SocketManager(zmq, context)
    manager.sockets = {
        "foobar": "SOME_TYPE"
    }
    manager.cache = {
        'socket_sentinel': "some-polling-mechanism"
    }
    # And that get_by_name will return "socket_sentinel"
    get_by_name.return_value = 'socket_sentinel'

    # When I call ready
    socket = manager.ready('foobar', 'some-polling-mechanism', timeout=1000)

    # Then the result should be the "socket_sentinel"
    socket.should.equal("socket_sentinel")

    # And engage should have been called
    engage.assert_called_once_with(1000)

# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import sockets [as 别名]
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"
    )

# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import sockets [as 别名]
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",
    )

# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import sockets [as 别名]
def test_socket_manager_repr():
    ("repr(SocketManager()) should be informative")
    # Given a zmq mock
    zmq = Mock()

    # And a context
    context = Mock()

    # And a socket manager with some fake sockets
    manager = SocketManager(zmq, context)
    manager.sockets = OrderedDict([("foobar", "WHATEVS"), ("awesome", "WHATEVS")])

    # When I get the string representation of the instance
    string = repr(manager)

    # Then it should
    string.should.equal("SocketManager(sockets=['foobar', 'awesome'])")