import pickle
import sys
import weakref

import pytest

from netaddr import IPAddress, IPNetwork, IPRange, IPSet, cidr_exclude


def test_ipset_basic_api():
    range1 = IPRange('192.0.2.1', '192.0.2.15')

    ip_list = [
        IPAddress('192.0.2.1'),
        '192.0.2.2/31',
        IPNetwork('192.0.2.4/31'),
        IPAddress('192.0.2.6'),
        IPAddress('192.0.2.7'),
        '192.0.2.8',
        '192.0.2.9',
        IPAddress('192.0.2.10'),
        IPAddress('192.0.2.11'),
        IPNetwork('192.0.2.12/30'),
    ]

    set1 = IPSet(range1.cidrs())

    set2 = IPSet(ip_list)

    assert set2 == IPSet(
        [
            '192.0.2.1/32',
            '192.0.2.2/31',
            '192.0.2.4/30',
            '192.0.2.8/29',
        ]
    )

    assert set1 == set2
    assert set2.pop() in set1
    assert set1 != set2


def test_ipset_empty():
    assert IPSet() == IPSet([])
    empty_set = IPSet([])
    assert IPSet([]) == empty_set
    assert len(empty_set) == 0


def test_ipset_constructor():
    assert IPSet(['192.0.2.0']) == IPSet(['192.0.2.0/32'])
    assert IPSet([IPAddress('192.0.2.0')]) == IPSet(['192.0.2.0/32'])
    assert IPSet([IPNetwork('192.0.2.0')]) == IPSet(['192.0.2.0/32'])
    assert IPSet(IPNetwork('1234::/32')) == IPSet(['1234::/32'])
    assert IPSet([IPNetwork('192.0.2.0/24')]) == IPSet(['192.0.2.0/24'])
    assert IPSet(IPSet(['192.0.2.0/32'])) == IPSet(['192.0.2.0/32'])
    assert IPSet(IPRange('10.0.0.0', '10.0.1.31')) == IPSet(['10.0.0.0/24', '10.0.1.0/27'])
    assert IPSet(IPRange('0.0.0.0', '255.255.255.255')) == IPSet(['0.0.0.0/0'])
    assert IPSet([IPRange('10.0.0.0', '10.0.1.31')]) == IPSet(IPRange('10.0.0.0', '10.0.1.31'))


def test_ipset_iteration():
    assert list(IPSet(['192.0.2.0/28', '::192.0.2.0/124'])) == [
        IPAddress('192.0.2.0'),
        IPAddress('192.0.2.1'),
        IPAddress('192.0.2.2'),
        IPAddress('192.0.2.3'),
        IPAddress('192.0.2.4'),
        IPAddress('192.0.2.5'),
        IPAddress('192.0.2.6'),
        IPAddress('192.0.2.7'),
        IPAddress('192.0.2.8'),
        IPAddress('192.0.2.9'),
        IPAddress('192.0.2.10'),
        IPAddress('192.0.2.11'),
        IPAddress('192.0.2.12'),
        IPAddress('192.0.2.13'),
        IPAddress('192.0.2.14'),
        IPAddress('192.0.2.15'),
        IPAddress('::192.0.2.0'),
        IPAddress('::192.0.2.1'),
        IPAddress('::192.0.2.2'),
        IPAddress('::192.0.2.3'),
        IPAddress('::192.0.2.4'),
        IPAddress('::192.0.2.5'),
        IPAddress('::192.0.2.6'),
        IPAddress('::192.0.2.7'),
        IPAddress('::192.0.2.8'),
        IPAddress('::192.0.2.9'),
        IPAddress('::192.0.2.10'),
        IPAddress('::192.0.2.11'),
        IPAddress('::192.0.2.12'),
        IPAddress('::192.0.2.13'),
        IPAddress('::192.0.2.14'),
        IPAddress('::192.0.2.15'),
    ]


def test_ipset_member_insertion_and_deletion():
    s1 = IPSet()
    s1.add('192.0.2.0')
    assert s1 == IPSet(['192.0.2.0/32'])

    s1.remove('192.0.2.0')
    assert s1 == IPSet([])

    s1.add(IPRange('10.0.0.0', '10.0.0.255'))
    assert s1 == IPSet(['10.0.0.0/24'])

    s1.remove(IPRange('10.0.0.128', '10.10.10.10'))
    assert s1 == IPSet(['10.0.0.0/25'])


