def query_args_func(key): args = { "limit": [0, 1, 2], "skip": [10, 20, 30] } return args.get(key, []) self.assertEqual((2, 30), get_skip_and_limit(query_args_func))
def test_get_skip_and_limit_only_limit(self): def query_args_func(key): args = { "limit": [1] } return args.get(key, []) self.assertEqual((0, 1), get_skip_and_limit(query_args_func))
def test_get_skip_and_limit_not_list(self): def query_args_func(key): return 1 self.assertEqual((0, 0), get_skip_and_limit(query_args_func))
def test_get_skip_and_limit_both_empty(self): def query_args_func(key): return [] self.assertEqual((0, 0), get_skip_and_limit(query_args_func))