aboutsummaryrefslogtreecommitdiff
path: root/tests/test_ip.py
blob: dc081a5a3a46dafa04ea0291668d7ce4bd5ea6ed (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
from edify.library import ipv4
from edify.library import ipv6

# Generate ipv4 dictionary
ipv4_dict = {
    "192.168.0.1": True,
    "244.232.123.233": True,
    "363.232.123.233": False,
    "234.234234.234.234": False,
    "12.12.12.12.12": False,
    "0.0.0.0": True,
    "987.987.987.987": False,
}

# Generate ipv6 dictionary
ipv6_dict = {
    "2001:0db8:85a3:0000:0000:8a2e:0370:7334": True,
    "2001:db8:85a3:0:0:8a2e:370:7334": True,
    "2001:db8:85a3::8a2e:370:7334": True,
    "2001:db8:85a3:0:0:8A2E:370:7334": True,
    "2001:db8:85a3:0:0:8a2e:370:7334:": False,
    "2001:db8:85a3:0:0:8a2e:370:7334:7334": False,
    "2001:db8:85a3:0:0:8a2e:370:7334:7334:7334": False,
    "2001:db8:85a3:0:0:8a2e:370:7334:7334:7334:7334": False,
}


def test_ipv4():
    for ip, expectation in ipv4_dict.items():
        assert ipv4(ip) == expectation


def test_ipv6():
    for ip, expectation in ipv6_dict.items():
        assert ipv6(ip) == expectation