1 ##############################################################################
2 # Copyright (c) 2016 Dan Radez (Red Hat)
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
12 from apex.ip_utils import IPUtilsException
13 from apex.ip_utils import get_interface
14 from apex.ip_utils import find_gateway
15 from apex.ip_utils import get_ip
16 from apex.ip_utils import get_ip_range
18 from nose.tools import assert_equal
19 from nose.tools import assert_raises
20 from nose.tools import assert_is_instance
21 from nose.tools import assert_regexp_matches
23 from ipaddress import IPv4Address
24 from ipaddress import IPv6Address
25 from ipaddress import ip_network
28 ip4_pattern = re.compile('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}')
29 ip4_range_pattern = re.compile('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3},\d{1,'
30 '3}\.\d{1,3}\.\d{1,3}\.\d{1,3}')
33 def get_default_gateway_linux():
34 """Read the default gateway directly from /proc."""
35 with open("/proc/net/route") as fh:
37 fields = line.strip().split()
38 if fields[2] not in ('00000000', 'Gateway'):
42 class TestIpUtils(object):
44 def setup_class(klass):
45 """This method is run once for each class before any tests are run"""
46 klass.iface_name = get_default_gateway_linux()
47 iface = get_interface(klass.iface_name)
51 def teardown_class(klass):
52 """This method is run once for each class _after_ all tests are run"""
55 """This method is run once before _each_ test method is executed"""
58 """This method is run once after _each_ test method is executed"""
60 def test_get_interface(self):
61 assert_equal(get_interface(''), None)
62 assert_equal(get_interface('notreal'), None)
63 assert_is_instance(get_interface(
65 address_family=4), IPv4Address)
66 # assert_is_instance(get_interface(
68 # address_family=6), IPv6Address)
69 assert_raises(IPUtilsException,
70 get_interface, self.iface_name, 0)
72 def test_find_gateway(self):
73 assert_is_instance(find_gateway(self.iface), str)
74 iface_virbr0 = get_interface('virbr0')
75 assert_equal(find_gateway(iface_virbr0), None)
77 def test_get_ip(self):
78 assert_equal(get_ip(1, cidr="10.10.10.0/24"), "0")
79 assert_regexp_matches(get_ip(1, interface=self.iface), ip4_pattern)
80 assert_raises(IPUtilsException, get_ip, 1)
82 def test_get_ip_range_raises(self):
83 assert_raises(IPUtilsException, get_ip_range)
84 assert_raises(IPUtilsException, get_ip_range, interface=self.iface)
86 def test_get_ip_range_with_interface(self):
87 assert_regexp_matches(get_ip_range(interface=self.iface,
88 start_offset=1, end_offset=20),
90 assert_regexp_matches(get_ip_range(interface=self.iface,
91 start_offset=1, count=10),
93 assert_regexp_matches(get_ip_range(interface=self.iface, end_offset=20,
94 count=10), ip4_range_pattern)
97 def test_get_ip_range_with_cidr():
98 cidr = ip_network('10.10.10.0/24')
99 assert_raises(IPUtilsException, get_ip_range, cidr=cidr)
100 assert_regexp_matches(get_ip_range(cidr=cidr, start_offset=1,
101 end_offset=20), ip4_pattern)
102 assert_regexp_matches(get_ip_range(cidr=cidr, start_offset=1,
103 count=10), ip4_pattern)
104 assert_regexp_matches(get_ip_range(cidr=cidr, end_offset=20,
105 count=10), ip4_pattern)