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