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
15 from __future__ import absolute_import
20 from yardstick.benchmark.scenarios.availability.result_checker import \
21 result_checker_general
24 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
25 'result_checker_general.ssh')
26 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
27 'result_checker_general.open')
28 class GeneralResultCheckerTestCase(unittest.TestCase):
34 "key_filename": "/root/.ssh/id_rsa"
36 self.context = {"node1": host}
38 'parameter': {'processname': 'process'},
39 'checker_type': 'general-result-checker',
42 'key': 'process-checker',
43 'checker_key': 'process-checker',
47 def test__result_checker_eq(self, mock_open, mock_ssh):
48 ins = result_checker_general.GeneralResultChecker(self.checker_cfg,
50 mock_ssh.SSH().execute.return_value = (0, "1", '')
52 self.assertTrue(ins.verify())
54 def test__result_checker_gt(self, mock_open, mock_ssh):
55 config = copy.deepcopy(self.checker_cfg)
56 config['condition'] = 'gt'
57 ins = result_checker_general.GeneralResultChecker(config,
59 mock_ssh.SSH().execute.return_value = (0, "2", '')
61 self.assertTrue(ins.verify())
63 def test__result_checker_gt_eq(self, mock_open, mock_ssh):
64 config = copy.deepcopy(self.checker_cfg)
65 config['condition'] = 'gt_eq'
66 ins = result_checker_general.GeneralResultChecker(config,
68 mock_ssh.SSH().execute.return_value = (0, "1", '')
70 self.assertTrue(ins.verify())
72 def test__result_checker_lt(self, mock_open, mock_ssh):
73 config = copy.deepcopy(self.checker_cfg)
74 config['condition'] = 'lt'
75 ins = result_checker_general.GeneralResultChecker(config,
77 mock_ssh.SSH().execute.return_value = (0, "0", '')
79 self.assertTrue(ins.verify())
81 def test__result_checker_lt_eq(self, mock_open, mock_ssh):
82 config = copy.deepcopy(self.checker_cfg)
83 config['condition'] = 'lt_eq'
84 ins = result_checker_general.GeneralResultChecker(config,
86 mock_ssh.SSH().execute.return_value = (0, "1", '')
88 self.assertTrue(ins.verify())
90 def test__result_checker_in(self, mock_open, mock_ssh):
91 config = copy.deepcopy(self.checker_cfg)
92 config['condition'] = 'in'
93 config['expectedValue'] = "value"
94 ins = result_checker_general.GeneralResultChecker(config,
96 mock_ssh.SSH().execute.return_value = (0, "value return", '')
98 self.assertTrue(ins.verify())
100 def test__result_checker_wrong(self, mock_open, mock_ssh):
101 config = copy.deepcopy(self.checker_cfg)
102 config['condition'] = 'wrong'
103 ins = result_checker_general.GeneralResultChecker(config,
105 mock_ssh.SSH().execute.return_value = (0, "1", '')
107 self.assertFalse(ins.verify())
109 def test__result_checker_fail(self, mock_open, mock_ssh):
110 config = copy.deepcopy(self.checker_cfg)
111 config.pop('parameter')
112 ins = result_checker_general.GeneralResultChecker(config,
114 mock_ssh.SSH().execute.return_value = (1, "fail", '')