yandex-tank/yandextank/plugins/RCAssert.py

''' Tank exit code check plugin '''
from yandextank.core import AbstractPlugin

class RCAssertPlugin(AbstractPlugin):
    ''' Apache Benchmark plugin '''
    SECTION = 'rcassert'

    def __init__(self, core):
        AbstractPlugin.__init__(self, core)
        self.ok_codes = []
        self.fail_code = 10

    @staticmethod
    def get_key():
        return __file__

    def get_available_options(self):
        return ["pass", "fail_code"]

    def configure(self):
        codes = self.get_option("pass", '').split(' ')
        for code in codes:
            if code:
                self.ok_codes.append(int(code))
        self.fail_code = int(self.get_option("fail_code", self.fail_code))

    def post_process(self, retcode):
        if not self.ok_codes:
            return retcode

        for code in self.ok_codes:
            self.log.debug("Comparing %s with %s codes", code, retcode)
            if code == int(retcode):
                self.log.info("Exit code %s was changed to 0 by RCAssert plugin", code)
                return 0

        self.log.info("Changing exit code to %s because RCAssert pass list was unsatisfied", self.fail_code)
        return self.fail_code