Revision b705c7a6 test/ganeti.rpc_unittest.py

b/test/ganeti.rpc_unittest.py
248 248
    node_list = ["node%d.example.com" % n for n in range(0, 255, 13)]
249 249
    ssc = GetFakeSimpleStoreClass(lambda s: [])
250 250
    node_addr_map = dict(zip(node_list, addr_list))
251
    nslookup_fn = lambda name: (None, None, [node_addr_map.get(name)])
251
    nslookup_fn = lambda name: node_addr_map.get(name)
252 252
    result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn)
253 253
    self.assertEqual(result, addr_list)
254 254

  
......
259 259
    node_addr_list = [ " ".join(t) for t in zip(node_list[n:], addr_list[n:])]
260 260
    ssc = GetFakeSimpleStoreClass(lambda s: node_addr_list)
261 261
    node_addr_map = dict(zip(node_list[:n], addr_list[:n]))
262
    nslookup_fn = lambda name: (None, None, [node_addr_map.get(name)])
262
    nslookup_fn = lambda name: node_addr_map.get(name)
263 263
    result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn)
264 264
    self.assertEqual(result, addr_list)
265 265

  

Also available in: Unified diff