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
136
137
138
139
140
141
142
143
|
#!/usr/bin/env python3
#
# Copyright (C) 2019-2020 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import os
import unittest
import vyos.config
import vyos.configsession
from netifaces import ifaddresses, AF_INET, AF_INET6
from vyos.validate import is_intf_addr_assigned, is_ipv6_link_local
from vyos.interfaces import list_interfaces_of_type
class BasicInterfaceTest:
class BaseTest(unittest.TestCase):
def setUp(self):
self.session = vyos.configsession.ConfigSession(os.getpid())
env = self.session.get_session_env()
self.config = vyos.config.Config(session_env=env)
self._test_addr = ['192.0.2.1/25', '2001:db8:1::ffff/64']
def tearDown(self):
self.session.delete(self._base_path)
self.session.commit()
def test_add_description(self):
""" Check if description can be added to interface """
for intf in self._interfaces:
test_string='Description-Test-{}'.format(intf)
self.session.set(self._base_path + [intf, 'description', test_string])
self.session.commit()
# Validate interface description
for intf in self._interfaces:
test_string='Description-Test-{}'.format(intf)
with open('/sys/class/net/{}/ifalias'.format(intf), 'r') as f:
tmp = f.read().rstrip()
self.assertTrue(tmp, test_string)
def test_add_address(self):
""" Check if address can be added to interface """
# Add address
for intf in self._interfaces:
for addr in self._test_addr:
self.session.set(self._base_path + [intf, 'address', addr])
self.session.commit()
# Validate address
for intf in self._interfaces:
for af in AF_INET, AF_INET6:
for addr in ifaddresses(intf)[af]:
# checking link local addresses makes no sense
if is_ipv6_link_local(addr['addr']):
continue
self.assertTrue(is_intf_addr_assigned(intf, addr['addr']))
class DummyInterfaceTest(BasicInterfaceTest.BaseTest):
def setUp(self):
super().setUp()
self._base_path = ['interfaces', 'dummy']
self._interfaces = ['dum0', 'dum1', 'dum2']
class LoopbackInterfaceTest(BasicInterfaceTest.BaseTest):
def setUp(self):
super().setUp()
self._base_path = ['interfaces', 'loopback']
self._interfaces = ['lo']
class BondInterfaceTest(BasicInterfaceTest.BaseTest):
def setUp(self):
super().setUp()
self._base_path = ['interfaces', 'bonding']
self._interfaces = ['bond0']
def test_add_remove_member(self):
members = []
# we need to filter out VLAN interfaces identified by a dot (.)
# in their name - just in case!
for tmp in list_interfaces_of_type("ethernet"):
if not '.' in tmp:
members.append(tmp)
for intf in self._interfaces:
for member in members:
# We can not enslave an interface when there is an address
# assigned - take care here - or find them dynamically if a user
# runs vyos-smoketest on his production device?
self.session.set(self._base_path + [intf, 'member', 'interface', member])
self.session.commit()
for intf in self._interfaces:
self.session.delete(self._base_path + [intf, 'member'])
self.session.commit()
class BridgeInterfaceTest(BasicInterfaceTest.BaseTest):
def setUp(self):
super().setUp()
self._base_path = ['interfaces', 'bridge']
self._interfaces = ['br0']
def test_add_remove_member(self):
members = list_interfaces_of_type("ethernet")
for intf in self._interfaces:
cost = 1000
priority = 10
self.session.set(self._base_path + [intf, 'stp'])
for member in members:
self.session.set(self._base_path + [intf, 'member', 'interface', member])
self.session.set(self._base_path + [intf, 'member', 'interface', member, 'cost', str(cost)])
self.session.set(self._base_path + [intf, 'member', 'interface', member, 'priority', str(priority)])
cost += 1
priority += 1
self.session.commit()
for intf in self._interfaces:
self.session.delete(self._base_path + [intf, 'member'])
self.session.commit()
if __name__ == '__main__':
unittest.main()
|
