def test_default_params_is_all_args(self):
self.assertEqual(ArgsKeyBit().params, '*')
def test_with_all_args(self):
self.kwargs['params'] = '*'
self.assertEqual(ArgsKeyBit().get_data(**self.kwargs), self.test_args)
def test_with_specified_args(self):
self.kwargs['params'] = test_arg_idx = [0, 2]
expected_args = [self.test_args[i] for i in test_arg_idx]
self.assertEqual(ArgsKeyBit().get_data(**self.kwargs), expected_args)
def test_with_no_args(self):
self.assertEqual(ArgsKeyBit().get_data(**self.kwargs), [])
class RequestKeyConstructor(DefaultKeyConstructor):
arguments = ArgsKeyBit()
key_arguments = KwargsKeyBit()
query_params = QueryParamsKeyBit()
user = UserKeyBit()
headers = HeadersKeyBit()
class ScoreboardKeyConstructor(DefaultKeyConstructor):
params_bit = QueryParamsKeyBit()
args_bit = ArgsKeyBit()
kwargs_bit = KwargsKeyBit()
class ContestTaskSolvedKeyConstructor(DefaultKeyConstructor):
args_bit = ArgsKeyBit()
kwargs_bit = KwargsKeyBit()
class ContestTaskListKeyConstructor(DefaultKeyConstructor):
user_bit = UserKeyBit()
args_bit = ArgsKeyBit()
kwargs_bit = KwargsKeyBit()
