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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
# -*- coding: utf-8 -*-
"""
test
~~~~
Flask-CORS is a simple extension to Flask allowing you to support cross
origin resource sharing (CORS) using a simple decorator.
:copyright: (c) 2016 by Cory Dolphin.
:license: MIT, see LICENSE for more details.
"""
from datetime import timedelta
from ..base_test import FlaskCorsTestCase
from flask import Flask
from flask_cors import *
from flask_cors.core import *
class MaxAgeTestCase(FlaskCorsTestCase):
def setUp(self):
self.app = Flask(__name__)
@self.app.route('/defaults')
@cross_origin()
def defaults():
return 'Should only return headers on OPTIONS'
@self.app.route('/test_string')
@cross_origin(max_age=600)
def test_string():
return 'Open!'
@self.app.route('/test_time_delta')
@cross_origin(max_age=timedelta(minutes=10))
def test_time_delta():
return 'Open!'
def test_defaults(self):
''' By default, no max-age headers should be returned
'''
for resp in self.iter_responses('/defaults', origin='www.example.com'):
self.assertFalse(ACL_MAX_AGE in resp.headers)
def test_string(self):
''' If the methods parameter is defined, always return the allowed
methods defined by the user.
'''
resp = self.preflight('/test_string', origin='www.example.com')
self.assertEqual(resp.headers.get(ACL_MAX_AGE), '600')
def test_time_delta(self):
''' If the methods parameter is defined, always return the allowed
methods defined by the user.
'''
resp = self.preflight('/test_time_delta', origin='www.example.com')
self.assertEqual(resp.headers.get(ACL_MAX_AGE), '600')
if __name__ == "__main__":
unittest.main()
|
