This repository has been archived by the owner on Oct 12, 2017. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 1
/
test_tidy.py
74 lines (54 loc) · 2.17 KB
/
test_tidy.py
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
from cherrypy.test import test
test.prefer_parent_path()
import os
localDir = os.path.join(os.getcwd(), os.path.dirname(__file__))
tidy_path = os.path.join(localDir, "tidy")
import cherrypy
from cherrypy import tools
doctype = ('<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" '
'"http://www.w3.org/TR/xhtml1/DTD/strict.dtd">')
def setup_server():
class Root:
_cp_config = {
'tools.tidy.on': True,
'tools.tidy.tidy_path': tidy_path,
'tools.tidy.temp_dir': localDir,
}
def plaintext(self):
yield "Hello, world"
plaintext.exposed = True
plaintext._cp_config = {'tools.tidy.warnings': False}
def validhtml(self):
return "<html><body><h1>This should be valid</h1></body></html>"
validhtml.exposed = True
validhtml._cp_config = {'tools.tidy.warnings': False}
def warning(self, skip_doctype=False):
if skip_doctype:
# This should raise a warning
pass
else:
yield doctype
yield "<html><head><title>Meh</title></head>"
yield "<body>Normal body</body></html>"
warning.exposed = True
cherrypy.tree.mount(Root())
from cherrypy.test import helper
class TidyTest(helper.CPWebCase):
def test_Tidy_Tool(self):
if not os.path.exists(tidy_path) and not os.path.exists(tidy_path + ".exe"):
return self.skip("skipped (tidy not found) ")
self.getPage('/validhtml')
self.assertStatus(200)
self.assertBody("<html><body><h1>This should be valid</h1></body></html>")
self.getPage('/plaintext')
self.assertStatus(200)
self.assertBody('Hello, world')
self.getPage('/warning')
self.assertStatus(200)
self.assertBody(doctype + "<html><head><title>Meh</title></head>"
"<body>Normal body</body></html>")
self.getPage('/warning?skip_doctype=YES')
self.assertStatus(200)
self.assertInBody("Wrong HTML")
if __name__ == "__main__":
helper.testmain()