mirror of
https://github.com/moparisthebest/wget
synced 2024-07-03 16:38:41 -04:00
17 lines
555 B
Python
17 lines
555 B
Python
|
from exc.test_failed import TestFailed
|
||
|
|
from conf import hook
|
||
|
|
|
||
|
|
|
||
|
|
@hook(alias='ExpectedRetcode')
|
||
|
|
class ExpectedRetCode:
|
||
|
|
def __init__(self, expected_ret_code):
|
||
|
|
self.expected_ret_code = expected_ret_code
|
||
|
|
|
||
|
|
def __call__(self, test_obj):
|
||
|
|
if test_obj.ret_code != self.expected_ret_code:
|
||
|
|
failure = "Return codes do not match.\n" \
|
||
|
|
"Expected: %s\n" \
|
||
|
|
"Actual: %s" % (self.expected_ret_code,
|
||
|
|
test_obj.ret_code)
|
||
|
|
raise TestFailed(failure)
|