Revision 065be3f0 lib/rpc.py

b/lib/rpc.py
397 397
    return results
398 398

  
399 399
  def __call__(self, hosts, procedure, body, read_timeout, resolver_opts,
400
               _req_process_fn=http.client.ProcessRequests):
400
               _req_process_fn=None):
401 401
    """Makes an RPC request to a number of nodes.
402 402

  
403 403
    @type hosts: sequence
......
413 413
    assert read_timeout is not None, \
414 414
      "Missing RPC read timeout for procedure '%s'" % procedure
415 415

  
416
    if _req_process_fn is None:
417
      _req_process_fn = http.client.ProcessRequests
418

  
416 419
    (results, requests) = \
417 420
      self._PrepareRequests(self._resolver(hosts, resolver_opts), self._port,
418 421
                            procedure, body, read_timeout)
......
425 428

  
426 429

  
427 430
class _RpcClientBase:
428
  def __init__(self, resolver, encoder_fn, lock_monitor_cb=None):
431
  def __init__(self, resolver, encoder_fn, lock_monitor_cb=None,
432
               _req_process_fn=None):
429 433
    """Initializes this class.
430 434

  
431 435
    """
432
    self._proc = _RpcProcessor(resolver,
433
                               netutils.GetDaemonPort(constants.NODED),
434
                               lock_monitor_cb=lock_monitor_cb)
436
    proc = _RpcProcessor(resolver,
437
                         netutils.GetDaemonPort(constants.NODED),
438
                         lock_monitor_cb=lock_monitor_cb)
439
    self._proc = compat.partial(proc, _req_process_fn=_req_process_fn)
435 440
    self._encoder = compat.partial(self._EncodeArg, encoder_fn)
436 441

  
437 442
  @staticmethod

Also available in: Unified diff