def test_generate_IPs(self): """ Test generate_IPs. """ dummy_list = ['127.0.0.1', '127.0.0.2', '127.0.0.3'] result = [] for new_ip in utils.generate_IPs("127.0.0.1-127.0.0.3"): result.append(new_ip) self.assertEqual(result, dummy_list) for new_ip in utils.generate_IPs('128.0.0.1-98.0.0.1'): self.assertIsNone(new_ip)
def parse_inbound_IPRule(self): try: action = int(self.cred['inbound_IPRule']['action']) temp_ip_inbound = [] if len(self.cred['inbound_IPRule']['ip_inbound']): list_of_IPs = str(self.cred['inbound_IPRule']['ip_inbound']) list_of_IPs = list_of_IPs.split(',') for IP in list_of_IPs: if '-' in IP: for new_ip in utils.generate_IPs(IP): if (new_ip not in temp_ip_inbound and utils.check_ip(new_ip)): temp_ip_inbound.append(str(new_ip).strip()) elif (utils.check_ip(IP)): if IP not in temp_ip_inbound: temp_ip_inbound.append(str(IP).strip()) return temp_ip_inbound, action except Exception as e: self.logger.log( "Error: " + str(e), logtype="error" ) # Return empty list and block action return [], 0