def mock_protocol():
mproto = Mock(RPCProtocol)
foo = Mock(RPCResponse)
foo.result = None
mproto.parse_reply = Mock(return_value=foo)
return mproto
def mock_protocol():
mproto = Mock(RPCProtocol)
foo = Mock(RPCResponse)
foo.result = None
mproto.parse_reply = Mock(return_value=foo)
return mproto
def client(mock_protocol, mock_transport, request):
client = RPCClient(mock_protocol, mock_transport, NoOpPublisher())
client.wait_for_task = Mock(return_value=None)
response = Mock(RPCResponse)
response.unicode = '0'
response.result = 'result1'
client.get_status = Mock(return_value={
response.unicode: ('done', response, 'some time', 'timeout')
})
def fin():
client.receiver.alive = False
request.addfinalizer(fin)
return client
