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