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 @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):
  29
  30     def setUp(self):
  31         host = {
  32             "ip": "10.20.0.5",
  33             "user": "root",
  34             "key_filename": "/root/.ssh/id_rsa"
  35         }
  36         self.context = {"node1": host}
  37         self.checker_cfg = {
  38             'parameter': {'processname': 'process'},
  39             'checker_type': 'general-result-checker',
  40             'condition': 'eq',
  41             'expectedValue': 1,
  42             'key': 'process-checker',
  43             'checker_key': 'process-checker',
  44             'host': 'node1'
  45         }
  46
  47     def test__result_checker_eq(self, mock_open, mock_ssh):
  48         ins = result_checker_general.GeneralResultChecker(self.checker_cfg,
  49                                                           self.context)
  50         mock_ssh.SSH().execute.return_value = (0, "1", '')
  51         ins.setup()
  52         self.assertTrue(ins.verify())
  53
  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,
  58                                                           self.context)
  59         mock_ssh.SSH().execute.return_value = (0, "2", '')
  60         ins.setup()
  61         self.assertTrue(ins.verify())
  62
  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,
  67                                                           self.context)
  68         mock_ssh.SSH().execute.return_value = (0, "1", '')
  69         ins.setup()
  70         self.assertTrue(ins.verify())
  71
  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,
  76                                                           self.context)
  77         mock_ssh.SSH().execute.return_value = (0, "0", '')
  78         ins.setup()
  79         self.assertTrue(ins.verify())
  80
  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,
  85                                                           self.context)
  86         mock_ssh.SSH().execute.return_value = (0, "1", '')
  87         ins.setup()
  88         self.assertTrue(ins.verify())
  89
  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,
  95                                                           self.context)
  96         mock_ssh.SSH().execute.return_value = (0, "value return", '')
  97         ins.setup()
  98         self.assertTrue(ins.verify())
  99
 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,
 104                                                           self.context)
 105         mock_ssh.SSH().execute.return_value = (0, "1", '')
 106         ins.setup()
 107         self.assertFalse(ins.verify())
 108
 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,
 113                                                           self.context)
 114         mock_ssh.SSH().execute.return_value = (1, "fail", '')
 115         ins.setup()
 116         ins.verify()