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
|
##############################################################################
# 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
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 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)
# 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):
assert_equal(get_ip(1, cidr="10.10.10.0/24"), "0")
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)
|
