def test_ips_to_ranges_range(self):
from ExportIndicators import ips_to_ranges, COLLAPSE_TO_RANGES
ip_list = [IPAddress("1.1.1.1"), IPAddress("25.24.23.22"), IPAddress("22.21.20.19"),
IPAddress("1.1.1.2"), IPAddress("1.2.3.4"), IPAddress("1.1.1.3"), IPAddress("2.2.2.2"),
IPAddress("1.2.3.5")]
ip_range_list = ips_to_ranges(ip_list, COLLAPSE_TO_RANGES)
assert "1.1.1.1-1.1.1.3" in ip_range_list
assert "1.2.3.4-1.2.3.5" in ip_range_list
assert "1.1.1.2" not in ip_range_list
assert "2.2.2.2" in ip_range_list
assert "25.24.23.22" in ip_range_list
def test_ips_to_ranges_cidr(self):
from ExportIndicators import ips_to_ranges, COLLAPSE_TO_CIDR
ip_list = [IPAddress("1.1.1.1"), IPAddress("25.24.23.22"), IPAddress("22.21.20.19"),
IPAddress("1.1.1.2"), IPAddress("1.2.3.4"), IPAddress("1.1.1.3"), IPAddress("2.2.2.2"),
IPAddress("1.2.3.5")]
ip_range_list = ips_to_ranges(ip_list, COLLAPSE_TO_CIDR)
assert "1.1.1.1" in ip_range_list
assert "1.1.1.2/31" in ip_range_list
assert "1.2.3.4/31" in ip_range_list
assert "1.2.3.5" not in ip_range_list
assert "1.1.1.3" not in ip_range_list
assert "2.2.2.2" in ip_range_list
assert "25.24.23.22" in ip_range_list
