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
|
# 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_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), [])
|
