yardstick/tests/unit/benchmark/scenarios/availability/test_result_checker_general.py

   1 #!/usr/bin/env python
   2
   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 ##############################################################################
  11
  12 # Unittest for yardstick.benchmark.scenarios.availability.result_checker
  13 # .result_checker_general
  14
  15 from __future__ import absolute_import
  16 import mock
  17 import unittest
  18 import copy
  19
  20 from yardstick.benchmark.scenarios.availability.result_checker import \
  21     result_checker_general
  22
  23
  24 # pylint: disable=unused-argument
  25 # disable this for now because I keep forgetting mock patch arg ordering
  26
  27
  28 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
  29             'result_checker_general.ssh')
  30 @mock.patch('yardstick.benchmark.scenarios.availability.result_checker.'
  31             'result_checker_general.open')
  32 class GeneralResultCheckerTestCase(unittest.TestCase):
  33
  34     def setUp(self):
  35         host = {
  36             "ip": "10.20.0.5",
  37             "user": "root",
  38             "key_filename": "/root/.ssh/id_rsa"
  39         }
  40         self.context = {"node1": host}
  41         self.checker_cfg = {
  42             'parameter': {'processname': 'process'},
  43             'checker_type': 'general-result-checker',
  44             'condition': 'eq',
  45             'expectedValue': 1,
  46             'key': 'process-checker',
  47             'checker_key': 'process-checker',
  48             'host': 'node1'
  49         }
  50
  51     def test__result_checker_eq(self, mock_open, mock_ssh):
  52         ins = result_checker_general.GeneralResultChecker(self.checker_cfg,
  53                                                           self.context)
  54         mock_ssh.SSH.from_node().execute.return_value = (0, "1", '')
  55         ins.setup()
  56         self.assertTrue(ins.verify())
  57
  58     def test__result_checker_gt(self, mock_open, mock_ssh):
  59         config = copy.deepcopy(self.checker_cfg)
  60         config['condition'] = 'gt'
  61         ins = result_checker_general.GeneralResultChecker(config,
  62                                                           self.context)
  63         mock_ssh.SSH.from_node().execute.return_value = (0, "2", '')
  64         ins.setup()
  65         self.assertTrue(ins.verify())
  66
  67     def test__result_checker_gt_eq(self, mock_open, mock_ssh):
  68         config = copy.deepcopy(self.checker_cfg)
  69         config['condition'] = 'gt_eq'
  70         ins = result_checker_general.GeneralResultChecker(config,
  71                                                           self.context)
  72         mock_ssh.SSH.from_node().execute.return_value = (0, "1", '')
  73         ins.setup()
  74         self.assertTrue(ins.verify())
  75
  76     def test__result_checker_lt(self, mock_open, mock_ssh):
  77         config = copy.deepcopy(self.checker_cfg)
  78         config['condition'] = 'lt'
  79         ins = result_checker_general.GeneralResultChecker(config,
  80                                                           self.context)
  81         mock_ssh.SSH.from_node().execute.return_value = (0, "0", '')
  82         ins.setup()
  83         self.assertTrue(ins.verify())
  84
  85     def test__result_checker_lt_eq(self, mock_open, mock_ssh):
  86         config = copy.deepcopy(self.checker_cfg)
  87         config['condition'] = 'lt_eq'
  88         ins = result_checker_general.GeneralResultChecker(config,
  89                                                           self.context)
  90         mock_ssh.SSH.from_node().execute.return_value = (0, "1", '')
  91         ins.setup()
  92         self.assertTrue(ins.verify())
  93
  94     def test__result_checker_in(self, mock_open, mock_ssh):
  95         config = copy.deepcopy(self.checker_cfg)
  96         config['condition'] = 'in'
  97         config['expectedValue'] = "value"
  98         ins = result_checker_general.GeneralResultChecker(config,
  99                                                           self.context)
 100         mock_ssh.SSH.from_node().execute.return_value = (0, "value return", '')
 101         ins.setup()
 102         self.assertTrue(ins.verify())
 103
 104     def test__result_checker_wrong(self, mock_open, mock_ssh):
 105         config = copy.deepcopy(self.checker_cfg)
 106         config['condition'] = 'wrong'
 107         ins = result_checker_general.GeneralResultChecker(config,
 108                                                           self.context)
 109         mock_ssh.SSH.from_node().execute.return_value = (0, "1", '')
 110         ins.setup()
 111         self.assertFalse(ins.verify())
 112
 113     def test__result_checker_fail(self, mock_open, mock_ssh):
 114         config = copy.deepcopy(self.checker_cfg)
 115         config.pop('parameter')
 116         ins = result_checker_general.GeneralResultChecker(config,
 117                                                           self.context)
 118         mock_ssh.SSH.from_node().execute.return_value = (1, "fail", '')
 119         ins.setup()
 120         ins.verify()