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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
##############################################################################
# Copyright (c) 2016 Dan Radez (Red Hat)
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
import re
import ipaddress
from apex.ip_utils import IPUtilsException
from apex.ip_utils import get_interface
from apex.ip_utils import find_gateway
from apex.ip_utils import get_ip
from apex.ip_utils import get_ip_range
from apex.ip_utils import _validate_ip_range
from nose.tools import assert_true
from nose.tools import assert_false
from nose.tools import assert_equal
from nose.tools import assert_raises
from nose.tools import assert_is_instance
from nose.tools import assert_regexp_matches
from ipaddress import IPv4Address
from ipaddress import IPv6Address
from ipaddress import ip_network
ip4_pattern = re.compile('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}')
ip4_range_pattern = re.compile('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3},\d{1,'
'3}\.\d{1,3}\.\d{1,3}\.\d{1,3}')
def get_default_gateway_linux():
"""Read the default gateway directly from /proc."""
with open("/proc/net/route") as fh:
for line in fh:
fields = line.strip().split()
if fields[2] not in ('00000000', 'Gateway'):
return fields[0]
class TestIpUtils(object):
@classmethod
def setup_class(klass):
"""This method is run once for each class before any tests are run"""
klass.iface_name = get_default_gateway_linux()
iface = get_interface(klass.iface_name)
klass.iface = iface
@classmethod
def teardown_class(klass):
"""This method is run once for each class _after_ all tests are run"""
def setUp(self):
"""This method is run once before _each_ test method is executed"""
def teardown(self):
"""This method is run once after _each_ test method is executed"""
def test_get_interface(self):
assert_equal(get_interface(''), None)
assert_equal(get_interface('notreal'), None)
assert_is_instance(get_interface(self.iface_name,
address_family=4),
IPv4Address)
# can't enable this until there's a v6 address on the ci hosts
# assert_is_instance(get_interface(
# self.iface_name,
# address_family=6), IPv6Address)
assert_raises(IPUtilsException,
get_interface, self.iface_name, 0)
def test_find_gateway(self):
assert_is_instance(find_gateway(self.iface), str)
iface_virbr0 = get_interface('virbr0')
assert_equal(find_gateway(iface_virbr0), None)
def test_get_ip(self):
cidr = ipaddress.ip_network("10.10.10.0/24")
assert_equal(get_ip(1, cidr=cidr), "10.10.10.1")
assert_raises(IPUtilsException, get_ip, 1000, interface=self.iface)
assert_regexp_matches(get_ip(1, interface=self.iface), ip4_pattern)
assert_raises(IPUtilsException, get_ip, 1)
def test_get_ip_range_raises(self):
assert_raises(IPUtilsException, get_ip_range)
assert_raises(IPUtilsException, get_ip_range, interface=self.iface)
def test_get_ip_range_with_interface(self):
assert_regexp_matches(get_ip_range(interface=self.iface,
start_offset=1, end_offset=20),
ip4_range_pattern)
assert_regexp_matches(get_ip_range(interface=self.iface,
start_offset=1, count=10),
ip4_range_pattern)
assert_regexp_matches(get_ip_range(interface=self.iface, end_offset=20,
count=10), ip4_range_pattern)
def test_get_ip_range_with_cidr(self):
cidr = ip_network('10.10.10.0/24')
assert_raises(IPUtilsException, get_ip_range, cidr=cidr)
assert_regexp_matches(get_ip_range(cidr=cidr, start_offset=1,
end_offset=20), ip4_pattern)
assert_regexp_matches(get_ip_range(cidr=cidr, start_offset=1,
count=10), ip4_pattern)
assert_regexp_matches(get_ip_range(cidr=cidr, end_offset=20,
count=10), ip4_pattern)
def test__validate_ip_range(self):
cidr = ip_network('10.10.10.0/24')
assert_true(_validate_ip_range(
start_ip=ipaddress.IPv4Address('10.10.10.1'),
end_ip=ipaddress.IPv4Address('10.10.10.10'),
cidr=cidr))
assert_false(_validate_ip_range(
start_ip=ipaddress.IPv4Address('10.10.10.10'),
end_ip=ipaddress.IPv4Address('10.10.10.1'),
cidr=cidr))
assert_false(_validate_ip_range(
start_ip=ipaddress.IPv4Address('10.10.0.1'),
end_ip=ipaddress.IPv4Address('10.10.10.10'),
cidr=cidr))
assert_false(_validate_ip_range(
start_ip=ipaddress.IPv4Address('10.10.10.1'),
end_ip=ipaddress.IPv4Address('10.10.11.10'),
cidr=cidr))
def test_exception(self):
e = IPUtilsException("test")
print(e)
assert_is_instance(e, IPUtilsException)
|
