Mercurial > mozilla > hg > talosnames
annotate talosnames/web.py @ 40:10945dedde84
get query string working for TBPL suites
author | Jeff Hammel <jhammel@mozilla.com> |
---|---|
date | Tue, 24 Jul 2012 22:51:40 -0700 |
parents | ef8590b55605 |
children | 4dfa9c298e3d |
rev | line source |
---|---|
0 | 1 #!/usr/bin/env python |
2 | |
3 """ | |
4 web handler for talosnames | |
5 """ | |
6 | |
35 | 7 import optparse |
19
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
8 import os |
22 | 9 import pprint |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
10 import talos.test |
19
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
11 import tempita |
18 | 12 from api import TalosNames |
31
c6580ad2612f
better formatting of command
Jeff Hammel <jhammel@mozilla.com>
parents:
26
diff
changeset
|
13 from subprocess import list2cmdline |
0 | 14 from webob import Request, Response, exc |
15 | |
19
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
16 here = os.path.dirname(os.path.abspath(__file__)) |
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
17 template = os.path.join(here, 'templates', 'index.html') |
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
18 |
0 | 19 class Handler(object): |
20 | |
21 def __init__(self, **kw): | |
18 | 22 self.api = TalosNames() |
0 | 23 |
25 | 24 # get data |
21 | 25 suites = sorted(self.api.suites.keys()) |
26 tests = {} | |
27 for suite in suites: | |
28 try: | |
29 test = self.api.test_config(suite) | |
30 tests[suite] = test | |
31 except: | |
32 tests[suite] = None | |
40
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
33 self.suites = suites |
21 | 34 |
40
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
35 self.data = {'suites': self.suites, |
25 | 36 'commands': self.api.buildbot_commands, |
37 'tbpl': dict([(suite, self.api.tbpl_name(suite)) | |
38 for suite in suites]), | |
39 'tests': tests, | |
40 'pprint': pprint.pformat, | |
31
c6580ad2612f
better formatting of command
Jeff Hammel <jhammel@mozilla.com>
parents:
26
diff
changeset
|
41 'list2cmdline': list2cmdline |
25 | 42 } |
43 | |
26 | 44 paint = {} |
45 chrome = {} | |
46 graphserver = {} | |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
47 test_type = {} |
26 | 48 for suite in suites: |
49 for test in tests.get(suite) or []: | |
50 config = self.api.talos_config(suite) | |
51 _paint = '--mozAfterPaint' in self.data['commands'][suite] | |
52 _chrome = '--noChrome' not in self.data['commands'][suite] | |
53 extension = config.get('test_name_extension', '') | |
54 _extension = '' | |
55 if not _chrome: | |
56 _extension += '_nochrome' | |
57 if _paint: | |
58 _extension += '_paint' | |
59 if extension != _extension: | |
60 raise AssertionError | |
61 paint[suite] = _paint | |
62 chrome[suite] = _chrome | |
36 | 63 |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
64 # determine test extension |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
65 # TODO: move this to api.py |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
66 testname = test |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
67 testobj = talos.test.test_dict[testname] |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
68 if issubclass(testobj, talos.test.TsBase): |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
69 test_type.setdefault(suite, {})[test] = 'Startup Test' |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
70 elif issubclass(testobj, talos.test.PageloaderTest): |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
71 test_type.setdefault(suite, {})[test] = 'Page Load Test' |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
72 testname += extension |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
73 else: |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
74 raise Exception |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
75 |
36 | 76 # get graphserver data |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
77 names = self.api.graphserver_name(testname) |
36 | 78 if names: |
79 graphserver.setdefault(suite, {})[test] = [names] | |
80 else: | |
81 graphserver.setdefault(suite, {})[test] = None | |
26 | 82 self.data['graphserver'] = graphserver |
83 self.data['paint'] = paint | |
84 self.data['chrome'] = chrome | |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
85 self.data['test_type'] = test_type |
25 | 86 |
87 def __call__(self, environ, start_response): | |
88 request = Request(environ) | |
89 response = Response(content_type='text/html', | |
40
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
90 body=self.render(request)) |
25 | 91 return response(environ, start_response) |
92 | |
40
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
93 def render(self, request=None): |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
94 data = self.data.copy() |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
95 if request and 'tbpl' in request.GET: |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
96 tbplnames = [] |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
97 for name in request.GET.getall('tbpl'): |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
98 if not name.startswith('Talos'): |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
99 name = 'Talos ' + name |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
100 tbplnames.append(name) |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
101 suites = [] |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
102 for suite, value in self.data['tbpl'].items(): |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
103 if value in tbplnames: |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
104 suites.append(suite) |
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
105 data['suites'] = sorted(suites) |
32
1a9ec766f2b0
add a thing for tbpl short names
Jeff Hammel <jhammel@mozilla.com>
parents:
31
diff
changeset
|
106 contents = file(template).read() |
1a9ec766f2b0
add a thing for tbpl short names
Jeff Hammel <jhammel@mozilla.com>
parents:
31
diff
changeset
|
107 _template = tempita.HTMLTemplate(contents) |
40
10945dedde84
get query string working for TBPL suites
Jeff Hammel <jhammel@mozilla.com>
parents:
38
diff
changeset
|
108 return _template.substitute(data) |
19
aae0a420c4f7
should now actually do something
Jeff Hammel <jhammel@mozilla.com>
parents:
18
diff
changeset
|
109 |
0 | 110 if __name__ == '__main__': |
35 | 111 |
112 parser = optparse.OptionParser() | |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
113 parser.add_option('-o', '--output', dest='output', |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
114 help="file to output to") |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
115 parser.add_option('-p', '--port', dest='port', |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
116 default=8080, type='int', |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
117 help="port to serve on") |
35 | 118 options, args = parser.parse_args() |
119 | |
0 | 120 app = Handler() |
38
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
121 |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
122 if options.output: |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
123 f = file(options.output, 'w') |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
124 f.write(app.render()) |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
125 f.close() |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
126 else: |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
127 from wsgiref import simple_server |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
128 server = simple_server.make_server(host='0.0.0.0', port=options.port, app=app) |
ef8590b55605
determine actually correct name extension
Jeff Hammel <jhammel@mozilla.com>
parents:
36
diff
changeset
|
129 server.serve_forever() |
0 | 130 |
18 | 131 |