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 import mock
  16 import unittest
  17 import copy
  18
  19 from yardstick.benchmark.scenarios.availability.result_checker import  result_checker_general
  20
  21
  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):
  27
  28     def setUp(self):
  29         host = {
  30             "ip": "10.20.0.5",
  31             "user": "root",
  32             "key_filename": "/root/.ssh/id_rsa"
  33         }
  34         self.context = {"node1": host}
  35         self.checker_cfg = {
  36             'parameter': {'processname': 'process'},
  37             'checker_type': 'general-result-checker',
  38             'condition' : 'eq',
  39             'expectedValue' : 1,
  40             'key' : 'process-checker',
  41             'host': 'node1'
  42         }
  43
  44     def test__result_checker_eq(self, mock_open, mock_ssh):
  45         ins = result_checker_general.GeneralResultChecker(self.checker_cfg,
  46             self.context);
  47         mock_ssh.SSH().execute.return_value = (0, "1", '')
  48         ins.setup()
  49         self.assertTrue(ins.verify())
  50
  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,
  55             self.context);
  56         mock_ssh.SSH().execute.return_value = (0, "2", '')
  57         ins.setup()
  58         self.assertTrue(ins.verify())
  59
  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,
  64             self.context);
  65         mock_ssh.SSH().execute.return_value = (0, "1", '')
  66         ins.setup()
  67         self.assertTrue(ins.verify())
  68
  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,
  73             self.context);
  74         mock_ssh.SSH().execute.return_value = (0, "0", '')
  75         ins.setup()
  76         self.assertTrue(ins.verify())
  77
  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,
  82             self.context);
  83         mock_ssh.SSH().execute.return_value = (0, "1", '')
  84         ins.setup()
  85         self.assertTrue(ins.verify())
  86
  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,
  92             self.context);
  93         mock_ssh.SSH().execute.return_value = (0, "value return", '')
  94         ins.setup()
  95         self.assertTrue(ins.verify())
  96
  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,
 101             self.context);
 102         mock_ssh.SSH().execute.return_value = (0, "1", '')
 103         ins.setup()
 104         self.assertFalse(ins.verify())
 105
 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,
 110             self.context);
 111         mock_ssh.SSH().execute.return_value = (1, "fail", '')
 112         ins.setup()
 113         ins.verify()