def test_ipset_membership():
    iprange = IPRange('192.0.1.255', '192.0.2.16')

    assert iprange.cidrs() == [
        IPNetwork('192.0.1.255/32'),
        IPNetwork('192.0.2.0/28'),
        IPNetwork('192.0.2.16/32'),
    ]

    ipset = IPSet(['192.0.2.0/28'])

    assert [(str(ip), ip in ipset) for ip in iprange] == [
        ('192.0.1.255', False),
        ('192.0.2.0', True),
        ('192.0.2.1', True),
        ('192.0.2.2', True),
        ('192.0.2.3', True),
        ('192.0.2.4', True),
        ('192.0.2.5', True),
        ('192.0.2.6', True),
        ('192.0.2.7', True),
        ('192.0.2.8', True),
        ('192.0.2.9', True),
        ('192.0.2.10', True),
        ('192.0.2.11', True),
        ('192.0.2.12', True),
        ('192.0.2.13', True),
        ('192.0.2.14', True),
        ('192.0.2.15', True),
        ('192.0.2.16', False),
    ]


def test_ipset_membership_largest():
    ipset = IPSet(['0.0.0.0/0'])

    assert IPAddress('10.0.0.1') in ipset
    assert IPAddress('0.0.0.0') in ipset
    assert IPAddress('255.255.255.0') in ipset
    assert IPNetwork('10.0.0.0/24') in ipset
    assert IPAddress('::1') not in ipset


def test_set_membership_smallest():
    ipset = IPSet(['10.0.0.42/32'])

    assert IPAddress('10.0.0.42') in ipset
    assert IPNetwork('10.0.0.42/32') in ipset

    assert IPAddress('10.0.0.41') not in ipset
    assert IPAddress('10.0.0.43') not in ipset
    assert IPNetwork('10.0.0.42/31') not in ipset


def test_ipset_unions():
    assert IPSet(['192.0.2.0']) == IPSet(['192.0.2.0/32'])
    assert IPSet(['192.0.2.0']) | IPSet(['192.0.2.1']) == IPSet(['192.0.2.0/31'])
    assert IPSet(['192.0.2.0']) | IPSet(['192.0.2.1']) | IPSet(['192.0.2.3']) == IPSet(
        ['192.0.2.0/31', '192.0.2.3/32']
    )
    assert IPSet(['192.0.2.0']) | IPSet(['192.0.2.1']) | IPSet(['192.0.2.3/30']) == IPSet(
        ['192.0.2.0/30']
    )
    assert IPSet(['192.0.2.0']) | IPSet(['192.0.2.1']) | IPSet(['192.0.2.3/31']) == IPSet(
        ['192.0.2.0/30']
    )
    assert IPSet(['192.0.2.0/24']) | IPSet(['192.0.3.0/24']) | IPSet(['192.0.4.0/24']) == IPSet(
        ['192.0.2.0/23', '192.0.4.0/24']
    )


def test_ipset_unions_intersections_differences():
    adj_cidrs = list(IPNetwork('192.0.2.0/24').subnet(28))
    even_cidrs = adj_cidrs[::2]
    evens = IPSet(even_cidrs)

    assert evens == IPSet(
        [
            '192.0.2.0/28',
            '192.0.2.32/28',
            '192.0.2.64/28',
            '192.0.2.96/28',
            '192.0.2.128/28',
            '192.0.2.160/28',
            '192.0.2.192/28',
            '192.0.2.224/28',
        ]
    )

    assert IPSet(['192.0.2.0/24']) & evens == IPSet(
        [
            '192.0.2.0/28',
            '192.0.2.32/28',
            '192.0.2.64/28',
            '192.0.2.96/28',
            '192.0.2.128/28',
            '192.0.2.160/28',
            '192.0.2.192/28',
            '192.0.2.224/28',
        ]
    )

    odds = IPSet(['192.0.2.0/24']) ^ evens
    assert odds == IPSet(
        [
            '192.0.2.16/28',
            '192.0.2.48/28',
            '192.0.2.80/28',
            '192.0.2.112/28',
            '192.0.2.144/28',
            '192.0.2.176/28',
            '192.0.2.208/28',
            '192.0.2.240/28',
        ]
    )

    assert evens | odds == IPSet(['192.0.2.0/24'])
    assert evens & odds == IPSet([])
    assert evens ^ odds == IPSet(['192.0.2.0/24'])


