Adding upstream version 0.10.1.upstream/0.10.1upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 195 insertions, 0 deletions

diff --git a/netaddr/tests/ip/test_cidr_v4.py b/netaddr/tests/ip/test_cidr_v4.py
new file mode 100644
index 0000000..30d7826
--- /dev/null
+++ b/netaddr/tests/ip/test_cidr_v4.py
@@ -0,0 +1,195 @@
+import random
+
+from netaddr import iprange_to_cidrs, IPNetwork, cidr_merge, cidr_exclude, largest_matching_cidr, smallest_matching_cidr, \
+    all_matching_cidrs
+
+
+def test_iprange_to_cidrs_worst_case_v4():
+    networks = iprange_to_cidrs('0.0.0.1', '255.255.255.254')
+    assert networks == [
+        IPNetwork('0.0.0.1/32'),
+        IPNetwork('0.0.0.2/31'),
+        IPNetwork('0.0.0.4/30'),
+        IPNetwork('0.0.0.8/29'),
+        IPNetwork('0.0.0.16/28'),
+        IPNetwork('0.0.0.32/27'),
+        IPNetwork('0.0.0.64/26'),
+        IPNetwork('0.0.0.128/25'),
+        IPNetwork('0.0.1.0/24'),
+        IPNetwork('0.0.2.0/23'),
+        IPNetwork('0.0.4.0/22'),
+        IPNetwork('0.0.8.0/21'),
+        IPNetwork('0.0.16.0/20'),
+        IPNetwork('0.0.32.0/19'),
+        IPNetwork('0.0.64.0/18'),
+        IPNetwork('0.0.128.0/17'),
+        IPNetwork('0.1.0.0/16'),
+        IPNetwork('0.2.0.0/15'),
+        IPNetwork('0.4.0.0/14'),
+        IPNetwork('0.8.0.0/13'),
+        IPNetwork('0.16.0.0/12'),
+        IPNetwork('0.32.0.0/11'),
+        IPNetwork('0.64.0.0/10'),
+        IPNetwork('0.128.0.0/9'),
+        IPNetwork('1.0.0.0/8'),
+        IPNetwork('2.0.0.0/7'),
+        IPNetwork('4.0.0.0/6'),
+        IPNetwork('8.0.0.0/5'),
+        IPNetwork('16.0.0.0/4'),
+        IPNetwork('32.0.0.0/3'),
+        IPNetwork('64.0.0.0/2'),
+        IPNetwork('128.0.0.0/2'),
+        IPNetwork('192.0.0.0/3'),
+        IPNetwork('224.0.0.0/4'),
+        IPNetwork('240.0.0.0/5'),
+        IPNetwork('248.0.0.0/6'),
+        IPNetwork('252.0.0.0/7'),
+        IPNetwork('254.0.0.0/8'),
+        IPNetwork('255.0.0.0/9'),
+        IPNetwork('255.128.0.0/10'),
+        IPNetwork('255.192.0.0/11'),
+        IPNetwork('255.224.0.0/12'),
+        IPNetwork('255.240.0.0/13'),
+        IPNetwork('255.248.0.0/14'),
+        IPNetwork('255.252.0.0/15'),
+        IPNetwork('255.254.0.0/16'),
+        IPNetwork('255.255.0.0/17'),
+        IPNetwork('255.255.128.0/18'),
+        IPNetwork('255.255.192.0/19'),
+        IPNetwork('255.255.224.0/20'),
+        IPNetwork('255.255.240.0/21'),
+        IPNetwork('255.255.248.0/22'),
+        IPNetwork('255.255.252.0/23'),
+        IPNetwork('255.255.254.0/24'),
+        IPNetwork('255.255.255.0/25'),
+        IPNetwork('255.255.255.128/26'),
+        IPNetwork('255.255.255.192/27'),
+        IPNetwork('255.255.255.224/28'),
+        IPNetwork('255.255.255.240/29'),
+        IPNetwork('255.255.255.248/30'),
+        IPNetwork('255.255.255.252/31'),
+        IPNetwork('255.255.255.254/32'),
+    ]
+
+
+def test_cidr_exclude_v4():
+    assert cidr_exclude('192.0.2.1/32', '192.0.2.1/32') == []
+    assert cidr_exclude('192.0.2.0/31', '192.0.2.1/32') == [IPNetwork('192.0.2.0/32')]
+    assert cidr_exclude('192.0.2.0/24', '192.0.2.128/25') == [IPNetwork('192.0.2.0/25')]
+    assert cidr_exclude('192.0.2.0/24', '192.0.2.128/27') == [
+        IPNetwork('192.0.2.0/25'),
+        IPNetwork('192.0.2.160/27'),
+        IPNetwork('192.0.2.192/26'),
+    ]
+
+    assert cidr_exclude('192.0.2.1/32', '192.0.2.0/24') == []
+    assert cidr_exclude('192.0.2.0/28', '192.0.2.16/32') == [IPNetwork('192.0.2.0/28')]
+    assert cidr_exclude('192.0.1.255/32', '192.0.2.0/28') == [IPNetwork('192.0.1.255/32')]
+
+
+def test_cidr_merge_v4():
+    assert cidr_merge(['192.0.128.0/24', '192.0.129.0/24']) == [IPNetwork('192.0.128.0/23')]
+    assert cidr_merge(['192.0.129.0/24', '192.0.130.0/24']) == [IPNetwork('192.0.129.0/24'), IPNetwork('192.0.130.0/24')]
+    assert cidr_merge(['192.0.2.112/30', '192.0.2.116/31', '192.0.2.118/31']) == [IPNetwork('192.0.2.112/29')]
+    assert cidr_merge(['192.0.2.112/30', '192.0.2.116/32', '192.0.2.118/31']) == [IPNetwork('192.0.2.112/30'), IPNetwork('192.0.2.116/32'), IPNetwork('192.0.2.118/31')]
+    assert cidr_merge(['192.0.2.112/31', '192.0.2.116/31', '192.0.2.118/31']) == [IPNetwork('192.0.2.112/31'), IPNetwork('192.0.2.116/30')]
+
+    assert cidr_merge([
+        '192.0.1.254/31',
+        '192.0.2.0/28',
+        '192.0.2.16/28',
+        '192.0.2.32/28',
+        '192.0.2.48/28',
+        '192.0.2.64/28',
+        '192.0.2.80/28',
+        '192.0.2.96/28',
+        '192.0.2.112/28',
+        '192.0.2.128/28',
+        '192.0.2.144/28',
+        '192.0.2.160/28',
+        '192.0.2.176/28',
+        '192.0.2.192/28',
+        '192.0.2.208/28',
+        '192.0.2.224/28',
+        '192.0.2.240/28',
+        '192.0.3.0/28']) == [
+        IPNetwork('192.0.1.254/31'),
+        IPNetwork('192.0.2.0/24'),
+        IPNetwork('192.0.3.0/28'),
+    ]
+
+
+def test_extended_cidr_merge():
+
+    orig_cidr_ipv4 = IPNetwork('192.0.2.0/23')
+    orig_cidr_ipv6 = IPNetwork('::192.0.2.0/120')
+
+    cidr_subnets = [str(c) for c in orig_cidr_ipv4.subnet(28)] + \
+                   list(orig_cidr_ipv4.subnet(28)) + \
+                   [str(c) for c in orig_cidr_ipv6.subnet(124)] + \
+                   list(orig_cidr_ipv6.subnet(124)) + \
+                   ['192.0.2.1/32', '192.0.2.128/25', '::192.0.2.92/128']
+
+    random.shuffle(cidr_subnets)
+
+    merged_cidrs = cidr_merge(cidr_subnets)
+
+    assert merged_cidrs == [IPNetwork('192.0.2.0/23'), IPNetwork('::192.0.2.0/120')]
+    assert merged_cidrs == [orig_cidr_ipv4, orig_cidr_ipv6]
+
+
+def test_whole_network_cidr_merge_v4():
+    assert cidr_merge(['0.0.0.0/0', '0.0.0.0']) == [IPNetwork('0.0.0.0/0')]
+    assert cidr_merge(['0.0.0.0/0', '255.255.255.255']) == [IPNetwork('0.0.0.0/0')]
+    assert cidr_merge(['0.0.0.0/0', '192.0.2.0/24', '10.0.0.0/8']) == [IPNetwork('0.0.0.0/0')]
+
+
+def test_largest_matching_cidr_v4():
+    assert largest_matching_cidr('192.0.2.0', ['192.0.2.0']) == IPNetwork('192.0.2.0/32')
+    assert largest_matching_cidr('192.0.2.0', ['10.0.0.1', '192.0.2.0']) == IPNetwork('192.0.2.0/32')
+    assert largest_matching_cidr('192.0.2.0', ['10.0.0.1', '192.0.2.0', '224.0.0.1']) == IPNetwork('192.0.2.0/32')
+    assert largest_matching_cidr('192.0.2.0', ['10.0.0.1', '224.0.0.1']) is None
+
+
+def test_smallest_matching_cidr_v4():
+    assert smallest_matching_cidr('192.0.2.0', ['10.0.0.1', '192.0.2.0', '224.0.0.1']) == IPNetwork('192.0.2.0/32')
+    assert smallest_matching_cidr('192.0.2.32', ['0.0.0.0/0', '10.0.0.0/8', '192.0.0.0/8', '192.0.1.0/24', '192.0.2.0/24', '192.0.3.0/24']) == IPNetwork('192.0.2.0/24')
+    assert smallest_matching_cidr('192.0.2.0', ['10.0.0.1', '224.0.0.1']) is None
+
+
+def test_all_matching_cidrs_v4():
+    assert all_matching_cidrs('192.0.2.32', ['0.0.0.0/0', '10.0.0.0/8', '192.0.0.0/8', '192.0.1.0/24', '192.0.2.0/24', '192.0.3.0/24']) == [
+        IPNetwork('0.0.0.0/0'),
+        IPNetwork('192.0.0.0/8'),
+        IPNetwork('192.0.2.0/24'),
+    ]
+
+def test_cidr_matching_v4():
+    networks = [str(c) for c in IPNetwork('192.0.2.128/27').supernet(22)]
+
+    assert networks == [
+        '192.0.0.0/22',
+        '192.0.2.0/23',
+        '192.0.2.0/24',
+        '192.0.2.128/25',
+        '192.0.2.128/26',
+    ]
+
+    assert all_matching_cidrs('192.0.2.0', networks) == [
+        IPNetwork('192.0.0.0/22'),
+        IPNetwork('192.0.2.0/23'),
+        IPNetwork('192.0.2.0/24'),
+    ]
+
+    assert smallest_matching_cidr('192.0.2.0', networks) == IPNetwork('192.0.2.0/24')
+    assert largest_matching_cidr('192.0.2.0', networks) == IPNetwork('192.0.0.0/22')
+
+# {{{
+# >>> all_matching_cidrs('192.0.2.0', ['192.0.2.0/24'])
+# [IPNetwork('192.0.2.0/24')]
+#
+# >>> all_matching_cidrs('192.0.2.0', ['::/96'])
+# []
+#
+#
+# }}}