def testAddressLookupIPv6(self): addr_list = ["2001:db8::%d" % n for n in range(0, 255, 13)] node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] node_addr_list = [ " ".join(t) for t in zip(node_list, addr_list)] ssc = GetFakeSimpleStoreClass(lambda _: node_addr_list) result = rpc._AddressLookup(node_list, ssc=ssc) self.assertEqual(result, addr_list)
def testAddressLookupNSLookup(self): addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] ssc = GetFakeSimpleStoreClass(lambda _: []) node_addr_map = dict(zip(node_list, addr_list)) nslookup_fn = lambda name, family=None: node_addr_map.get(name) result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn) self.assertEqual(result, addr_list)
def testAddressLookupBoth(self): addr_list = ["192.0.2.%d" % n for n in range(0, 255, 13)] node_list = ["node%d.example.com" % n for n in range(0, 255, 13)] n = len(addr_list) / 2 node_addr_list = [ " ".join(t) for t in zip(node_list[n:], addr_list[n:])] ssc = GetFakeSimpleStoreClass(lambda _: node_addr_list) node_addr_map = dict(zip(node_list[:n], addr_list[:n])) nslookup_fn = lambda name, family=None: node_addr_map.get(name) result = rpc._AddressLookup(node_list, ssc=ssc, nslookup_fn=nslookup_fn) self.assertEqual(result, addr_list)