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
|
# Copyright VyOS maintainers and contributors <maintainers@vyos.io>
#
# 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/>.
from unittest import TestCase
from unittest.mock import patch
class TestVyOSUtils(TestCase):
def test_cmdl_basic(self):
from vyos.utils.process import cmdl
out = cmdl(['echo', 'hello'])
self.assertEqual(out, 'hello')
def test_cmdl_requires_list(self):
from vyos.utils.process import cmdl
with self.assertRaises(TypeError):
cmdl('echo hello')
def test_cmdl_sudo_prepends(self):
from vyos.utils.process import cmdl
with patch('vyos.utils.process.popen') as mock_popen:
mock_popen.return_value = ('', 0)
cmdl(['id'], sudo=True)
args, kwargs = mock_popen.call_args
self.assertEqual(args[0][0], 'sudo')
self.assertEqual(args[0][1], 'id')
def test_cmdl_no_shell(self):
from vyos.utils.process import cmdl
with patch('vyos.utils.process.popen') as mock_popen:
mock_popen.return_value = ('', 0)
cmdl(['true'])
_, kwargs = mock_popen.call_args
self.assertEqual(kwargs.get('shell'), False)
def test_cmdl_raises_on_nonzero(self):
from vyos.utils.process import cmdl
with self.assertRaises(OSError):
cmdl(['false'])
def test_key_mangling(self):
from vyos.utils.dict import mangle_dict_keys
data = {"foo-bar": {"baz-quux": None}}
expected_data = {"foo_bar": {"baz_quux": None}}
new_data = mangle_dict_keys(data, '-', '_')
self.assertEqual(new_data, expected_data)
def test_sysctl_read(self):
from vyos.utils.system import sysctl_read
self.assertEqual(sysctl_read(['net', 'ipv4', 'conf', 'lo', 'forwarding']), '1')
def test_sysctl_key_normalization(self):
from vyos.utils.system import sysctl_read
with patch('vyos.utils.system.run') as mock_run:
mock_run.return_value.stdout = b'1\n'
sysctl_read(['net', 'ipv4', 'conf', 'eth0.10', 'forwarding'])
mock_run.assert_called_with(
['sysctl', '-nb', 'net.ipv4.conf.eth0/10.forwarding'],
capture_output=True,
)
def test_list_strip(self):
from vyos.utils.list import list_strip
lst = ['a', 'b', 'c', 'd', 'e']
sub = ['a', 'b']
rsb = ['d', 'e']
non = ['a', 'e']
self.assertEqual(list_strip(lst, sub), ['c', 'd', 'e'])
self.assertEqual(list_strip(lst, rsb, right=True), ['a', 'b', 'c'])
self.assertEqual(list_strip(lst, non), [])
self.assertEqual(list_strip(sub, lst), [])
def test_range_str_to_list(self):
from vyos.utils.convert import range_str_to_list
# basic cases
self.assertEqual(range_str_to_list('1-3'), [1, 2, 3])
self.assertEqual(range_str_to_list('1-3,5,7-8'), [1, 2, 3, 5, 7, 8])
self.assertEqual(range_str_to_list('3'), [3])
# empty string
self.assertEqual(range_str_to_list(''), [])
# unordered input
self.assertEqual(range_str_to_list('5,1-3,4'), [1, 2, 3, 4, 5])
self.assertEqual(range_str_to_list('7-9,1-3'), [1, 2, 3, 7, 8, 9])
# overlapping ranges
self.assertEqual(range_str_to_list('1-5,3-7'), [1, 2, 3, 4, 5, 6, 7])
self.assertEqual(range_str_to_list('1-3,2-4,3-5'), [1, 2, 3, 4, 5])
# duplicated values
self.assertEqual(range_str_to_list('1,1,2,2,3'), [1, 2, 3])
self.assertEqual(range_str_to_list('5,1-3,2,3'), [1, 2, 3, 5])
# adjacent ranges
self.assertEqual(range_str_to_list('1-3,4-6'), [1, 2, 3, 4, 5, 6])
def test_list_to_range_str(self):
from vyos.utils.convert import list_to_range_str
# basic cases
self.assertEqual(list_to_range_str([1, 2, 3]), '1-3')
self.assertEqual(list_to_range_str([1, 2, 3, 5, 7, 8]), '1-3,5,7-8')
self.assertEqual(list_to_range_str([1, 3]), '1,3')
self.assertEqual(list_to_range_str([3]), '3')
# empty list
self.assertEqual(list_to_range_str([]), '')
# unordered input
self.assertEqual(list_to_range_str([5, 1, 2, 3, 4]), '1-5')
self.assertEqual(list_to_range_str([7, 8, 9, 1, 2, 3]), '1-3,7-9')
# duplicated values
self.assertEqual(list_to_range_str([1, 1, 2, 2, 3, 3]), '1-3')
self.assertEqual(list_to_range_str([5, 1, 2, 2, 3, 5]), '1-3,5')
|
