3 ##############################################################################
4 # Copyright (c) 2016 Huan Li and others
5 # lihuansse@tongji.edu.cn
6 # All rights reserved. This program and the accompanying materials
7 # are made available under the terms of the Apache License, Version 2.0
8 # which accompanies this distribution, and is available at
9 # http://www.apache.org/licenses/LICENSE-2.0
10 ##############################################################################
12 # Unittest for yardstick.benchmark.scenarios.availability.result_checker
13 # .result_checker_general
19 from yardstick.benchmark.scenarios.availability.result_checker import result_checker_general
22 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
23 'result_checker_general.ssh')
24 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
25 'result_checker_general.open')
26 class GeneralResultCheckerTestCase(unittest.TestCase):
32 "key_filename": "/root/.ssh/id_rsa"
34 self.context = {"node1": host}
36 'parameter': {'processname': 'process'},
37 'checker_type': 'general-result-checker',
40 'key' : 'process-checker',
44 def test__result_checker_eq(self, mock_open, mock_ssh):
45 ins = result_checker_general.GeneralResultChecker(self.checker_cfg,
47 mock_ssh.SSH().execute.return_value = (0, "1", '')
49 self.assertTrue(ins.verify())
51 def test__result_checker_gt(self, mock_open, mock_ssh):
52 config = copy.deepcopy(self.checker_cfg)
53 config['condition'] = 'gt'
54 ins = result_checker_general.GeneralResultChecker(config,
56 mock_ssh.SSH().execute.return_value = (0, "2", '')
58 self.assertTrue(ins.verify())
60 def test__result_checker_gt_eq(self, mock_open, mock_ssh):
61 config = copy.deepcopy(self.checker_cfg)
62 config['condition'] = 'gt_eq'
63 ins = result_checker_general.GeneralResultChecker(config,
65 mock_ssh.SSH().execute.return_value = (0, "1", '')
67 self.assertTrue(ins.verify())
69 def test__result_checker_lt(self, mock_open, mock_ssh):
70 config = copy.deepcopy(self.checker_cfg)
71 config['condition'] = 'lt'
72 ins = result_checker_general.GeneralResultChecker(config,
74 mock_ssh.SSH().execute.return_value = (0, "0", '')
76 self.assertTrue(ins.verify())
78 def test__result_checker_lt_eq(self, mock_open, mock_ssh):
79 config = copy.deepcopy(self.checker_cfg)
80 config['condition'] = 'lt_eq'
81 ins = result_checker_general.GeneralResultChecker(config,
83 mock_ssh.SSH().execute.return_value = (0, "1", '')
85 self.assertTrue(ins.verify())
87 def test__result_checker_in(self, mock_open, mock_ssh):
88 config = copy.deepcopy(self.checker_cfg)
89 config['condition'] = 'in'
90 config['expectedValue'] = "value"
91 ins = result_checker_general.GeneralResultChecker(config,
93 mock_ssh.SSH().execute.return_value = (0, "value return", '')
95 self.assertTrue(ins.verify())
97 def test__result_checker_wrong(self, mock_open, mock_ssh):
98 config = copy.deepcopy(self.checker_cfg)
99 config['condition'] = 'wrong'
100 ins = result_checker_general.GeneralResultChecker(config,
102 mock_ssh.SSH().execute.return_value = (0, "1", '')
104 self.assertFalse(ins.verify())
106 def test__result_checker_fail(self, mock_open, mock_ssh):
107 config = copy.deepcopy(self.checker_cfg)
108 config.pop('parameter')
109 ins = result_checker_general.GeneralResultChecker(config,
111 mock_ssh.SSH().execute.return_value = (1, "fail", '')