Revision db04ce5d test/ganeti.rpc_unittest.py

b/test/ganeti.rpc_unittest.py
73 73
    resolver = rpc._StaticResolver(["127.0.0.1"])
74 74
    http_proc = _FakeRequestProcessor(self._GetVersionResponse)
75 75
    proc = rpc._RpcProcessor(resolver, 24094)
76
    result = proc(["localhost"], "version", None, _req_process_fn=http_proc)
76
    result = proc(["localhost"], "version", None, _req_process_fn=http_proc,
77
                  read_timeout=60)
77 78
    self.assertEqual(result.keys(), ["localhost"])
78 79
    lhresp = result["localhost"]
79 80
    self.assertFalse(lhresp.offline)
......
113 114
    resolver = rpc._StaticResolver([rpc._OFFLINE])
114 115
    http_proc = _FakeRequestProcessor(NotImplemented)
115 116
    proc = rpc._RpcProcessor(resolver, 30668)
116
    result = proc(["n17296"], "version", None, _req_process_fn=http_proc)
117
    result = proc(["n17296"], "version", None, _req_process_fn=http_proc,
118
                  read_timeout=60)
117 119
    self.assertEqual(result.keys(), ["n17296"])
118 120
    lhresp = result["n17296"]
119 121
    self.assertTrue(lhresp.offline)
......
143 145
    resolver = rpc._StaticResolver(nodes)
144 146
    http_proc = _FakeRequestProcessor(self._GetMultiVersionResponse)
145 147
    proc = rpc._RpcProcessor(resolver, 23245)
146
    result = proc(nodes, "version", None, _req_process_fn=http_proc)
148
    result = proc(nodes, "version", None, _req_process_fn=http_proc,
149
                  read_timeout=60)
147 150
    self.assertEqual(sorted(result.keys()), sorted(nodes))
148 151

  
149 152
    for name in nodes:
......
171 174
        _FakeRequestProcessor(compat.partial(self._GetVersionResponseFail,
172 175
                                             errinfo))
173 176
      result = proc(["aef9ur4i.example.com"], "version", None,
174
                    _req_process_fn=http_proc)
177
                    _req_process_fn=http_proc, read_timeout=60)
175 178
      self.assertEqual(result.keys(), ["aef9ur4i.example.com"])
176 179
      lhresp = result["aef9ur4i.example.com"]
177 180
      self.assertFalse(lhresp.offline)
......
263 266
    for fn in [self._GetInvalidResponseA, self._GetInvalidResponseB]:
264 267
      http_proc = _FakeRequestProcessor(fn)
265 268
      result = proc(["oqo7lanhly.example.com"], "version", None,
266
                    _req_process_fn=http_proc)
269
                    _req_process_fn=http_proc, read_timeout=60)
267 270
      self.assertEqual(result.keys(), ["oqo7lanhly.example.com"])
268 271
      lhresp = result["oqo7lanhly.example.com"]
269 272
      self.assertFalse(lhresp.offline)

Also available in: Unified diff