def test_ipset_supersets_and_subsets():
    s1 = IPSet(['192.0.2.0/24', '192.0.4.0/24'])
    s2 = IPSet(['192.0.2.0', '192.0.4.0'])

    assert s1.issuperset(s2)
    assert s2.issubset(s1)
    assert not s2.issuperset(s1)
    assert not s1.issubset(s2)

    ipv4_addr_space = IPSet(['0.0.0.0/0'])
    private = IPSet(
        ['10.0.0.0/8', '172.16.0.0/12', '192.0.2.0/24', '192.168.0.0/16', '239.192.0.0/14']
    )
    reserved = IPSet(
        [
            '225.0.0.0/8',
            '226.0.0.0/7',
            '228.0.0.0/6',
            '234.0.0.0/7',
            '236.0.0.0/7',
            '238.0.0.0/8',
            '240.0.0.0/4',
        ]
    )
    unavailable = reserved | private
    available = ipv4_addr_space ^ unavailable

    assert [tuple(map(str, (cidr, cidr[0], cidr[-1]))) for cidr in available.iter_cidrs()] == [
        ('0.0.0.0/5', '0.0.0.0', '7.255.255.255'),
        ('8.0.0.0/7', '8.0.0.0', '9.255.255.255'),
        ('11.0.0.0/8', '11.0.0.0', '11.255.255.255'),
        ('12.0.0.0/6', '12.0.0.0', '15.255.255.255'),
        ('16.0.0.0/4', '16.0.0.0', '31.255.255.255'),
        ('32.0.0.0/3', '32.0.0.0', '63.255.255.255'),
        ('64.0.0.0/2', '64.0.0.0', '127.255.255.255'),
        ('128.0.0.0/3', '128.0.0.0', '159.255.255.255'),
        ('160.0.0.0/5', '160.0.0.0', '167.255.255.255'),
        ('168.0.0.0/6', '168.0.0.0', '171.255.255.255'),
        ('172.0.0.0/12', '172.0.0.0', '172.15.255.255'),
        ('172.32.0.0/11', '172.32.0.0', '172.63.255.255'),
        ('172.64.0.0/10', '172.64.0.0', '172.127.255.255'),
        ('172.128.0.0/9', '172.128.0.0', '172.255.255.255'),
        ('173.0.0.0/8', '173.0.0.0', '173.255.255.255'),
        ('174.0.0.0/7', '174.0.0.0', '175.255.255.255'),
        ('176.0.0.0/4', '176.0.0.0', '191.255.255.255'),
        ('192.0.0.0/23', '192.0.0.0', '192.0.1.255'),
        ('192.0.3.0/24', '192.0.3.0', '192.0.3.255'),
        ('192.0.4.0/22', '192.0.4.0', '192.0.7.255'),
        ('192.0.8.0/21', '192.0.8.0', '192.0.15.255'),
        ('192.0.16.0/20', '192.0.16.0', '192.0.31.255'),
        ('192.0.32.0/19', '192.0.32.0', '192.0.63.255'),
        ('192.0.64.0/18', '192.0.64.0', '192.0.127.255'),
        ('192.0.128.0/17', '192.0.128.0', '192.0.255.255'),
        ('192.1.0.0/16', '192.1.0.0', '192.1.255.255'),
        ('192.2.0.0/15', '192.2.0.0', '192.3.255.255'),
        ('192.4.0.0/14', '192.4.0.0', '192.7.255.255'),
        ('192.8.0.0/13', '192.8.0.0', '192.15.255.255'),
        ('192.16.0.0/12', '192.16.0.0', '192.31.255.255'),
        ('192.32.0.0/11', '192.32.0.0', '192.63.255.255'),
        ('192.64.0.0/10', '192.64.0.0', '192.127.255.255'),
        ('192.128.0.0/11', '192.128.0.0', '192.159.255.255'),
        ('192.160.0.0/13', '192.160.0.0', '192.167.255.255'),
        ('192.169.0.0/16', '192.169.0.0', '192.169.255.255'),
        ('192.170.0.0/15', '192.170.0.0', '192.171.255.255'),
        ('192.172.0.0/14', '192.172.0.0', '192.175.255.255'),
        ('192.176.0.0/12', '192.176.0.0', '192.191.255.255'),
        ('192.192.0.0/10', '192.192.0.0', '192.255.255.255'),
        ('193.0.0.0/8', '193.0.0.0', '193.255.255.255'),
        ('194.0.0.0/7', '194.0.0.0', '195.255.255.255'),
        ('196.0.0.0/6', '196.0.0.0', '199.255.255.255'),
        ('200.0.0.0/5', '200.0.0.0', '207.255.255.255'),
        ('208.0.0.0/4', '208.0.0.0', '223.255.255.255'),
        ('224.0.0.0/8', '224.0.0.0', '224.255.255.255'),
        ('232.0.0.0/7', '232.0.0.0', '233.255.255.255'),
        ('239.0.0.0/9', '239.0.0.0', '239.127.255.255'),
        ('239.128.0.0/10', '239.128.0.0', '239.191.255.255'),
        ('239.196.0.0/14', '239.196.0.0', '239.199.255.255'),
        ('239.200.0.0/13', '239.200.0.0', '239.207.255.255'),
        ('239.208.0.0/12', '239.208.0.0', '239.223.255.255'),
        ('239.224.0.0/11', '239.224.0.0', '239.255.255.255'),
    ]

    assert ipv4_addr_space ^ available == IPSet(
        [
            '10.0.0.0/8',
            '172.16.0.0/12',
            '192.0.2.0/24',
            '192.168.0.0/16',
            '225.0.0.0/8',
            '226.0.0.0/7',
            '228.0.0.0/6',
            '234.0.0.0/7',
            '236.0.0.0/7',
            '238.0.0.0/8',
            '239.192.0.0/14',
            '240.0.0.0/4',
        ]
    )


