Commit | Line | Data |
---|---|---|
f49f6323 PD |
1 | #!/usr/bin/env python |
2 | ||
11cbb815 PD |
3 | # The software in this package is distributed under the GNU General |
4 | # Public License version 2 (with a special exception described below). | |
5 | # | |
6 | # A copy of GNU General Public License (GPL) is included in this distribution, | |
7 | # in the file COPYING.GPL. | |
8 | # | |
9 | # As a special exception, if other files instantiate templates or use macros | |
10 | # or inline functions from this file, or you compile this file and link it | |
11 | # with other works to produce a work based on this file, this file | |
12 | # does not by itself cause the resulting work to be covered | |
13 | # by the GNU General Public License. | |
14 | # | |
15 | # However the source code for this file must still be made available | |
16 | # in accordance with section (3) of the GNU General Public License. | |
17 | # | |
18 | # This exception does not invalidate any other reasons why a work based | |
19 | # on this file might be covered by the GNU General Public License. | |
20 | # | |
21 | # Copyright (c) 2016-2018 Intra2net AG <info@intra2net.com> | |
22 | ||
f49f6323 PD |
23 | import unittest |
24 | import unittest.mock as mock | |
25 | import subprocess | |
26 | ||
2f3212fe | 27 | from src import arnied_wrapper |
f49f6323 PD |
28 | |
29 | ||
30 | class DummyCmdOutputMapping(object): | |
31 | ||
32 | fail_switch = False | |
33 | cmds = [ | |
34 | {"cmd": "pgrep -l -x arnied", "stdout": b"", "returncode": 0}, | |
80d82f13 | 35 | |
f49f6323 PD |
36 | {"cmd": 'echo "LICENSE_ACCEPTED,0: \\"1\\"" | set_cnf', "stdout": b"", "returncode": 0}, |
37 | {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-program-end GENERATE', "stdout": b"", "returncode": 0}, | |
80d82f13 | 38 | |
d6d4904f | 39 | {"cmd": 'get_cnf PROVIDER 1', "stdout": b"1 PROVIDER,1: \"sample-provider\"", "returncode": 0}, |
f49f6323 | 40 | {"cmd": 'tell-connd --online P1', "stdout": b"", "returncode": 0}, |
d6d4904f | 41 | {"cmd": '/usr/intranator/bin/get_var ONLINE', "stdout": b"DEFAULT: 2", "returncode": 0}, |
80d82f13 | 42 | |
d9446155 | 43 | {"cmd": "pgrep -l -x arnied", "stdout": b"", "returncode": 0}, |
f49f6323 PD |
44 | {"cmd": 'get_cnf VIRSCAN_UPDATE_CRON | set_cnf -x', "stdout": b"", "returncode": 0}, |
45 | {"cmd": '/usr/intranator/bin/arnied_helper --is-scheduled-or-running GENERATE', "stdout": b"", "returncode": 1}, | |
80d82f13 | 46 | {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-program-end GENERATE --wait-for-program-timeout 300', "stdout": b"", "returncode": 1}, |
f49f6323 | 47 | {"cmd": '/usr/intranator/bin/arnied_helper --is-scheduled-or-running GENERATE_OFFLINE', "stdout": b"", "returncode": 1}, |
80d82f13 | 48 | {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-program-end GENERATE_OFFLINE --wait-for-program-timeout 300', "stdout": b"", "returncode": 1}, |
f49f6323 PD |
49 | {"cmd": 'echo \'VIRSCAN_UPDATE_DNS_PUSH,0:"0"\' |set_cnf', "stdout": b"", "returncode": 0}, |
50 | {"cmd": 'rm -f /var/intranator/schedule/UPDATE_VIRSCAN_NODIAL*', "stdout": b"", "returncode": 0}, | |
80d82f13 | 51 | |
f49f6323 PD |
52 | {"cmd": '/usr/intranator/bin/arnied_helper --transfer-mail', "stdout": b"", "returncode": 0}, |
53 | ] | |
54 | asserted_cmds = [] | |
55 | ||
80d82f13 | 56 | def __init__(self, cmd="", ignore_errors=False, vm=None, timeout=60): |
f49f6323 | 57 | self.returncode = 0 |
7102665a | 58 | self.stdout = b"" |
f49f6323 PD |
59 | self._get_result(cmd) |
60 | if self.fail_switch: | |
61 | self.returncode = 1 | |
62 | ||
63 | def __str__(self): | |
64 | return "status %i, stdout %s" % (self.returncode, self.stdout) | |
65 | ||
66 | def _get_result(self, cmd): | |
67 | for dummy_cmd in self.asserted_cmds: | |
68 | if dummy_cmd['cmd'] == cmd: | |
69 | self.returncode = dummy_cmd['returncode'] | |
70 | self.stdout = dummy_cmd['stdout'] | |
71 | return | |
72 | raise ValueError("Could not locate the command '%s' among the known answers " | |
73 | "for the universe" % cmd) | |
74 | ||
75 | ||
7102665a | 76 | @mock.patch('src.arnied_wrapper.run_cmd', DummyCmdOutputMapping) |
f49f6323 PD |
77 | class ArniedWrapperTest(unittest.TestCase): |
78 | ||
79 | def setUp(self): | |
80 | DummyCmdOutputMapping.fail_switch = False | |
81 | DummyCmdOutputMapping.asserted_cmds = [] | |
82 | self.cmd_db = DummyCmdOutputMapping.cmds | |
83 | ||
84 | def test_verify_running(self): | |
85 | DummyCmdOutputMapping.asserted_cmds = self.cmd_db[0:1] | |
86 | arnied_wrapper.verify_running(timeout=1) | |
87 | DummyCmdOutputMapping.fail_switch = True | |
88 | with self.assertRaises(RuntimeError): | |
89 | arnied_wrapper.verify_running(timeout=1) | |
90 | ||
91 | def test_accept_license(self): | |
92 | DummyCmdOutputMapping.asserted_cmds = self.cmd_db[1:3] | |
93 | arnied_wrapper.accept_licence() | |
94 | # make sure an error is ignored since license might | |
95 | # already be accepted | |
96 | DummyCmdOutputMapping.fail_switch = True | |
97 | arnied_wrapper.accept_licence() | |
98 | ||
99 | def test_go_online(self): | |
d6d4904f | 100 | DummyCmdOutputMapping.asserted_cmds = self.cmd_db[3:6] |
f49f6323 | 101 | arnied_wrapper.go_online(1) |
f49f6323 PD |
102 | |
103 | def test_disable_virscan(self): | |
d6d4904f | 104 | DummyCmdOutputMapping.asserted_cmds = self.cmd_db[6:14] |
f49f6323 PD |
105 | arnied_wrapper.disable_virscan() |
106 | ||
107 | def test_email_transfer(self): | |
d6d4904f | 108 | DummyCmdOutputMapping.asserted_cmds = self.cmd_db[14:15] |
f49f6323 | 109 | arnied_wrapper.email_transfer() |