From ba4e9e6e47bd10ecc803bab920178ea973c2fa86 Mon Sep 17 00:00:00 2001
From: Emma Foley <emma.l.foley@intel.com>
Date: Wed, 29 Nov 2017 22:56:21 +0000
Subject: Move tests: unit/benchmark

* Fix pylint errors
* Add TODOs

Some errors are ignored locally, as they were a symptom of other problems.
These issues have been flagged with a TODO, and should be fixed later.

Change-Id: I30eb4b0aafe0575d0cddbc946108291f21a98ed8
Jira: YARDSTICK-837
Signed-off-by: Emma Foley <emma.l.foley@intel.com>
Signed-off-by: Ross Brattain <ross.b.brattain@intel.com>
---
 .../scenarios/availability/test_director.py        | 108 +++++++++++++++++++++
 1 file changed, 108 insertions(+)
 create mode 100644 yardstick/tests/unit/benchmark/scenarios/availability/test_director.py

(limited to 'yardstick/tests/unit/benchmark/scenarios/availability/test_director.py')

diff --git a/yardstick/tests/unit/benchmark/scenarios/availability/test_director.py b/yardstick/tests/unit/benchmark/scenarios/availability/test_director.py
new file mode 100644
index 000000000..72ce7b0d5
--- /dev/null
+++ b/yardstick/tests/unit/benchmark/scenarios/availability/test_director.py
@@ -0,0 +1,108 @@
+#!/usr/bin/env python
+
+##############################################################################
+# Copyright (c) 2016 Huan Li and others
+# lihuansse@tongji.edu.cn
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+# Unittest for yardstick.benchmark.scenarios.availability.director
+
+from __future__ import absolute_import
+import mock
+import unittest
+
+from yardstick.benchmark.scenarios.availability.director import Director
+
+
+# pylint: disable=unused-argument
+# disable this for now because I keep forgetting mock patch arg ordering
+
+
+@mock.patch('yardstick.benchmark.scenarios.availability.director.basemonitor')
+@mock.patch('yardstick.benchmark.scenarios.availability.director.baseattacker')
+@mock.patch(
+    'yardstick.benchmark.scenarios.availability.director.baseoperation')
+@mock.patch(
+    'yardstick.benchmark.scenarios.availability.director.baseresultchecker')
+class DirectorTestCase(unittest.TestCase):
+
+    def setUp(self):
+        self.scenario_cfg = {
+            'type': "general_scenario",
+            'options': {
+                'attackers': [{
+                    'fault_type': "general-attacker",
+                    'key': "kill-process"}],
+                'monitors': [{
+                    'monitor_type': "general-monitor",
+                    'key': "service-status"}],
+                'operations': [{
+                    'operation_type': 'general-operation',
+                    'key': 'service-status'}],
+                'resultCheckers': [{
+                    'checker_type': 'general-result-checker',
+                    'key': 'process-checker', }],
+                'steps': [
+                    {
+                        'actionKey': "service-status",
+                        'actionType': "operation",
+                        'index': 1},
+                    {
+                        'actionKey': "kill-process",
+                        'actionType': "attacker",
+                        'index': 2},
+                    {
+                        'actionKey': "process-checker",
+                        'actionType': "resultchecker",
+                        'index': 3},
+                    {
+                        'actionKey': "service-status",
+                        'actionType': "monitor",
+                        'index': 4},
+                ]
+            }
+        }
+        host = {
+            "ip": "10.20.0.5",
+            "user": "root",
+            "key_filename": "/root/.ssh/id_rsa"
+        }
+        self.ctx = {"nodes": {"node1": host}}
+
+    def test_director_all_successful(self, mock_checer, mock_opertion,
+                                     mock_attacker, mock_monitor):
+        ins = Director(self.scenario_cfg, self.ctx)
+        opertion_action = ins.createActionPlayer("operation", "service-status")
+        attacker_action = ins.createActionPlayer("attacker", "kill-process")
+        checker_action = ins.createActionPlayer("resultchecker",
+                                                "process-checker")
+        monitor_action = ins.createActionPlayer("monitor", "service-status")
+
+        opertion_rollback = ins.createActionRollbacker("operation",
+                                                       "service-status")
+        attacker_rollback = ins.createActionRollbacker("attacker",
+                                                       "kill-process")
+        ins.executionSteps.append(opertion_rollback)
+        ins.executionSteps.append(attacker_rollback)
+
+        opertion_action.action()
+        attacker_action.action()
+        checker_action.action()
+        monitor_action.action()
+
+        attacker_rollback.rollback()
+        opertion_rollback.rollback()
+
+        ins.stopMonitors()
+        ins.verify()
+        ins.knockoff()
+
+    def test_director_get_wrong_item(self, mock_checer, mock_opertion,
+                                     mock_attacker, mock_monitor):
+        ins = Director(self.scenario_cfg, self.ctx)
+        ins.createActionPlayer("wrong_type", "wrong_key")
+        ins.createActionRollbacker("wrong_type", "wrong_key")
-- 
cgit 1.2.3-korg