def test_combined_ipv4_and_ipv6_ipsets():
    s1 = IPSet(['192.0.2.0', '::192.0.2.0', '192.0.2.2', '::192.0.2.2'])
    s2 = IPSet(['192.0.2.2', '::192.0.2.2', '192.0.2.4', '::192.0.2.4'])

    assert s1 | s2 == IPSet(
        [
            '192.0.2.0/32',
            '192.0.2.2/32',
            '192.0.2.4/32',
            '::192.0.2.0/128',
            '::192.0.2.2/128',
            '::192.0.2.4/128',
        ]
    )

    assert s2 | s1 == IPSet(
        [
            '192.0.2.0/32',
            '192.0.2.2/32',
            '192.0.2.4/32',
            '::192.0.2.0/128',
            '::192.0.2.2/128',
            '::192.0.2.4/128',
        ]
    )

    assert s1 & s2 == IPSet(['192.0.2.2/32', '::192.0.2.2/128'])
    assert s1 - s2 == IPSet(['192.0.2.0/32', '::192.0.2.0/128'])
    assert s2 - s1 == IPSet(['192.0.2.4/32', '::192.0.2.4/128'])
    assert s1 ^ s2 == IPSet(['192.0.2.0/32', '192.0.2.4/32', '::192.0.2.0/128', '::192.0.2.4/128'])


def test_disjointed_ipsets():
    s1 = IPSet(['192.0.2.0', '192.0.2.1', '192.0.2.2'])
    s2 = IPSet(['192.0.2.2', '192.0.2.3', '192.0.2.4'])

    assert s1 & s2 == IPSet(['192.0.2.2/32'])
    assert not s1.isdisjoint(s2)

    s3 = IPSet(['192.0.2.0', '192.0.2.1'])
    s4 = IPSet(['192.0.2.3', '192.0.2.4'])

    assert s3 & s4 == IPSet([])
    assert s3.isdisjoint(s4)


def test_ipset_updates():
    s1 = IPSet(['192.0.2.0/25'])
    s2 = IPSet(['192.0.2.128/25'])

    s1.update(s2)
    assert s1 == IPSet(['192.0.2.0/24'])

    s1.update(['192.0.0.0/24', '192.0.1.0/24', '192.0.3.0/24'])
    assert s1 == IPSet(['192.0.0.0/22'])

    expected = IPSet(['192.0.1.0/24', '192.0.2.0/24'])

    s3 = IPSet(['192.0.1.0/24'])
    s3.update(IPRange('192.0.2.0', '192.0.2.255'))
    assert s3 == expected

    s4 = IPSet(['192.0.1.0/24'])
    s4.update([IPRange('192.0.2.0', '192.0.2.100'), IPRange('192.0.2.50', '192.0.2.255')])
    assert s4 == expected


