3 # The software in this package is distributed under the GNU General
4 # Public License version 2 (with a special exception described below).
6 # A copy of GNU General Public License (GPL) is included in this distribution,
7 # in the file COPYING.GPL.
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.
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.
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.
21 # Copyright (c) 2016-2018 Intra2net AG <info@intra2net.com>
24 import unittest.mock as mock
27 from src import arnied_wrapper
30 class DummyCmdOutputMapping(object):
34 {"cmd": "pgrep -l -x arnied", "stdout": b"", "returncode": 0},
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},
39 {"cmd": 'get_cnf PROVIDER 1', "stdout": b"1 PROVIDER,1: \"sample-provider\"", "returncode": 0},
40 {"cmd": 'tell-connd --online P1', "stdout": b"", "returncode": 0},
41 {"cmd": '/usr/intranator/bin/get_var ONLINE', "stdout": b"DEFAULT: 2", "returncode": 0},
43 {"cmd": "pgrep -l -x arnied", "stdout": b"", "returncode": 0},
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},
46 {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-program-end GENERATE --wait-for-program-timeout 300', "stdout": b"", "returncode": 1},
47 {"cmd": '/usr/intranator/bin/arnied_helper --is-scheduled-or-running GENERATE_OFFLINE', "stdout": b"", "returncode": 1},
48 {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-program-end GENERATE_OFFLINE --wait-for-program-timeout 300', "stdout": b"", "returncode": 1},
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},
52 {"cmd": '/usr/intranator/bin/arnied_helper --transfer-mail', "stdout": b"", "returncode": 0},
54 {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-arnied-socket --wait-for-arnied-socket-timeout 10', "stdout": b"", "returncode": 0},
55 {"cmd": '/usr/intranator/bin/arnied_helper --wait-for-arnied-socket --wait-for-arnied-socket-timeout 30', "stdout": b"", "returncode": 0},
59 def __init__(self, cmd="", ignore_errors=False, vm=None, timeout=60):
67 return "status %i, stdout %s" % (self.returncode, self.stdout)
69 def _get_result(self, cmd):
70 for dummy_cmd in self.asserted_cmds:
71 if dummy_cmd['cmd'] == cmd:
72 self.returncode = dummy_cmd['returncode']
73 self.stdout = dummy_cmd['stdout']
75 raise ValueError("Could not locate the command '%s' among the known answers "
76 "for the universe" % cmd)
79 @mock.patch('src.arnied_wrapper.run_cmd', DummyCmdOutputMapping)
80 class ArniedWrapperTest(unittest.TestCase):
83 DummyCmdOutputMapping.fail_switch = False
84 DummyCmdOutputMapping.asserted_cmds = []
85 self.cmd_db = DummyCmdOutputMapping.cmds
87 def test_verify_running(self):
88 DummyCmdOutputMapping.asserted_cmds = self.cmd_db[0:1]
89 arnied_wrapper.verify_running(timeout=1)
90 DummyCmdOutputMapping.fail_switch = True
91 with self.assertRaises(RuntimeError):
92 arnied_wrapper.verify_running(timeout=1)
94 def test_wait_for_arnied(self):
95 DummyCmdOutputMapping.asserted_cmds = self.cmd_db
96 arnied_wrapper.wait_for_arnied(timeout=10)
98 def test_accept_license(self):
99 DummyCmdOutputMapping.asserted_cmds = self.cmd_db[1:3]
100 arnied_wrapper.accept_licence()
101 # make sure an error is ignored since license might
102 # already be accepted
103 DummyCmdOutputMapping.fail_switch = True
104 arnied_wrapper.accept_licence()
106 def test_go_online(self):
107 DummyCmdOutputMapping.asserted_cmds = self.cmd_db[3:6]
108 arnied_wrapper.go_online(1)
110 def test_disable_virscan(self):
111 DummyCmdOutputMapping.asserted_cmds = self.cmd_db[6:]
112 arnied_wrapper.disable_virscan()
114 def test_email_transfer(self):
115 DummyCmdOutputMapping.asserted_cmds = self.cmd_db[14:15]
116 arnied_wrapper.email_transfer()