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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
# coding: utf8
from __future__ import unicode_literals
from flask_api import renderers, status, FlaskAPI
from flask_api.decorators import set_renderers
from flask_api.mediatypes import MediaType
import unittest
class RendererTests(unittest.TestCase):
def test_render_json(self):
renderer = renderers.JSONRenderer()
content = renderer.render({'example': 'example'}, MediaType('application/json'))
expected = '{"example": "example"}'
self.assertEqual(content, expected)
def test_render_json_with_indent(self):
renderer = renderers.JSONRenderer()
content = renderer.render({'example': 'example'}, MediaType('application/json; indent=4'))
expected = '{\n "example": "example"\n}'
self.assertEqual(content, expected)
def test_render_browsable_encoding(self):
app = FlaskAPI(__name__)
@app.route('/_love', methods=['GET'])
def love():
return {"test": "I <3 Python"}
with app.test_client() as client:
response = client.get('/_love',
headers={"Accept": "text/html"})
html = str(response.get_data())
self.assertTrue('I <3 Python' in html)
self.assertTrue('<h1>Love</h1>' in html)
self.assertTrue('/_love' in html)
def test_render_browsable_linking(self):
app = FlaskAPI(__name__)
@app.route('/_happiness', methods=['GET'])
def happiness():
return {"url": "http://example.org",
"a tag": "<br />"}
with app.test_client() as client:
response = client.get('/_happiness',
headers={"Accept": "text/html"})
html = str(response.get_data())
self.assertTrue('<a href="http://example.org">http://example.org</a>' in html)
self.assertTrue('<br />'in html)
self.assertTrue('<h1>Happiness</h1>' in html)
self.assertTrue('/_happiness' in html)
def test_renderer_negotiation_not_implemented(self):
renderer = renderers.BaseRenderer()
with self.assertRaises(NotImplementedError) as context:
renderer.render(None, None)
msg = str(context.exception)
expected = '`render()` method must be implemented for class "BaseRenderer"'
self.assertEqual(msg, expected)
class OverrideParserSettings(unittest.TestCase):
def setUp(self):
class CustomRenderer1(renderers.BaseRenderer):
media_type = 'application/example1'
def render(self, data, media_type, **options):
return 'custom renderer 1'
class CustomRenderer2(renderers.BaseRenderer):
media_type = 'application/example2'
def render(self, data, media_type, **options):
return 'custom renderer 2'
app = FlaskAPI(__name__)
app.config['DEFAULT_RENDERERS'] = [CustomRenderer1]
app.config['PROPAGATE_EXCEPTIONS'] = True
@app.route('/custom_renderer_1/', methods=['GET'])
def custom_renderer_1():
return {'data': 'example'}
@app.route('/custom_renderer_2/', methods=['GET'])
@set_renderers([CustomRenderer2])
def custom_renderer_2():
return {'data': 'example'}
@app.route('/custom_renderer_2_as_args/', methods=['GET'])
@set_renderers(CustomRenderer2)
def custom_renderer_2_as_args():
return {'data': 'example'}
self.app = app
def test_overridden_parsers_with_settings(self):
with self.app.test_client() as client:
response = client.get('/custom_renderer_1/')
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.headers['Content-Type'], 'application/example1')
data = response.get_data().decode('utf8')
self.assertEqual(data, "custom renderer 1")
def test_overridden_parsers_with_decorator(self):
with self.app.test_client() as client:
data = {'example': 'example'}
response = client.get('/custom_renderer_2/', data=data)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.headers['Content-Type'], 'application/example2')
data = response.get_data().decode('utf8')
self.assertEqual(data, "custom renderer 2")
def test_overridden_parsers_with_decorator_as_args(self):
with self.app.test_client() as client:
data = {'example': 'example'}
response = client.get('/custom_renderer_2_as_args/', data=data)
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(response.headers['Content-Type'], 'application/example2')
data = response.get_data().decode('utf8')
self.assertEqual(data, "custom renderer 2")
|
