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
|
import unittest
import os
from flask import Flask, render_template
from flask_compress import Compress
class DefaultsTest(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.testing = True
Compress(self.app)
def test_mimetypes_default(self):
""" Tests COMPRESS_MIMETYPES default value is correctly set. """
defaults = ['text/html', 'text/css', 'text/xml', 'application/json',
'application/javascript']
self.assertEqual(self.app.config['COMPRESS_MIMETYPES'], defaults)
def test_level_default(self):
""" Tests COMPRESS_LEVEL default value is correctly set. """
self.assertEqual(self.app.config['COMPRESS_LEVEL'], 6)
def test_min_size_default(self):
""" Tests COMPRESS_MIN_SIZE default value is correctly set. """
self.assertEqual(self.app.config['COMPRESS_MIN_SIZE'], 500)
class InitTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.testing = True
def test_constructor_init(self):
Compress(self.app)
def test_delayed_init(self):
compress = Compress()
compress.init_app(self.app)
class UrlTests(unittest.TestCase):
def setUp(self):
self.app = Flask(__name__)
self.app.testing = True
small_path = os.path.join(os.getcwd(), 'tests', 'templates',
'small.html')
large_path = os.path.join(os.getcwd(), 'tests', 'templates',
'large.html')
self.small_size = os.path.getsize(small_path) - 1
self.large_size = os.path.getsize(large_path) - 1
Compress(self.app)
@self.app.route('/small/')
def small():
return render_template('small.html')
@self.app.route('/large/')
def large():
return render_template('large.html')
def client_get(self, ufs):
client = self.app.test_client()
response = client.get(ufs, headers=[('Accept-Encoding', 'gzip')])
self.assertEqual(response.status_code, 200)
return response
def test_compress_level(self):
""" Tests COMPRESS_LEVEL correctly affects response data. """
self.app.config['COMPRESS_LEVEL'] = 1
response = self.client_get('/large/')
response1_size = len(response.data)
self.app.config['COMPRESS_LEVEL'] = 6
response = self.client_get('/large/')
response6_size = len(response.data)
self.assertNotEqual(response1_size, response6_size)
def test_compress_min_size(self):
""" Tests COMPRESS_MIN_SIZE correctly affects response data. """
response = self.client_get('/small/')
self.assertEqual(self.small_size, len(response.data))
response = self.client_get('/large/')
self.assertNotEqual(self.large_size, len(response.data))
def test_mimetype_mismatch(self):
""" Tests if mimetype not in COMPRESS_MIMETYPES. """
response = self.client_get('/static/1.png')
self.assertEqual(response.mimetype, 'image/png')
def test_content_length_options(self):
client = self.app.test_client()
headers = [('Accept-Encoding', 'gzip')]
response = client.options('/small/', headers=headers)
self.assertEqual(response.status_code, 200)
if __name__ == '__main__':
unittest.main()
|
