mirror of
https://github.com/valitydev/yandex-tank.git
synced 2024-11-07 02:45:21 +00:00
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
''' Tank exit code check plugin '''
|
|
from ...core.interfaces import AbstractPlugin
|
|
|
|
|
|
class RCAssertPlugin(AbstractPlugin):
|
|
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
|