Signed-off-by: Niels Thykier <niels@thykier.net>ubuntu/rebased
parent
1fcccea83f
commit
ee12b76885
@ -0,0 +1,58 @@
|
||||
import unittest
|
||||
|
||||
from britney2.hints import HintParser
|
||||
|
||||
from . import MockObject, HINTS_ALL, TEST_HINTER
|
||||
|
||||
|
||||
def new_hint_paser(logger=None):
|
||||
if logger is None:
|
||||
def empty_logger(x, type='I'):
|
||||
pass
|
||||
logger = empty_logger
|
||||
fake_britney = MockObject(log=logger)
|
||||
hint_parser = HintParser(fake_britney)
|
||||
return hint_parser
|
||||
|
||||
|
||||
def parse_should_not_call_this_function(*args, **kwargs):
|
||||
raise AssertionError("Should not be called")
|
||||
|
||||
|
||||
class HintParsing(unittest.TestCase):
|
||||
|
||||
def test_parse_invalid_hints(self):
|
||||
hint_log = []
|
||||
hint_parser = new_hint_paser(lambda x, type='I': hint_log.append(x))
|
||||
|
||||
hint_parser.register_hint_type('min-10-arg', parse_should_not_call_this_function, min_args=10)
|
||||
hint_parser.register_hint_type('simple-hint', parse_should_not_call_this_function)
|
||||
|
||||
tests = [
|
||||
{
|
||||
'hint_text': 'min-10-arg foo bar',
|
||||
'permissions': HINTS_ALL,
|
||||
'error_message_contains': 'Needs at least 10 argument(s), got'
|
||||
},
|
||||
{
|
||||
'hint_text': 'undefined-hint with some arguments',
|
||||
'permissions': HINTS_ALL,
|
||||
'error_message_contains': 'Unknown hint found in'
|
||||
},
|
||||
{
|
||||
'hint_text': 'simple-hint foo/1.0',
|
||||
'permissions': ['not-this-hint'],
|
||||
'error_message_contains': 'not a part of the permitted hints for'
|
||||
},
|
||||
]
|
||||
|
||||
for test in tests:
|
||||
hint_parser.parse_hints(TEST_HINTER, test['permissions'], 'test-parse-hint', [test['hint_text']])
|
||||
assert len(hint_log) == 1
|
||||
assert test['error_message_contains'] in hint_log[0]
|
||||
assert hint_parser.hints.is_empty
|
||||
hint_log.clear()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
Loading…
Reference in new issue