--- /dev/null
+#!/usr/bin/env python
+
+##############################################################################
+# Copyright (c) 2016 Juan Qiu and others
+# juan_ qiu@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.attacker
+# .attacker_general
+
+import mock
+import unittest
+
+from yardstick.benchmark.scenarios.availability.attacker import baseattacker
+
+@mock.patch('yardstick.benchmark.scenarios.availability.attacker.'
+ 'attacker_general.ssh')
+class GeneralAttackerServiceTestCase(unittest.TestCase):
+
+ def setUp(self):
+ host = {
+ "ip": "10.20.0.5",
+ "user": "root",
+ "key_filename": "/root/.ssh/id_rsa"
+ }
+ self.context = {"node1": host}
+ self.attacker_cfg = {
+ 'fault_type': 'general-attacker',
+ 'action_parameter':{'process_name':'nova_api'},
+ 'rollback_parameter':{'process_name':'nova_api'},
+ 'key':'stop_service',
+ 'host': 'node1',
+ }
+
+ def test__attacker_service_all_successful(self, mock_ssh):
+
+ cls = baseattacker.BaseAttacker.get_attacker_cls(self.attacker_cfg)
+ ins = cls(self.attacker_cfg, self.context)
+
+ mock_ssh.SSH().execute.return_value = (0, "running", '')
+ ins.setup()
+ ins.inject_fault()
+ ins.recover()
+
+ def test__attacker_service_check_failuer(self, mock_ssh):
+
+ cls = baseattacker.BaseAttacker.get_attacker_cls(self.attacker_cfg)
+ ins = cls(self.attacker_cfg, self.context)
+
+ mock_ssh.SSH().execute.return_value = (0, "error check", '')
+ ins.setup()
--- /dev/null
+##############################################################################
+# Copyright (c) 2016 Juan Qiu and others
+# juan_ qiu@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
+##############################################################################
+import logging
+
+from baseattacker import BaseAttacker
+import yardstick.ssh as ssh
+from yardstick.benchmark.scenarios.availability import util
+
+LOG = logging.getLogger(__name__)
+
+
+class GeneralAttacker(BaseAttacker):
+
+ __attacker_type__ = 'general-attacker'
+
+ def setup(self):
+ LOG.debug("config:%s context:%s" % (self._config, self._context))
+ host = self._context.get(self._config['host'], None)
+ ip = host.get("ip", None)
+ user = host.get("user", "root")
+ key_filename = host.get("key_filename", "~/.ssh/id_rsa")
+
+ self.connection = ssh.SSH(user, ip, key_filename=key_filename)
+ self.connection.wait(timeout=600)
+ LOG.debug("ssh host success!")
+
+ self.key = self._config['key']
+
+ if "action_parameter" in self._config:
+ actionParameter = self._config['action_parameter']
+ str = util.buildshellparams(actionParameter)
+ LOG.debug("inject parameter is: {0}".format(actionParameter))
+ LOG.debug("inject parameter values are: {0}"
+ .format(actionParameter.values()))
+ l = list(item for item in actionParameter.values())
+ self.action_param = str.format(*l)
+
+ if "rollback_parameter" in self._config:
+ rollbackParameter = self._config['rollback_parameter']
+ str = util.buildshellparams(rollbackParameter)
+ LOG.debug("recover parameter is: {0}".format(rollbackParameter))
+ LOG.debug("recover parameter values are: {0}".
+ format(rollbackParameter.values()))
+ l = list(item for item in rollbackParameter.values())
+ self.rollback_param = str.format(*l)
+
+ self.fault_cfg = BaseAttacker.attacker_cfgs.get(self.key)
+ self.inject_script = self.get_script_fullpath(
+ self.fault_cfg['inject_script'])
+ self.recovery_script = self.get_script_fullpath(
+ self.fault_cfg['recovery_script'])
+
+ def inject_fault(self):
+ LOG.debug("{0} starting inject!".format(self.key))
+ LOG.debug("the inject_script path:{0}".format(self.inject_script))
+
+ if "action_parameter" in self._config:
+ LOG.debug("the shell command is: {0}".format(self.action_param))
+ exit_status, stdout, stderr = self.connection.execute(
+ self.action_param,
+ stdin=open(self.inject_script, "r"))
+ else:
+ exit_status, stdout, stderr = self.connection.execute(
+ "/bin/bash -s ",
+ stdin=open(self.inject_script, "r"))
+
+ LOG.debug("the inject_fault's exit status is: {0}".format(exit_status))
+ if exit_status == 0:
+ LOG.debug("success,the inject_fault's output is: {0}"
+ .format(stdout))
+ else:
+ LOG.error(
+ "the inject_fault's error, stdout:%s, stderr:%s" %
+ (stdout, stderr))
+
+ def recover(self):
+ if "rollback_parameter" in self._config:
+ LOG.debug("the shell command is: {0}".format(self.rollback_param))
+ exit_status, stdout, stderr = self.connection.execute(
+ self.rollback_param,
+ stdin=open(self.recovery_script, "r"))
+ else:
+ exit_status, stdout, stderr = self.connection.execute(
+ "/bin/bash -s ",
+ stdin=open(self.recovery_script, "r"))
"attacker_conf.yaml")
+class AttackerMgr(object):
+
+ def __init__(self):
+ self._attacker_list = []
+
+ def init_attackers(self, attacker_cfgs, context):
+ LOG.debug("attackerMgr confg: %s" % attacker_cfgs)
+
+ for cfg in attacker_cfgs:
+ attacker_cls = BaseAttacker.get_attacker_cls(cfg)
+ attacker_ins = attacker_cls(cfg, context)
+ attacker_ins.key = cfg['key']
+ attacker_ins.setup()
+ self._attacker_list.append(attacker_ins)
+
+ def __getitem__(self, item):
+ for obj in self._attacker_list:
+ if(obj.key == item):
+ return obj
+
+ def recover(self):
+ for _instance in self._attacker_list:
+ _instance.recover()
+
+
class BaseAttacker(object):
attacker_cfgs = {}
def get_script_fullpath(self, path):
base_path = os.path.dirname(attacker_conf_path)
return os.path.join(base_path, path)
+
+ def recover(self):
+ pass
Code Review / yardstick.git / commitdiff