From f863d7aad52ad89828dadd7e03d4329b62f688c8 Mon Sep 17 00:00:00 2001
From: Michael Hanselmann <hansmi@google.com>
Date: Wed, 4 Jan 2012 19:59:13 +0100
Subject: [PATCH] rpc._RpcProcessor: Make read timeout a non-keyword argument
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

It is always used and can be set to β€œNone” if intended.

Signed-off-by: Michael Hanselmann <hansmi@google.com>
Reviewed-by: Iustin Pop <iustin@google.com>
---
 lib/rpc.py                  |  5 ++---
 test/ganeti.rpc_unittest.py | 28 ++++++++++++----------------
 2 files changed, 14 insertions(+), 19 deletions(-)

diff --git a/lib/rpc.py b/lib/rpc.py
index bf05cfb25..256dd7028 100644
--- a/lib/rpc.py
+++ b/lib/rpc.py
@@ -391,7 +391,7 @@ class _RpcProcessor:
 
     return results
 
-  def __call__(self, hosts, procedure, body, read_timeout=None,
+  def __call__(self, hosts, procedure, body, read_timeout,
                _req_process_fn=http.client.ProcessRequests):
     """Makes an RPC request to a number of nodes.
 
@@ -463,8 +463,7 @@ class _RpcClientBase:
       pnbody = dict((n, serializer.DumpJson(prep_fn(n, enc_args)))
                     for n in node_list)
 
-    result = self._proc(node_list, procedure, pnbody,
-                        read_timeout=read_timeout)
+    result = self._proc(node_list, procedure, pnbody, read_timeout)
 
     if postproc_fn:
       return dict(map(lambda (key, value): (key, postproc_fn(value)),
diff --git a/test/ganeti.rpc_unittest.py b/test/ganeti.rpc_unittest.py
index e16802eb9..7eb529bcf 100755
--- a/test/ganeti.rpc_unittest.py
+++ b/test/ganeti.rpc_unittest.py
@@ -73,8 +73,8 @@ class TestRpcProcessor(unittest.TestCase):
     resolver = rpc._StaticResolver(["127.0.0.1"])
     http_proc = _FakeRequestProcessor(self._GetVersionResponse)
     proc = rpc._RpcProcessor(resolver, 24094)
-    result = proc(["localhost"], "version", {"localhost": ""},
-                  _req_process_fn=http_proc, read_timeout=60)
+    result = proc(["localhost"], "version", {"localhost": ""}, 60,
+                  _req_process_fn=http_proc)
     self.assertEqual(result.keys(), ["localhost"])
     lhresp = result["localhost"]
     self.assertFalse(lhresp.offline)
@@ -100,8 +100,7 @@ class TestRpcProcessor(unittest.TestCase):
     proc = rpc._RpcProcessor(resolver, 19176)
     host = "node31856"
     body = {host: ""}
-    result = proc([host], "version", body, _req_process_fn=http_proc,
-                  read_timeout=12356)
+    result = proc([host], "version", body, 12356, _req_process_fn=http_proc)
     self.assertEqual(result.keys(), [host])
     lhresp = result[host]
     self.assertFalse(lhresp.offline)
@@ -118,8 +117,7 @@ class TestRpcProcessor(unittest.TestCase):
     proc = rpc._RpcProcessor(resolver, 30668)
     host = "n17296"
     body = {host: ""}
-    result = proc([host], "version", body, _req_process_fn=http_proc,
-                  read_timeout=60)
+    result = proc([host], "version", body, 60, _req_process_fn=http_proc)
     self.assertEqual(result.keys(), [host])
     lhresp = result[host]
     self.assertTrue(lhresp.offline)
@@ -150,8 +148,7 @@ class TestRpcProcessor(unittest.TestCase):
     resolver = rpc._StaticResolver(nodes)
     http_proc = _FakeRequestProcessor(self._GetMultiVersionResponse)
     proc = rpc._RpcProcessor(resolver, 23245)
-    result = proc(nodes, "version", body, _req_process_fn=http_proc,
-                  read_timeout=60)
+    result = proc(nodes, "version", body, 60, _req_process_fn=http_proc,)
     self.assertEqual(sorted(result.keys()), sorted(nodes))
 
     for name in nodes:
@@ -180,8 +177,8 @@ class TestRpcProcessor(unittest.TestCase):
                                              errinfo))
       host = "aef9ur4i.example.com"
       body = {host: ""}
-      result = proc(body.keys(), "version", body,
-                    _req_process_fn=http_proc, read_timeout=60)
+      result = proc(body.keys(), "version", body, 60,
+                    _req_process_fn=http_proc)
       self.assertEqual(result.keys(), [host])
       lhresp = result[host]
       self.assertFalse(lhresp.offline)
@@ -230,8 +227,8 @@ class TestRpcProcessor(unittest.TestCase):
     http_proc = \
       _FakeRequestProcessor(compat.partial(self._GetHttpErrorResponse,
                                            httperrnodes, failnodes))
-    result = proc(nodes, "vg_list", body, _req_process_fn=http_proc,
-                  read_timeout=rpc._TMO_URGENT)
+    result = proc(nodes, "vg_list", body, rpc._TMO_URGENT,
+                  _req_process_fn=http_proc)
     self.assertEqual(sorted(result.keys()), sorted(nodes))
 
     for name in nodes:
@@ -275,8 +272,8 @@ class TestRpcProcessor(unittest.TestCase):
       http_proc = _FakeRequestProcessor(fn)
       host = "oqo7lanhly.example.com"
       body = {host: ""}
-      result = proc([host], "version", body,
-                    _req_process_fn=http_proc, read_timeout=60)
+      result = proc([host], "version", body, 60,
+                    _req_process_fn=http_proc)
       self.assertEqual(result.keys(), [host])
       lhresp = result[host]
       self.assertFalse(lhresp.offline)
@@ -307,8 +304,7 @@ class TestRpcProcessor(unittest.TestCase):
     proc = rpc._RpcProcessor(resolver, 18700)
     host = "node19759"
     body = {host: serializer.DumpJson(test_data)}
-    result = proc([host], "upload_file", body, _req_process_fn=http_proc,
-                  read_timeout=30)
+    result = proc([host], "upload_file", body, 30, _req_process_fn=http_proc)
     self.assertEqual(result.keys(), [host])
     lhresp = result[host]
     self.assertFalse(lhresp.offline)
-- 
GitLab