def test_ipset_clear():
    ipset = IPSet(['10.0.0.0/16'])
    ipset.update(IPRange('10.1.0.0', '10.1.255.255'))
    assert ipset == IPSet(['10.0.0.0/15'])

    ipset.clear()
    assert ipset == IPSet([])


def test_ipset_cidr_fracturing():
    s1 = IPSet(['0.0.0.0/0'])
    s1.remove('255.255.255.255')
    assert s1 == IPSet(
        [
            '0.0.0.0/1',
            '128.0.0.0/2',
            '192.0.0.0/3',
            '224.0.0.0/4',
            '240.0.0.0/5',
            '248.0.0.0/6',
            '252.0.0.0/7',
            '254.0.0.0/8',
            '255.0.0.0/9',
            '255.128.0.0/10',
            '255.192.0.0/11',
            '255.224.0.0/12',
            '255.240.0.0/13',
            '255.248.0.0/14',
            '255.252.0.0/15',
            '255.254.0.0/16',
            '255.255.0.0/17',
            '255.255.128.0/18',
            '255.255.192.0/19',
            '255.255.224.0/20',
            '255.255.240.0/21',
            '255.255.248.0/22',
            '255.255.252.0/23',
            '255.255.254.0/24',
            '255.255.255.0/25',
            '255.255.255.128/26',
            '255.255.255.192/27',
            '255.255.255.224/28',
            '255.255.255.240/29',
            '255.255.255.248/30',
            '255.255.255.252/31',
            '255.255.255.254/32',
        ]
    )

    cidrs = s1.iter_cidrs()
    assert len(cidrs) == 32
    assert list(cidrs) == [
        IPNetwork('0.0.0.0/1'),
        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'),
    ]

    assert cidrs == cidr_exclude('0.0.0.0/0', '255.255.255.255')

    s1.remove('0.0.0.0')

    assert s1 == IPSet(
        [
            '0.0.0.1/32',
            '0.0.0.2/31',
            '0.0.0.4/30',
            '0.0.0.8/29',
            '0.0.0.16/28',
            '0.0.0.32/27',
            '0.0.0.64/26',
            '0.0.0.128/25',
            '0.0.1.0/24',
            '0.0.2.0/23',
            '0.0.4.0/22',
            '0.0.8.0/21',
            '0.0.16.0/20',
            '0.0.32.0/19',
            '0.0.64.0/18',
            '0.0.128.0/17',
            '0.1.0.0/16',
            '0.2.0.0/15',
            '0.4.0.0/14',
            '0.8.0.0/13',
            '0.16.0.0/12',
            '0.32.0.0/11',
            '0.64.0.0/10',
            '0.128.0.0/9',
            '1.0.0.0/8',
            '2.0.0.0/7',
            '4.0.0.0/6',
            '8.0.0.0/5',
            '16.0.0.0/4',
            '32.0.0.0/3',
            '64.0.0.0/2',
            '128.0.0.0/2',
            '192.0.0.0/3',
            '224.0.0.0/4',
            '240.0.0.0/5',
            '248.0.0.0/6',
            '252.0.0.0/7',
            '254.0.0.0/8',
            '255.0.0.0/9',
            '255.128.0.0/10',
            '255.192.0.0/11',
            '255.224.0.0/12',
            '255.240.0.0/13',
            '255.248.0.0/14',
            '255.252.0.0/15',
            '255.254.0.0/16',
            '255.255.0.0/17',
            '255.255.128.0/18',
            '255.255.192.0/19',
            '255.255.224.0/20',
            '255.255.240.0/21',
            '255.255.248.0/22',
            '255.255.252.0/23',
            '255.255.254.0/24',
            '255.255.255.0/25',
            '255.255.255.128/26',
            '255.255.255.192/27',
            '255.255.255.224/28',
            '255.255.255.240/29',
            '255.255.255.248/30',
            '255.255.255.252/31',
            '255.255.255.254/32',
        ]
    )

    assert len(list(s1.iter_cidrs())) == 62

    s1.add('255.255.255.255')
    s1.add('0.0.0.0')

    assert s1 == IPSet(['0.0.0.0/0'])


