def testCorrectRankingAgainstMultipleRefIps(self): reference_ips = [ "130.83.1.1/16", "83.27.1.1/8" ] unranked_ips = [ "130.83.27.100", "83.57.32.2", "76.32.32.12" ] policy = SameIPPrefixPolicy(reference_ips) ranked_ips = policy.getRankedPeers(unranked_ips) self.assertEquals(ranked_ips['130.83.27.100'], 100) self.assertEquals(ranked_ips['83.57.32.2'], 100) self.assertEquals(ranked_ips['76.32.32.12'], 0)
def testCorrectRankingAgainstMultipleRefIps(self): reference_ips = ["130.83.1.1/16", "83.27.1.1/8"] unranked_ips = ["130.83.27.100", "83.57.32.2", "76.32.32.12"] policy = SameIPPrefixPolicy(reference_ips) ranked_ips = policy.getRankedPeers(unranked_ips) self.assertEquals(ranked_ips['130.83.27.100'], 100) self.assertEquals(ranked_ips['83.57.32.2'], 100) self.assertEquals(ranked_ips['76.32.32.12'], 0)
def testDefaultIfNoRefIpsProvided(self): try: policy = SameIPPrefixPolicy() unranked_ips = ["1.1.1.1", "192.168.1.1"] ranked_ips = policy.getRankedPeers(unranked_ips) #self.assertEquals(ranked_ips['1.1.1.1'], 100) #self.assertEquals(ranked_ips['192.168.1.1'], 0) raise Exception("should never happen!") except AssertionError: pass
def testCorrectRankingAgainstSingleRefIp(self): reference_ip = "130.83.1.1/16" policy = SameIPPrefixPolicy([reference_ip]) list_of_unranked_ips = ["130.83.85.17", "88.56.172.31", "127.0.0.1", "130.83.32.67"] dict_of_ranked_ips = policy.getRankedPeers(list_of_unranked_ips) self.assertEquals(dict_of_ranked_ips['130.83.85.17'], 100) self.assertEquals(dict_of_ranked_ips['130.83.32.67'], 100) self.assertEquals(dict_of_ranked_ips['88.56.172.31'], 0) self.assertEquals(dict_of_ranked_ips['127.0.0.1'], 0)
def testDefaultIfNoRefIpsProvided(self): try: policy = SameIPPrefixPolicy() unranked_ips = ["1.1.1.1", "192.168.1.1"] ranked_ips = policy.getRankedPeers(unranked_ips) #self.assertEquals(ranked_ips['1.1.1.1'], 100) #self.assertEquals(ranked_ips['192.168.1.1'], 0) raise Exception("should never happen!") except AssertionError: pass
def testCorrectRankingAgainstSingleRefIp(self): reference_ip = "130.83.1.1/16" policy = SameIPPrefixPolicy([reference_ip]) list_of_unranked_ips = [ "130.83.85.17", "88.56.172.31", "127.0.0.1", "130.83.32.67" ] dict_of_ranked_ips = policy.getRankedPeers(list_of_unranked_ips) self.assertEquals(dict_of_ranked_ips['130.83.85.17'], 100) self.assertEquals(dict_of_ranked_ips['130.83.32.67'], 100) self.assertEquals(dict_of_ranked_ips['88.56.172.31'], 0) self.assertEquals(dict_of_ranked_ips['127.0.0.1'], 0)