本文整理汇总了Python中agentzero.core.SocketManager.ensure_and_connect方法的典型用法代码示例。如果您正苦于以下问题：Python SocketManager.ensure_and_connect方法的具体用法？Python SocketManager.ensure_and_connect怎么用？Python SocketManager.ensure_and_connect使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类agentzero.core.SocketManager的用法示例。
在下文中一共展示了SocketManager.ensure_and_connect方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_socket_manager_can_poll_asynchronously
# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import ensure_and_connect [as 别名]
def test_socket_manager_can_poll_asynchronously():
("SocketManager should leverage a non-blocking socket "
"collection can be used seamlessly in a blocking fashion [TCP SOCKET]")
# Given a socket manager for a server
server = SocketManager(zmq, context)
# And a reply socket listening on a tcp port
server.ensure_and_bind('reply-server', zmq.REP, 'tcp://0.0.0.0:3458', zmq.POLLIN | zmq.POLLOUT)
# And a socket manager for a client
client = SocketManager(zmq, context)
# And a request client connected to the server
client.ensure_and_connect('request-client', zmq.REQ, 'tcp://0.0.0.0:3458', zmq.POLLIN | zmq.POLLOUT)
# And send a request from the client
requester = client.wait_until_ready('request-client', zmq.POLLOUT, timeout=2)
requester.send_json({'client': 42})
# Then I should receive a request from the client
replier = server.wait_until_ready('reply-server', zmq.POLLIN, timeout=2)
replier.should_not.be.none
request = replier.recv_json()
# And the request should be the one that the client just sent
request.should.equal({'client': 42})示例2: test_socket_ensure_and_connect
# 需要导入模块: from agentzero.core import SocketManager [as 别名]
# 或者: from agentzero.core.SocketManager import ensure_and_connect [as 别名]
def test_socket_ensure_and_connect(get_or_create, connect, engage):
(
"SocketManager().ensure_and_connect() should ensure the socket existence then "
"connect to the given address."
)
# Given a zmq mock
zmq = Mock()
# And a context
context = Mock()
# And a socket manager
manager = SocketManager(zmq, context)
# When I call ensure_and_connect
result = manager.ensure_and_connect(
"foobar", "SOME_TYPE", "inproc://whatevs", "some-mechanism"
)
# Then it should have returned a socket from connect
result.should.equal(connect.return_value)
# And get_or_create should have been called correctly
get_or_create.assert_called_once_with(
"foobar", "SOME_TYPE", "some-mechanism"
)
# And connect should have been called with the given address
connect.assert_called_once_with(
"foobar", "inproc://whatevs", "some-mechanism"
)
# And engage should have been called with the default value
engage.assert_called_once_with()