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
|
# -*- coding: utf-8 -*-
from __future__ import absolute_import, division, print_function
__metaclass__ = type
import unittest
from unittest.mock import MagicMock
from ansible_collections.vyos.rest.plugins.modules.vyos_command import (
evaluate_conditions,
parse_command,
run_commands,
)
class TestVyOSCommandParseCommand(unittest.TestCase):
def test_string_single_word(self):
self.assertEqual(parse_command("version"), ["version"])
def test_string_multi_word(self):
self.assertEqual(parse_command("ip route"), ["ip", "route"])
def test_list_passthrough(self):
self.assertEqual(parse_command(["ip", "route"]), ["ip", "route"])
class TestVyOSCommandEvaluateConditions(unittest.TestCase):
def _stdout(self):
return ["VyOS 1.5.0 output", "eth0 192.168.1.1"]
def test_contains_match(self):
failed, conds = evaluate_conditions(
self._stdout(),
["result[0] contains VyOS"],
"all",
)
self.assertFalse(failed)
self.assertEqual(conds, [])
def test_contains_no_match(self):
failed, conds = evaluate_conditions(
self._stdout(),
["result[0] contains NonExistent"],
"all",
)
self.assertTrue(failed)
self.assertIn("result[0] contains NonExistent", conds)
def test_match_all_both_pass(self):
failed, conditions = evaluate_conditions(
self._stdout(),
["result[0] contains VyOS", "result[1] contains eth0"],
"all",
)
self.assertFalse(failed)
def test_match_all_one_fails(self):
failed, conditions = evaluate_conditions(
self._stdout(),
["result[0] contains VyOS", "result[1] contains NonExistent"],
"all",
)
self.assertTrue(failed)
def test_match_any_one_passes(self):
failed, conditions = evaluate_conditions(
self._stdout(),
["result[0] contains VyOS", "result[1] contains NonExistent"],
"any",
)
self.assertFalse(failed)
def test_empty_conditions(self):
failed, conds = evaluate_conditions(self._stdout(), [], "all")
self.assertFalse(failed)
self.assertEqual(conds, [])
class TestVyOSCommandRunCommands(unittest.TestCase):
def setUp(self):
self.mock_vyos = MagicMock()
def test_run_list_command(self):
self.mock_vyos.show = MagicMock(return_value="VyOS 1.5.0")
result = run_commands(self.mock_vyos, [["version"]])
self.mock_vyos.show.assert_called_once_with(["version"])
self.assertEqual(result, ["VyOS 1.5.0"])
def test_run_string_command(self):
self.mock_vyos.show = MagicMock(return_value="uptime")
run_commands(self.mock_vyos, ["system uptime"])
self.mock_vyos.show.assert_called_once_with(["system", "uptime"])
def test_run_multiple_commands(self):
self.mock_vyos.show = MagicMock(side_effect=["out1", "out2"])
result = run_commands(self.mock_vyos, [["version"], ["interfaces"]])
self.assertEqual(result, ["out1", "out2"])
def test_run_none_returns_empty_string(self):
self.mock_vyos.show = MagicMock(return_value=None)
result = run_commands(self.mock_vyos, [["version"]])
self.assertEqual(result, [""])
if __name__ == "__main__":
unittest.main()
|
