1 from cherrypy.test import test
2 test.prefer_parent_path()
3
4 import gzip, StringIO
5 import cherrypy
6 europoundUnicode = u'\x80\xa3'
7 europoundUtf8 = u'\x80\xa3'.encode('utf-8')
8 sing = u"\u6bdb\u6cfd\u4e1c: Sing, Little Birdie?"
9 sing8 = sing.encode('utf-8')
10 sing16 = sing.encode('utf-16')
11
12
18 index.exposed = True
19
20 def mao_zedong(self):
21 return sing
22 mao_zedong.exposed = True
23
24 def utf8(self):
25 return sing8
26 utf8.exposed = True
27 utf8._cp_config = {'tools.encode.encoding': 'utf-8'}
28
29 class GZIP:
30 def index(self):
31 yield "Hello, world"
32 index.exposed = True
33
34 def noshow(self):
35
36
37 raise IndexError()
38 yield "Here be dragons"
39 noshow.exposed = True
40
41 def noshow_stream(self):
42
43
44 raise IndexError()
45 yield "Here be dragons"
46 noshow_stream.exposed = True
47 noshow_stream._cp_config = {'response.stream': True}
48
49 cherrypy.config.update({
50 'environment': 'test_suite',
51 'tools.encode.on': True,
52 'tools.decode.on': True,
53 })
54
55 root = Root()
56 root.gzip = GZIP()
57 cherrypy.tree.mount(root, config={'/gzip': {'tools.gzip.on': True}})
58
59
60
61 from cherrypy.test import helper
62
63
65
70
72
73 self.getPage('/mao_zedong')
74 self.assertBody(sing8)
75
76
77 self.getPage('/mao_zedong', [('Accept-Charset', 'utf-16')])
78 self.assertBody(sing16)
79
80
81
82 self.getPage('/mao_zedong', [('Accept-Charset',
83 'iso-8859-1;q=1, utf-16;q=0.5')])
84 self.assertBody(sing16)
85
86
87 self.getPage('/mao_zedong', [('Accept-Charset', '*;q=1, utf-7;q=.2')])
88 self.assertBody(sing8)
89
90
91 self.getPage('/mao_zedong', [('Accept-Charset', 'iso-8859-1, *;q=0')])
92 self.assertStatus("406 Not Acceptable")
93 self.assertInBody("Your client sent this Accept-Charset header: "
94 "iso-8859-1, *;q=0. We tried these charsets: "
95 "iso-8859-1.")
96
97
98 self.getPage('/mao_zedong', [('Accept-Charset',
99 'us-ascii, ISO-8859-1, x-mac-ce')])
100 self.assertStatus("406 Not Acceptable")
101 self.assertInBody("Your client sent this Accept-Charset header: "
102 "us-ascii, ISO-8859-1, x-mac-ce. We tried these "
103 "charsets: x-mac-ce, us-ascii, ISO-8859-1.")
104
105
106 self.getPage('/utf8')
107 self.assertBody(sing8)
108 self.getPage('/utf8', [('Accept-Charset', 'us-ascii, ISO-8859-1')])
109 self.assertStatus("406 Not Acceptable")
110
112 zbuf = StringIO.StringIO()
113 zfile = gzip.GzipFile(mode='wb', fileobj=zbuf, compresslevel=9)
114 zfile.write("Hello, world")
115 zfile.close()
116
117 self.getPage('/gzip/', headers=[("Accept-Encoding", "gzip")])
118 self.assertInBody(zbuf.getvalue()[:3])
119 self.assertHeader("Vary", "Accept-Encoding")
120
121
122 self.getPage('/gzip/', headers=[("Accept-Encoding", "identity")])
123 self.assertNoHeader("Vary")
124 self.assertBody("Hello, world")
125
126 self.getPage('/gzip/', headers=[("Accept-Encoding", "gzip;q=0")])
127 self.assertNoHeader("Vary")
128 self.assertBody("Hello, world")
129
130 self.getPage('/gzip/', headers=[("Accept-Encoding", "*;q=0")])
131 self.assertStatus(406)
132 self.assertNoHeader("Vary")
133 self.assertErrorPage(406, "identity, gzip")
134
135
136 self.getPage('/gzip/noshow', headers=[("Accept-Encoding", "gzip")])
137 self.assertNoHeader('Content-Encoding')
138 self.assertStatus(500)
139 self.assertErrorPage(500, pattern="IndexError\n")
140
141
142
143
144
145 self.getPage('/gzip/noshow_stream',
146 headers=[("Accept-Encoding", "gzip")])
147 self.assertHeader('Content-Encoding', 'gzip')
148 self.assertMatchesBody(r"Unrecoverable error in the server.$")
149
150
151 if __name__ == "__main__":
152 setup_server()
153 helper.testmain()
154