def test_ipset_with_iprange():
    s1 = IPSet(['10.0.0.0/25', '10.0.0.128/25'])
    assert s1.iprange() == IPRange('10.0.0.0', '10.0.0.255')

    assert s1.iscontiguous()

    s1.remove('10.0.0.16')
    assert s1 == IPSet(
        [
            '10.0.0.0/28',
            '10.0.0.17/32',
            '10.0.0.18/31',
            '10.0.0.20/30',
            '10.0.0.24/29',
            '10.0.0.32/27',
            '10.0.0.64/26',
            '10.0.0.128/25',
        ]
    )

    assert not s1.iscontiguous()

    with pytest.raises(ValueError):
        s1.iprange()

    assert list(s1.iter_ipranges()) == [
        IPRange('10.0.0.0', '10.0.0.15'),
        IPRange('10.0.0.17', '10.0.0.255'),
    ]

    s2 = IPSet(['0.0.0.0/0'])
    assert s2.iscontiguous()
    assert s2.iprange() == IPRange('0.0.0.0', '255.255.255.255')
    #
    s3 = IPSet()
    assert s3.iscontiguous()
    assert s3.iprange() is None

    s4 = IPSet(IPRange('10.0.0.0', '10.0.0.8'))
    assert s4.iscontiguous()


def test_ipset_pickling():
    ip_data = IPSet(['10.0.0.0/16', 'fe80::/64'])
    buf = pickle.dumps(ip_data)
    ip_data_unpickled = pickle.loads(buf)
    assert ip_data == ip_data_unpickled


def test_ipset_comparison():
    s1 = IPSet(['fc00::/2'])
    s2 = IPSet(['fc00::/3'])

    assert s1 > s2
    assert not s1 < s2
    assert s1 != s2


def test_ipset_adding_and_removing_members_ip_addresses_as_ints():
    s1 = IPSet(['10.0.0.0/25'])

    s1.add('10.0.0.0/24')
    assert s1 == IPSet(['10.0.0.0/24'])

    integer1 = int(IPAddress('10.0.0.1'))
    integer2 = int(IPAddress('fe80::'))
    integer3 = int(IPAddress('10.0.0.2'))

    s2 = IPSet([integer1, integer2])
    assert s2 == IPSet(['10.0.0.1/32', 'fe80::/128'])

    s2.add(integer3)
    assert s2 == IPSet(['10.0.0.1/32', '10.0.0.2/32', 'fe80::/128'])

    s2.remove(integer2)
    assert s2 == IPSet(['10.0.0.1/32', '10.0.0.2/32'])

    s2.update([integer2])
    assert s2 == IPSet(['10.0.0.1/32', '10.0.0.2/32', 'fe80::/128'])


