blob: a73052e486592e61103c67288b8556d03fb8893f (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
from tests import SiteTest
class DecoratorTests(SiteTest):
def test_decorator_api_json(self):
""" Check the json validation decorator """
from pysite.decorators import api_params
from pysite.constants import ValidationTypes
from schema import Schema
SCHEMA = Schema([{"user_id": int, "role": int}])
@api_params(schema=SCHEMA, validation_type=ValidationTypes.json)
def try_json_type(data):
return data
with self.assertRaises(AttributeError):
try_json_type("not json")
def test_decorator_params(self):
""" Check the params validation decorator """
response = self.client.post('/testparams?test=params')
self.assertEqual(response.status_code, 200)
self.assertEqual(response.json, [{'test': 'params'}])
|
