diff options
Diffstat (limited to 'pysite/views/tests/index.py')
-rw-r--r-- | pysite/views/tests/index.py | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/pysite/views/tests/index.py b/pysite/views/tests/index.py index b96590c0..f99e3f3c 100644 --- a/pysite/views/tests/index.py +++ b/pysite/views/tests/index.py @@ -1,20 +1,23 @@ from flask import jsonify from schema import Schema -from pysite.base_route import RouteView +from pysite.base_route import APIView from pysite.constants import ValidationTypes from pysite.decorators import api_params -SCHEMA = Schema([{"test": str}]) +LIST_SCHEMA = Schema([{"test": str}]) +DICT_SCHEMA = Schema({"segfault": str}) -REQUIRED_KEYS = ["test"] - -class TestParamsView(RouteView): +class TestParamsView(APIView): path = "/testparams" name = "testparams" - @api_params(schema=SCHEMA, validation_type=ValidationTypes.params) + @api_params(schema=DICT_SCHEMA, validation_type=ValidationTypes.params) + def get(self, data): + return jsonify(data) + + @api_params(schema=LIST_SCHEMA, validation_type=ValidationTypes.params) def post(self, data): jsonified = jsonify(data) return jsonified |