def test_ipset_operations_with_combined_ipv4_and_ipv6():
    s1 = IPSet(['192.0.2.0', '::192.0.2.0', '192.0.2.2', '::192.0.2.2'])
    s2 = IPSet(['192.0.2.2', '::192.0.2.2', '192.0.2.4', '::192.0.2.4'])
    s3 = IPSet(['0.0.0.1', '10.0.0.64/30', '255.255.255.1'])
    s4 = IPSet(['10.0.0.64', '10.0.0.66'])
    s4b = IPSet(['10.0.0.64', '10.0.0.66', '111.111.111.111'])
    s5 = IPSet(['10.0.0.65', '10.0.0.67'])
    s6 = IPSet(['2405:8100::/32'])

    assert bool(s6)
    assert not bool(IPSet())

    #   set intersection
    assert s2 & s1 == IPSet(['192.0.2.2/32', '::192.0.2.2/128'])
    assert s3 & s4 == IPSet(['10.0.0.64/32', '10.0.0.66/32'])
    assert s4 & s3 == IPSet(['10.0.0.64/32', '10.0.0.66/32'])
    assert s3 & s5 == IPSet(['10.0.0.65/32', '10.0.0.67/32'])
    assert s5 & s3 == IPSet(['10.0.0.65/32', '10.0.0.67/32'])

    #   set difference
    assert s3 - s4 == IPSet(['0.0.0.1/32', '10.0.0.65/32', '10.0.0.67/32', '255.255.255.1/32'])
    assert s4 - s3 == IPSet([])
    assert s3 - s4b == IPSet(['0.0.0.1/32', '10.0.0.65/32', '10.0.0.67/32', '255.255.255.1/32'])
    assert s3 - s5 == IPSet(['0.0.0.1/32', '10.0.0.64/32', '10.0.0.66/32', '255.255.255.1/32'])
    assert s5 - s3 == IPSet([])

    #   set symmetric difference
    assert s2 ^ s1 == IPSet(['192.0.2.0/32', '192.0.2.4/32', '::192.0.2.0/128', '::192.0.2.4/128'])
    assert IPSet([]) ^ IPSet([]) == IPSet([])
    assert IPSet(['0.0.0.1/32']) ^ IPSet([]) == IPSet(['0.0.0.1/32'])
    assert IPSet(['0.0.0.1/32']) ^ IPSet(['0.0.0.1/32']) == IPSet([])
    assert s3 ^ s4 == IPSet(['0.0.0.1/32', '10.0.0.65/32', '10.0.0.67/32', '255.255.255.1/32'])
    assert s4 ^ s3 == IPSet(['0.0.0.1/32', '10.0.0.65/32', '10.0.0.67/32', '255.255.255.1/32'])
    assert s3 ^ s4b == IPSet(
        ['0.0.0.1/32', '10.0.0.65/32', '10.0.0.67/32', '111.111.111.111/32', '255.255.255.1/32']
    )
    assert s3 ^ s5 == IPSet(['0.0.0.1/32', '10.0.0.64/32', '10.0.0.66/32', '255.255.255.1/32'])
    assert s5 ^ s3 == IPSet(['0.0.0.1/32', '10.0.0.64/32', '10.0.0.66/32', '255.255.255.1/32'])


def test_converting_ipsets_to_ipranges():
    assert list(IPSet().iter_ipranges()) == []
    assert list(IPSet([IPAddress('10.0.0.1')]).iter_ipranges()) == [IPRange('10.0.0.1', '10.0.0.1')]
    assert list(IPSet([IPAddress('10.0.0.1'), IPAddress('10.0.0.2')]).iter_ipranges()) == [
        IPRange('10.0.0.1', '10.0.0.2')
    ]


def test_len_on_ipset_failure_with_large_ipv6_addresses():
    s1 = IPSet(IPRange(IPAddress('::0'), IPAddress(sys.maxsize, 6)))
    with pytest.raises(IndexError):
        len(s1)

    s2 = IPSet(IPRange(IPAddress('::0'), IPAddress(sys.maxsize - 1, 6)))
    assert len(s2) == sys.maxsize


def test_ipset_ipv4_and_ipv4_separation():
    assert list(IPSet([IPAddress(1, 4), IPAddress(1, 6)]).iter_ipranges()) == [
        IPRange('0.0.0.1', '0.0.0.1'),
        IPRange('::1', '::1'),
    ]


def test_ipset_exceptions():
    s1 = IPSet(['10.0.0.1'])

    #   IPSet objects are not hashable.
    with pytest.raises(TypeError):
        hash(s1)

    #   Bad update argument type.
    with pytest.raises(TypeError):
        s1.update(42)


def test_ipset_comparison_with_int_is_invalid():
    s1 = IPSet(['10.0.0.1'])
    assert not s1 == 42
    s1 != 42


def test_ipset_converts_to_cidr_networks_v4():
    s1 = IPSet(IPNetwork('10.1.2.3/8'))
    s1.add(IPNetwork('192.168.1.2/16'))
    assert list(s1.iter_cidrs()) == [
        IPNetwork('10.0.0.0/8'),
        IPNetwork('192.168.0.0/16'),
    ]


def test_ipset_converts_to_cidr_networks_v6():
    s1 = IPSet(IPNetwork('fe80::4242/64'))
    s1.add(IPNetwork('fe90::4343/64'))
    assert list(s1.iter_cidrs()) == [
        IPNetwork('fe80::/64'),
        IPNetwork('fe90::/64'),
    ]


def test_ipset_is_weak_referencable():
    weakref.ref(IPSet())