def test_validate(self): for action, args in self.test_funcs.items(): for arg_pair in args['args']: arg = Argument(key=arg_pair['name'], format=arg_pair['type']) self.assertTrue(arg.validate(action=action)) test_funcs = {'func_name2': {'args': []}, 'func_name3': {'args': [{'name': 'junk_name1', 'type': 'junk_type1'}, {'name': 'junk_name2', 'type': 'junk_type2'}]}} for action, args in test_funcs.items(): for arg_pair in args['args']: arg = Argument(key=arg_pair['name'], format=arg_pair['type']) self.assertFalse(arg.validate(action=action))
def test_validate_against_invalid_action(self): for action, args in self.test_funcs.items(): for arg_pair in args['args']: arg = Argument(key=arg_pair['name'], format=arg_pair['type']) with self.assertRaises(KeyError): self.assertTrue(arg.validate(action='junkName'))