OLD | NEW |
1 # Copyright 2017 The Chromium Authors. All rights reserved. | 1 # Copyright 2017 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import base64 | 5 import base64 |
6 import json | 6 import json |
7 import mock | 7 import mock |
8 import sys | 8 import sys |
9 import urlparse | 9 import urlparse |
10 import webapp2 | 10 import webapp2 |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
45 testing_common.SetIsInternalUser('foo@bar.com', True) | 45 testing_common.SetIsInternalUser('foo@bar.com', True) |
46 self.SetCurrentUser('foo@bar.com', is_admin=True) | 46 self.SetCurrentUser('foo@bar.com', is_admin=True) |
47 oauth_patcher = mock.patch.object(api_auth, 'oauth') | 47 oauth_patcher = mock.patch.object(api_auth, 'oauth') |
48 self.addCleanup(oauth_patcher.stop) | 48 self.addCleanup(oauth_patcher.stop) |
49 mock_oauth = oauth_patcher.start() | 49 mock_oauth = oauth_patcher.start() |
50 SetGooglerOAuth(mock_oauth) | 50 SetGooglerOAuth(mock_oauth) |
51 | 51 |
52 def testPostHistogramEndToEnd(self): | 52 def testPostHistogramEndToEnd(self): |
53 data = json.dumps([ | 53 data = json.dumps([ |
54 { | 54 { |
55 'benchmarkName': 'benchmark', | 55 'values': ['benchmark'], |
56 'canonicalUrl': '', | |
57 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 56 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
58 'label': '', | 57 'type': 'GenericSet', |
59 'legacyTIRLabel': '', | |
60 'storyDisplayName': 'story', | |
61 'type': 'TelemetryInfo' | |
62 }, { | 58 }, { |
63 'values': [424242], | 59 'values': [424242], |
64 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 60 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
65 'type': 'GenericSet', | 61 'type': 'GenericSet', |
66 }, { | 62 }, { |
67 'buildNumber': 0, | 63 'values': ['master'], |
68 'buildbotMasterName': '', | |
69 'buildbotName': '', | |
70 'displayBotName': 'bot', | |
71 'displayMasterName': 'master', | |
72 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 64 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
73 'logUri': '', | 65 'type': 'GenericSet' |
74 'type': 'BuildbotInfo' | 66 }, { |
| 67 'values': ['bot'], |
| 68 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 69 'type': 'GenericSet' |
75 }, { | 70 }, { |
76 'binBoundaries': [1, [1, 1000, 20]], | 71 'binBoundaries': [1, [1, 1000, 20]], |
77 'diagnostics': { | 72 'diagnostics': { |
78 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 73 reserved_infos.MASTERS.name: |
| 74 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 75 reserved_infos.BOTS.name: |
| 76 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
79 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 77 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
80 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 78 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
81 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 79 reserved_infos.BENCHMARKS.name: |
| 80 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
82 }, | 81 }, |
83 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', | 82 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', |
84 'name': 'foo2', | 83 'name': 'foo2', |
85 'unit': 'count' | 84 'unit': 'count' |
86 } | 85 } |
87 ]) | 86 ]) |
88 self.testapp.post('/add_histograms', {'data': data}) | 87 self.testapp.post('/add_histograms', {'data': data}) |
89 self.ExecuteTaskQueueTasks('/add_histograms_queue', | 88 self.ExecuteTaskQueueTasks('/add_histograms_queue', |
90 add_histograms.TASK_QUEUE_NAME) | 89 add_histograms.TASK_QUEUE_NAME) |
91 diagnostics = histogram.SparseDiagnostic.query().fetch() | 90 diagnostics = histogram.SparseDiagnostic.query().fetch() |
92 self.assertEqual(2, len(diagnostics)) | 91 self.assertEqual(3, len(diagnostics)) |
93 histograms = histogram.Histogram.query().fetch() | 92 histograms = histogram.Histogram.query().fetch() |
94 self.assertEqual(1, len(histograms)) | 93 self.assertEqual(1, len(histograms)) |
95 | 94 |
96 class AddHistogramsTest(testing_common.TestCase): | 95 class AddHistogramsTest(testing_common.TestCase): |
97 | 96 |
98 def setUp(self): | 97 def setUp(self): |
99 super(AddHistogramsTest, self).setUp() | 98 super(AddHistogramsTest, self).setUp() |
100 app = webapp2.WSGIApplication([ | 99 app = webapp2.WSGIApplication([ |
101 ('/add_histograms', add_histograms.AddHistogramsHandler)]) | 100 ('/add_histograms', add_histograms.AddHistogramsHandler)]) |
102 self.testapp = webtest.TestApp(app) | 101 self.testapp = webtest.TestApp(app) |
103 testing_common.SetIsInternalUser('foo@bar.com', True) | 102 testing_common.SetIsInternalUser('foo@bar.com', True) |
104 self.SetCurrentUser('foo@bar.com', is_admin=True) | 103 self.SetCurrentUser('foo@bar.com', is_admin=True) |
105 oauth_patcher = mock.patch.object(api_auth, 'oauth') | 104 oauth_patcher = mock.patch.object(api_auth, 'oauth') |
106 self.addCleanup(oauth_patcher.stop) | 105 self.addCleanup(oauth_patcher.stop) |
107 mock_oauth = oauth_patcher.start() | 106 mock_oauth = oauth_patcher.start() |
108 SetGooglerOAuth(mock_oauth) | 107 SetGooglerOAuth(mock_oauth) |
109 | 108 |
110 def TaskParamsByGuid(self): | 109 def TaskParamsByGuid(self): |
111 tasks = self.GetTaskQueueTasks(add_histograms.TASK_QUEUE_NAME) | 110 tasks = self.GetTaskQueueTasks(add_histograms.TASK_QUEUE_NAME) |
112 params_by_guid = {} | 111 params_by_guid = {} |
113 for task in tasks: | 112 for task in tasks: |
114 params = urlparse.parse_qs(base64.b64decode(task['body'])) | 113 params = urlparse.parse_qs(base64.b64decode(task['body'])) |
115 guid = json.loads(params['data'][0])['guid'] | 114 guid = json.loads(params['data'][0])['guid'] |
116 params_by_guid[guid] = params | 115 params_by_guid[guid] = params |
117 return params_by_guid | 116 return params_by_guid |
118 | 117 |
119 def testPostHistogramSetsTestPathAndRevision(self): | 118 def testPostHistogramSetsTestPathAndRevision(self): |
120 data = json.dumps([ | 119 data = json.dumps([ |
121 { | 120 { |
122 'benchmarkName': 'benchmark', | 121 'values': ['benchmark'], |
123 'canonicalUrl': '', | |
124 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 122 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
125 'label': '', | 123 'type': 'GenericSet', |
126 'legacyTIRLabel': '', | |
127 'storyDisplayName': 'story', | |
128 'type': 'TelemetryInfo' | |
129 }, { | 124 }, { |
130 'values': [424242], | 125 'values': [424242], |
131 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 126 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
132 'type': 'GenericSet', | 127 'type': 'GenericSet', |
133 }, { | 128 }, { |
134 'buildNumber': 0, | 129 'values': ['master'], |
135 'buildbotMasterName': '', | |
136 'buildbotName': '', | |
137 'displayBotName': 'bot', | |
138 'displayMasterName': 'master', | |
139 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 130 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
140 'logUri': '', | 131 'type': 'GenericSet' |
141 'type': 'BuildbotInfo' | 132 }, { |
| 133 'values': ['bot'], |
| 134 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 135 'type': 'GenericSet' |
| 136 }, { |
| 137 'values': ['story'], |
| 138 'guid': 'dc894bd9-0b73-4400-9d95-b21ee371031d', |
| 139 'type': 'GenericSet', |
142 }, { | 140 }, { |
143 'binBoundaries': [1, [1, 1000, 20]], | 141 'binBoundaries': [1, [1, 1000, 20]], |
144 'diagnostics': { | 142 'diagnostics': { |
145 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 143 reserved_infos.STORIES.name: |
| 144 'dc894bd9-0b73-4400-9d95-b21ee371031d', |
| 145 reserved_infos.MASTERS.name: |
| 146 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 147 reserved_infos.BOTS.name: |
| 148 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
146 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 149 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
147 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 150 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
148 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 151 reserved_infos.BENCHMARKS.name: |
| 152 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
149 }, | 153 }, |
150 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 154 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
151 'name': 'foo', | 155 'name': 'foo', |
152 'unit': 'count' | 156 'unit': 'count' |
153 }, { | 157 }, { |
154 'binBoundaries': [1, [1, 1000, 20]], | 158 'binBoundaries': [1, [1, 1000, 20]], |
155 'diagnostics': { | 159 'diagnostics': { |
156 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 160 reserved_infos.MASTERS.name: |
| 161 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 162 reserved_infos.BOTS.name: |
| 163 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
157 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 164 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
158 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 165 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
159 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 166 reserved_infos.BENCHMARKS.name: |
| 167 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
| 168 reserved_infos.STORIES.name: |
| 169 'dc894bd9-0b73-4400-9d95-b21ee371031d', |
160 }, | 170 }, |
161 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', | 171 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', |
162 'name': 'foo2', | 172 'name': 'foo2', |
163 'unit': 'count' | 173 'unit': 'count' |
164 } | 174 } |
165 ]) | 175 ]) |
166 self.testapp.post('/add_histograms', {'data': data}) | 176 self.testapp.post('/add_histograms', {'data': data}) |
167 params_by_guid = self.TaskParamsByGuid() | 177 params_by_guid = self.TaskParamsByGuid() |
168 | 178 |
169 self.assertEqual(2, len(params_by_guid)) | 179 self.assertEqual(2, len(params_by_guid)) |
170 self.assertEqual( | 180 self.assertEqual( |
171 'master/bot/benchmark/foo/story', | 181 'master/bot/benchmark/foo/story', |
172 params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0']['test_path'][0]) | 182 params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0']['test_path'][0]) |
173 self.assertEqual( | 183 self.assertEqual( |
174 '424242', | 184 '424242', |
175 params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0']['revision'][0]) | 185 params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0']['revision'][0]) |
176 self.assertEqual( | 186 self.assertEqual( |
177 'master/bot/benchmark/foo2/story', | 187 'master/bot/benchmark/foo2/story', |
178 params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388']['test_path'][0]) | 188 params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388']['test_path'][0]) |
179 self.assertEqual( | 189 self.assertEqual( |
180 '424242', | 190 '424242', |
181 params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388']['revision'][0]) | 191 params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388']['revision'][0]) |
182 | 192 |
183 def testPostHistogramPassesHistogramLevelSparseDiagnostics(self): | 193 def testPostHistogramPassesHistogramLevelSparseDiagnostics(self): |
184 data = json.dumps([ | 194 data = json.dumps([ |
185 { | 195 { |
186 'benchmarkName': 'benchmark', | 196 'values': ['benchmark'], |
187 'canonicalUrl': '', | 197 'guid': '876d0fba-1d12-4c00-a7e9-5fed467e19e3', |
| 198 'type': 'GenericSet', |
| 199 }, { |
| 200 'values': [], |
188 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 201 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
189 'label': '', | 202 'type': 'GenericSet', |
190 'legacyTIRLabel': '', | |
191 'storyDisplayName': 'story', | |
192 'type': 'TelemetryInfo' | |
193 }, { | 203 }, { |
194 'values': [424242], | 204 'values': [424242], |
195 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 205 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
196 'type': 'GenericSet', | 206 'type': 'GenericSet', |
197 }, { | 207 }, { |
198 'buildNumber': 0, | 208 'values': ['master'], |
199 'buildbotMasterName': '', | |
200 'buildbotName': '', | |
201 'displayBotName': 'bot', | |
202 'displayMasterName': 'master', | |
203 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 209 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
204 'logUri': '', | 210 'type': 'GenericSet' |
205 'type': 'BuildbotInfo' | 211 }, { |
| 212 'values': ['bot'], |
| 213 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 214 'type': 'GenericSet' |
206 }, { | 215 }, { |
207 'binBoundaries': [1, [1, 1000, 20]], | 216 'binBoundaries': [1, [1, 1000, 20]], |
208 'diagnostics': { | 217 'diagnostics': { |
209 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 218 reserved_infos.MASTERS.name: |
| 219 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 220 reserved_infos.BOTS.name: |
| 221 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
210 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 222 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
211 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 223 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
212 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 224 reserved_infos.BENCHMARKS.name: |
| 225 '876d0fba-1d12-4c00-a7e9-5fed467e19e3', |
213 }, | 226 }, |
214 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 227 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
215 'name': 'foo', | 228 'name': 'foo', |
216 'unit': 'count' | 229 'unit': 'count' |
217 }, { | 230 }, { |
218 'binBoundaries': [1, [1, 1000, 20]], | 231 'binBoundaries': [1, [1, 1000, 20]], |
219 'diagnostics': { | 232 'diagnostics': { |
220 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 233 reserved_infos.MASTERS.name: |
| 234 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 235 reserved_infos.BOTS.name: |
| 236 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 237 reserved_infos.GPUS.name: |
| 238 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
221 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 239 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
222 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 240 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
223 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 241 reserved_infos.BENCHMARKS.name: |
| 242 '876d0fba-1d12-4c00-a7e9-5fed467e19e3', |
224 }, | 243 }, |
225 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', | 244 'guid': '2a714c36-f4ef-488d-8bee-93c7e3149388', |
226 'name': 'foo2', | 245 'name': 'foo2', |
227 'unit': 'count' | 246 'unit': 'count' |
228 } | 247 } |
229 ]) | 248 ]) |
230 self.testapp.post('/add_histograms', {'data': data}) | 249 self.testapp.post('/add_histograms', {'data': data}) |
231 | 250 |
232 params_by_guid = self.TaskParamsByGuid() | 251 params_by_guid = self.TaskParamsByGuid() |
233 params = params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388'] | 252 params = params_by_guid['2a714c36-f4ef-488d-8bee-93c7e3149388'] |
234 diagnostics = json.loads(params['diagnostics'][0]) | 253 diagnostics = json.loads(params['diagnostics'][0]) |
235 | 254 |
236 self.assertEqual(1, len(diagnostics)) | 255 self.assertEqual(1, len(diagnostics)) |
237 self.assertEqual( | 256 self.assertEqual( |
238 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', diagnostics[0]['guid']) | 257 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', diagnostics[0]['guid']) |
239 | 258 |
240 def testPostHistogram_AddsNewSparseDiagnostic(self): | 259 def testPostHistogram_AddsNewSparseDiagnostic(self): |
241 diag_dict = { | 260 diag_dict = { |
242 'buildNumber': 0, | 261 'values': ['master'], |
243 'buildbotMasterName': '', | 262 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
244 'buildbotName': 'buildbotmaster0', | 263 'type': 'GenericSet' |
245 'displayBotName': 'bot', | |
246 'displayMasterName': 'master', | |
247 'guid': '6ce177ab-3fdb-44cb-aa8d-9ed49765d810', | |
248 'logUri': '', | |
249 'type': 'BuildbotInfo' | |
250 } | 264 } |
251 diag = histogram.SparseDiagnostic( | 265 diag = histogram.SparseDiagnostic( |
252 data=diag_dict, start_revision=1, end_revision=sys.maxint, | 266 data=diag_dict, start_revision=1, end_revision=sys.maxint, |
253 test=utils.TestKey('master/bot/benchmark')) | 267 test=utils.TestKey('master/bot/benchmark')) |
254 diag.put() | 268 diag.put() |
255 data = json.dumps([ | 269 data = json.dumps([ |
256 { | 270 { |
257 'benchmarkName': 'benchmark', | 271 'values': ['benchmark'], |
258 'canonicalUrl': '', | |
259 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 272 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
260 'label': '', | 273 'type': 'GenericSet', |
261 'legacyTIRLabel': '', | |
262 'storyDisplayName': 'story', | |
263 'type': 'TelemetryInfo' | |
264 }, { | 274 }, { |
265 'values': [424242], | 275 'values': [424242], |
266 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 276 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
267 'type': 'GenericSet', | 277 'type': 'GenericSet', |
268 }, { | 278 }, { |
269 'buildNumber': 0, | 279 'values': ['master'], |
270 'buildbotMasterName': '', | |
271 'buildbotName': 'buildbotmaster1', | |
272 'displayBotName': 'bot', | |
273 'displayMasterName': 'master', | |
274 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 280 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
275 'logUri': '', | 281 'type': 'GenericSet' |
276 'type': 'BuildbotInfo' | 282 }, { |
| 283 'values': ['bot'], |
| 284 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 285 'type': 'GenericSet' |
277 }, { | 286 }, { |
278 'binBoundaries': [1, [1, 1000, 20]], | 287 'binBoundaries': [1, [1, 1000, 20]], |
279 'diagnostics': { | 288 'diagnostics': { |
280 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 289 reserved_infos.MASTERS.name: |
| 290 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 291 reserved_infos.BOTS.name: |
| 292 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
281 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 293 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
282 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 294 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
283 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 295 reserved_infos.BENCHMARKS.name: |
| 296 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
284 }, | 297 }, |
285 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 298 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
286 'name': 'foo', | 299 'name': 'foo', |
287 'unit': 'count'} | 300 'unit': 'count'} |
288 ]) | 301 ]) |
289 self.testapp.post('/add_histograms', {'data': data}) | 302 self.testapp.post('/add_histograms', {'data': data}) |
290 | 303 |
291 diagnostics = histogram.SparseDiagnostic.query().fetch() | 304 diagnostics = histogram.SparseDiagnostic.query().fetch() |
292 params_by_guid = self.TaskParamsByGuid() | 305 params_by_guid = self.TaskParamsByGuid() |
293 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] | 306 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] |
294 hist = json.loads(params['data'][0]) | 307 hist = json.loads(params['data'][0]) |
295 buildbot_info = hist['diagnostics']['buildbot'] | |
296 | 308 |
297 self.assertEqual(2, len(diagnostics)) | 309 self.assertEqual(4, len(diagnostics)) |
298 self.assertEqual('e9c2891d-2b04-413f-8cf4-099827e67626', buildbot_info) | 310 self.assertEqual( |
| 311 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 312 hist['diagnostics'][reserved_infos.MASTERS.name]) |
299 | 313 |
300 def testPostHistogram_DeduplicatesSameSparseDiagnostic(self): | 314 def testPostHistogram_DeduplicatesSameSparseDiagnostic(self): |
301 diag_dict = { | 315 diag_dict = { |
302 'buildNumber': 0, | 316 'values': ['master'], |
303 'buildbotMasterName': '', | 317 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
304 'buildbotName': 'buildbotmaster', | 318 'type': 'GenericSet' |
305 'displayBotName': 'bot', | |
306 'displayMasterName': 'master', | |
307 'guid': '6ce177ab-3fdb-44cb-aa8d-9ed49765d810', | |
308 'logUri': '', | |
309 'type': 'BuildbotInfo' | |
310 } | 319 } |
311 diag = histogram.SparseDiagnostic( | 320 diag = histogram.SparseDiagnostic( |
312 id='e9c2891d-2b04-413f-8cf4-099827e67626', data=diag_dict, | 321 id='e9c2891d-2b04-413f-8cf4-099827e67626', data=diag_dict, |
313 start_revision=1, end_revision=sys.maxint, | 322 start_revision=1, end_revision=sys.maxint, |
314 test=utils.TestKey('master/bot/benchmark')) | 323 test=utils.TestKey('master/bot/benchmark')) |
315 diag.put() | 324 diag.put() |
316 data = json.dumps([ | 325 data = json.dumps([ |
317 { | 326 { |
318 'benchmarkName': 'benchmark', | 327 'values': ['benchmark'], |
319 'canonicalUrl': '', | |
320 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 328 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
321 'label': '', | 329 'type': 'GenericSet', |
322 'legacyTIRLabel': '', | |
323 'storyDisplayName': 'story', | |
324 'type': 'TelemetryInfo' | |
325 }, { | 330 }, { |
326 'values': [424242], | 331 'values': [424242], |
327 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 332 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
328 'type': 'GenericSet', | 333 'type': 'GenericSet', |
329 }, { | 334 }, { |
330 'buildNumber': 0, | 335 'values': ['master'], |
331 'buildbotMasterName': '', | |
332 'buildbotName': 'buildbotmaster', | |
333 'displayBotName': 'bot', | |
334 'displayMasterName': 'master', | |
335 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 336 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
336 'logUri': '', | 337 'type': 'GenericSet' |
337 'type': 'BuildbotInfo' | 338 }, { |
| 339 'values': ['bot'], |
| 340 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 341 'type': 'GenericSet' |
338 }, { | 342 }, { |
339 'binBoundaries': [1, [1, 1000, 20]], | 343 'binBoundaries': [1, [1, 1000, 20]], |
340 'diagnostics': { | 344 'diagnostics': { |
341 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 345 reserved_infos.MASTERS.name: |
| 346 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 347 reserved_infos.BOTS.name: |
| 348 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
342 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 349 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
343 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 350 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
344 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 351 reserved_infos.BENCHMARKS.name: |
| 352 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
345 }, | 353 }, |
346 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 354 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
347 'name': 'foo', | 355 'name': 'foo', |
348 'unit': 'count' | 356 'unit': 'count' |
349 } | 357 } |
350 ]) | 358 ]) |
351 self.testapp.post('/add_histograms', {'data': data}) | 359 self.testapp.post('/add_histograms', {'data': data}) |
352 | 360 |
353 diagnostics = histogram.SparseDiagnostic.query().fetch() | 361 diagnostics = histogram.SparseDiagnostic.query().fetch() |
354 params_by_guid = self.TaskParamsByGuid() | 362 params_by_guid = self.TaskParamsByGuid() |
355 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] | 363 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] |
356 hist = json.loads(params['data'][0]) | 364 hist = json.loads(params['data'][0]) |
357 buildbot_info = hist['diagnostics']['buildbot'] | |
358 | 365 |
359 self.assertEqual(1, len(diagnostics)) | 366 self.assertEqual(3, len(diagnostics)) |
360 self.assertEqual('6ce177ab-3fdb-44cb-aa8d-9ed49765d810', buildbot_info) | 367 self.assertEqual( |
| 368 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 369 hist['diagnostics'][reserved_infos.MASTERS.name]) |
361 | 370 |
362 def testPostHistogramFailsWithoutHistograms(self): | 371 def testPostHistogramFailsWithoutHistograms(self): |
363 data = json.dumps([ | 372 data = json.dumps([ |
364 { | 373 { |
365 'benchmarkName': 'benchmark', | 374 'values': ['benchmark'], |
366 'canonicalUrl': '', | |
367 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 375 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
368 'label': '', | 376 'type': 'GenericSet', |
369 'legacyTIRLabel': '', | |
370 'storyDisplayName': 'story', | |
371 'type': 'TelemetryInfo' | |
372 }, { | 377 }, { |
373 'buildNumber': 0, | 378 'values': ['master'], |
374 'buildbotMasterName': '', | |
375 'buildbotName': 'buildbotmaster1', | |
376 'displayBotName': 'bot', | |
377 'displayMasterName': 'master', | |
378 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 379 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
379 'logUri': '', | 380 'type': 'GenericSet' |
380 'type': 'BuildbotInfo' | 381 }, { |
| 382 'values': ['bot'], |
| 383 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 384 'type': 'GenericSet' |
381 } | 385 } |
382 ]) | 386 ]) |
383 self.testapp.post('/add_histograms', {'data': data}, status=400) | 387 self.testapp.post('/add_histograms', {'data': data}, status=400) |
384 | 388 |
385 def testPostHistogramFailsWithoutBuildbotInfo(self): | 389 def testPostHistogramFailsWithoutBuildbotInfo(self): |
386 data = json.dumps([ | 390 data = json.dumps([ |
387 { | 391 { |
388 'benchmarkName': 'benchmark', | 392 'values': ['benchmark'], |
389 'canonicalUrl': '', | |
390 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 393 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
391 'label': '', | 394 'type': 'GenericSet' |
392 'legacyTIRLabel': '', | |
393 'storyDisplayName': 'story', | |
394 'type': 'TelemetryInfo' | |
395 }, { | 395 }, { |
396 'values': [424242], | 396 'values': [424242], |
397 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 397 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
398 'type': 'GenericSet', | 398 'type': 'GenericSet', |
399 }, { | 399 }, { |
400 'binBoundaries': [1, [1, 1000, 20]], | 400 'binBoundaries': [1, [1, 1000, 20]], |
401 'diagnostics': { | 401 'diagnostics': { |
402 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 402 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
403 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 403 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
404 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 404 reserved_infos.BENCHMARKS.name: |
| 405 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
405 }, | 406 }, |
406 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 407 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
407 'name': 'foo', | 408 'name': 'foo', |
408 'unit': 'count' | 409 'unit': 'count' |
409 } | 410 } |
410 ]) | 411 ]) |
411 self.testapp.post('/add_histograms', {'data': data}, status=400) | 412 self.testapp.post('/add_histograms', {'data': data}, status=400) |
412 | 413 |
413 def testPostHistogramFailsWithoutChromiumCommit(self): | 414 def testPostHistogramFailsWithoutChromiumCommit(self): |
414 data = json.dumps([ | 415 data = json.dumps([ |
415 { | 416 { |
416 'benchmarkName': 'benchmark', | 417 'values': ['benchmark'], |
417 'canonicalUrl': '', | |
418 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 418 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
419 'label': '', | 419 'type': 'GenericSet', |
420 'legacyTIRLabel': '', | |
421 'storyDisplayName': 'story', | |
422 'type': 'TelemetryInfo' | |
423 }, { | 420 }, { |
424 'buildNumber': 0, | 421 'values': ['master'], |
425 'buildbotMasterName': '', | |
426 'buildbotName': 'buildbotmaster1', | |
427 'displayBotName': 'bot', | |
428 'displayMasterName': 'master', | |
429 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 422 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
430 'logUri': '', | 423 'type': 'GenericSet' |
431 'type': 'BuildbotInfo' | 424 }, { |
| 425 'values': ['bot'], |
| 426 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 427 'type': 'GenericSet' |
432 }, { | 428 }, { |
433 'binBoundaries': [1, [1, 1000, 20]], | 429 'binBoundaries': [1, [1, 1000, 20]], |
434 'diagnostics': { | 430 'diagnostics': { |
435 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 431 reserved_infos.MASTERS.name: |
436 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae' | 432 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 433 reserved_infos.BOTS.name: |
| 434 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 435 reserved_infos.BENCHMARKS.name: |
| 436 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
437 }, | 437 }, |
438 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 438 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
439 'name': 'foo', | 439 'name': 'foo', |
440 'unit': 'count'} | 440 'unit': 'count'} |
441 ]) | 441 ]) |
442 self.testapp.post('/add_histograms', {'data': data}, status=400) | 442 self.testapp.post('/add_histograms', {'data': data}, status=400) |
443 | 443 |
444 def testPostHistogramFailsWithoutTelemetryInfo(self): | 444 def testPostHistogramFailsWithoutBenchmark(self): |
445 data = json.dumps([ | 445 data = json.dumps([ |
446 { | 446 { |
447 'values': [424242], | 447 'values': [424242], |
448 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 448 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
449 'type': 'GenericSet', | 449 'type': 'GenericSet', |
450 }, { | 450 }, { |
451 'buildNumber': 0, | 451 'values': ['master'], |
452 'buildbotMasterName': '', | |
453 'buildbotName': '', | |
454 'displayBotName': 'bot', | |
455 'displayMasterName': 'master', | |
456 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 452 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
457 'logUri': '', | 453 'type': 'GenericSet' |
458 'type': 'BuildbotInfo' | 454 }, { |
| 455 'values': ['bot'], |
| 456 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 457 'type': 'GenericSet' |
459 }, { | 458 }, { |
460 'binBoundaries': [1, [1, 1000, 20]], | 459 'binBoundaries': [1, [1, 1000, 20]], |
461 'diagnostics': { | 460 'diagnostics': { |
462 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 461 reserved_infos.MASTERS.name: |
| 462 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 463 reserved_infos.BOTS.name: |
| 464 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
463 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 465 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
464 '25f0a111-9bb4-4cea-b0c1-af2609623160' | 466 '25f0a111-9bb4-4cea-b0c1-af2609623160' |
465 }, | 467 }, |
466 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 468 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
467 'name': 'foo', | 469 'name': 'foo', |
468 'unit': 'count' | 470 'unit': 'count' |
469 } | 471 } |
470 ]) | 472 ]) |
471 self.testapp.post('/add_histograms', {'data': data}, status=400) | 473 self.testapp.post('/add_histograms', {'data': data}, status=400) |
472 | 474 |
473 def testPostHistogram_AddsSparseDiagnosticByName(self): | 475 def testPostHistogram_AddsSparseDiagnosticByName(self): |
474 data = json.dumps([ | 476 data = json.dumps([ |
475 { | 477 { |
476 'type': 'GenericSet', | 478 'type': 'GenericSet', |
477 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', | 479 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', |
478 'values': ['alice@chromium.org'] | 480 'values': ['alice@chromium.org'] |
479 }, | 481 }, |
480 { | 482 { |
481 'benchmarkName': 'benchmark', | 483 'values': ['benchmark'], |
482 'canonicalUrl': '', | |
483 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 484 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
484 'label': '', | 485 'type': 'GenericSet', |
485 'legacyTIRLabel': '', | |
486 'storyDisplayName': 'story', | |
487 'type': 'TelemetryInfo' | |
488 }, { | 486 }, { |
489 'values': [424242], | 487 'values': [424242], |
490 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 488 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
491 'type': 'GenericSet', | 489 'type': 'GenericSet', |
492 }, { | 490 }, { |
493 'buildNumber': 0, | 491 'values': ['master'], |
494 'buildbotMasterName': '', | |
495 'buildbotName': 'buildbotmaster1', | |
496 'displayBotName': 'bot', | |
497 'displayMasterName': 'master', | |
498 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 492 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
499 'logUri': '', | 493 'type': 'GenericSet' |
500 'type': 'BuildbotInfo' | 494 }, { |
| 495 'values': ['bot'], |
| 496 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 497 'type': 'GenericSet' |
501 }, { | 498 }, { |
502 'binBoundaries': [1, [1, 1000, 20]], | 499 'binBoundaries': [1, [1, 1000, 20]], |
503 'diagnostics': { | 500 'diagnostics': { |
504 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 501 reserved_infos.OWNERS.name: |
| 502 'cabb59fe-4bcf-4512-881c-d038c7a80635', |
| 503 reserved_infos.MASTERS.name: |
| 504 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 505 reserved_infos.BOTS.name: |
| 506 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
505 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 507 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
506 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 508 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
507 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 509 reserved_infos.BENCHMARKS.name: |
508 reserved_infos.OWNERS.name: | 510 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
509 'cabb59fe-4bcf-4512-881c-d038c7a80635' | |
510 }, | 511 }, |
511 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 512 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
512 'name': 'foo', | 513 'name': 'foo', |
513 'unit': 'count'} | 514 'unit': 'count'} |
514 ]) | 515 ]) |
515 | 516 |
516 self.testapp.post('/add_histograms', {'data': data}) | 517 self.testapp.post('/add_histograms', {'data': data}) |
517 | 518 |
518 diagnostics = histogram.SparseDiagnostic.query().fetch() | 519 diagnostics = histogram.SparseDiagnostic.query().fetch() |
519 params_by_guid = self.TaskParamsByGuid() | 520 params_by_guid = self.TaskParamsByGuid() |
520 | 521 |
521 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] | 522 params = params_by_guid['4989617a-14d6-4f80-8f75-dafda2ff13b0'] |
522 hist = json.loads(params['data'][0]) | 523 hist = json.loads(params['data'][0]) |
523 owners_info = hist['diagnostics'][reserved_infos.OWNERS.name] | 524 owners_info = hist['diagnostics'][reserved_infos.OWNERS.name] |
524 self.assertEqual(2, len(diagnostics)) | 525 self.assertEqual(4, len(diagnostics)) |
525 self.assertEqual(reserved_infos.OWNERS.name, diagnostics[0].name) | 526 self.assertEqual(reserved_infos.BENCHMARKS.name, diagnostics[0].name) |
526 self.assertEqual(['alice@chromium.org'], diagnostics[0].data['values']) | 527 self.assertEqual(reserved_infos.BOTS.name, diagnostics[1].name) |
| 528 self.assertEqual(reserved_infos.OWNERS.name, diagnostics[2].name) |
| 529 self.assertEqual(reserved_infos.MASTERS.name, diagnostics[3].name) |
| 530 self.assertEqual(['benchmark'], diagnostics[0].data['values']) |
| 531 self.assertEqual(['bot'], diagnostics[1].data['values']) |
| 532 self.assertEqual(['alice@chromium.org'], diagnostics[2].data['values']) |
| 533 self.assertEqual(['master'], diagnostics[3].data['values']) |
527 self.assertEqual('cabb59fe-4bcf-4512-881c-d038c7a80635', owners_info) | 534 self.assertEqual('cabb59fe-4bcf-4512-881c-d038c7a80635', owners_info) |
528 | 535 |
529 def testPostHistogram_AddsSparseDiagnosticByName_OnlyOnce(self): | 536 def testPostHistogram_AddsSparseDiagnosticByName_OnlyOnce(self): |
530 data = json.dumps([ | 537 data = json.dumps([ |
531 { | 538 { |
532 'type': 'GenericSet', | 539 'type': 'GenericSet', |
533 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', | 540 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', |
534 'values': ['alice@chromium.org'] | 541 'values': ['alice@chromium.org'] |
535 }, | 542 }, |
536 { | 543 { |
537 'benchmarkName': 'benchmark', | 544 'values': ['benchmark'], |
538 'canonicalUrl': '', | |
539 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 545 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
540 'label': '', | 546 'type': 'GenericSet', |
541 'legacyTIRLabel': '', | |
542 'storyDisplayName': 'story', | |
543 'type': 'TelemetryInfo' | |
544 }, { | 547 }, { |
545 'values': [424242], | 548 'values': [424242], |
546 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 549 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
547 'type': 'GenericSet' | 550 'type': 'GenericSet' |
548 }, { | 551 }, { |
549 'buildNumber': 0, | 552 'values': ['master'], |
550 'buildbotMasterName': '', | |
551 'buildbotName': 'buildbotmaster1', | |
552 'displayBotName': 'bot', | |
553 'displayMasterName': 'master', | |
554 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 553 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
555 'logUri': '', | 554 'type': 'GenericSet' |
556 'type': 'BuildbotInfo' | 555 }, { |
| 556 'values': ['bot'], |
| 557 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 558 'type': 'GenericSet' |
557 }, { | 559 }, { |
558 'binBoundaries': [1, [1, 1000, 20]], | 560 'binBoundaries': [1, [1, 1000, 20]], |
559 'diagnostics': { | 561 'diagnostics': { |
560 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 562 reserved_infos.MASTERS.name: |
| 563 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 564 reserved_infos.BOTS.name: |
| 565 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
561 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 566 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
562 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 567 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
563 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 568 reserved_infos.BENCHMARKS.name: |
| 569 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
564 reserved_infos.OWNERS.name: | 570 reserved_infos.OWNERS.name: |
565 'cabb59fe-4bcf-4512-881c-d038c7a80635' | 571 'cabb59fe-4bcf-4512-881c-d038c7a80635' |
566 }, | 572 }, |
567 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 573 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
568 'name': 'foo', | 574 'name': 'foo', |
569 'unit': 'count' | 575 'unit': 'count' |
570 }, { | 576 }, { |
571 'binBoundaries': [1, [1, 1000, 20]], | 577 'binBoundaries': [1, [1, 1000, 20]], |
572 'diagnostics': { | 578 'diagnostics': { |
573 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 579 reserved_infos.MASTERS.name: |
| 580 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 581 reserved_infos.BOTS.name: |
| 582 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
574 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 583 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
575 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 584 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
576 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | |
577 reserved_infos.OWNERS.name: | 585 reserved_infos.OWNERS.name: |
578 'cabb59fe-4bcf-4512-881c-d038c7a80635' | 586 'cabb59fe-4bcf-4512-881c-d038c7a80635', |
| 587 reserved_infos.BENCHMARKS.name: |
| 588 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
579 }, | 589 }, |
580 'guid': '5239617a-14d6-4f80-8f75-dafda2ff13b1', | 590 'guid': '5239617a-14d6-4f80-8f75-dafda2ff13b1', |
581 'name': 'bar', | 591 'name': 'bar', |
582 'unit': 'count' | 592 'unit': 'count' |
583 }]) | 593 }]) |
584 | 594 |
585 self.testapp.post('/add_histograms', {'data': data}) | 595 self.testapp.post('/add_histograms', {'data': data}) |
586 | 596 |
587 diagnostics = histogram.SparseDiagnostic.query().fetch() | 597 diagnostics = histogram.SparseDiagnostic.query().fetch() |
588 | 598 |
589 self.assertEqual(2, len(diagnostics)) | 599 self.assertEqual(4, len(diagnostics)) |
590 self.assertEqual(reserved_infos.OWNERS.name, diagnostics[0].name) | 600 self.assertEqual(reserved_infos.BOTS.name, diagnostics[1].name) |
591 self.assertNotEqual(reserved_infos.OWNERS.name, diagnostics[1].name) | 601 self.assertNotEqual(reserved_infos.BOTS.name, diagnostics[0].name) |
592 | 602 |
593 def testPostHistogram_AddsSparseDiagnosticByName_ErrorsIfDiverging(self): | 603 def testPostHistogram_AddsSparseDiagnosticByName_ErrorsIfDiverging(self): |
594 data = json.dumps([ | 604 data = json.dumps([ |
595 { | 605 { |
596 'type': 'GenericSet', | 606 'type': 'GenericSet', |
597 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', | 607 'guid': 'cabb59fe-4bcf-4512-881c-d038c7a80635', |
598 'values': ['alice@chromium.org'] | 608 'values': ['alice@chromium.org'] |
599 }, { | 609 }, { |
600 'type': 'GenericSet', | 610 'type': 'GenericSet', |
601 'guid': '7c5bd92f-4146-411b-9192-248ffc1be92c', | 611 'guid': '7c5bd92f-4146-411b-9192-248ffc1be92c', |
602 'values': ['bob@chromium.org'] | 612 'values': ['bob@chromium.org'] |
603 }, { | 613 }, { |
604 'benchmarkName': 'benchmark', | 614 'values': ['benchmark'], |
605 'canonicalUrl': '', | |
606 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 615 'guid': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
607 'label': '', | 616 'type': 'GenericSet' |
608 'legacyTIRLabel': '', | |
609 'storyDisplayName': 'story', | |
610 'type': 'TelemetryInfo' | |
611 }, { | 617 }, { |
612 'values': [424242], | 618 'values': [424242], |
613 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', | 619 'guid': '25f0a111-9bb4-4cea-b0c1-af2609623160', |
614 'type': 'GenericSet' | 620 'type': 'GenericSet' |
615 }, { | 621 }, { |
616 'buildNumber': 0, | 622 'values': ['master'], |
617 'buildbotMasterName': '', | |
618 'buildbotName': 'buildbotmaster1', | |
619 'displayBotName': 'bot', | |
620 'displayMasterName': 'master', | |
621 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 623 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
622 'logUri': '', | 624 'type': 'GenericSet' |
623 'type': 'BuildbotInfo' | 625 }, { |
| 626 'values': ['bot'], |
| 627 'guid': '53fb5448-9f8d-407a-8891-e7233fe1740f', |
| 628 'type': 'GenericSet' |
624 }, { | 629 }, { |
625 'binBoundaries': [1, [1, 1000, 20]], | 630 'binBoundaries': [1, [1, 1000, 20]], |
626 'diagnostics': { | 631 'diagnostics': { |
627 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 632 reserved_infos.MASTERS.name: |
| 633 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 634 reserved_infos.BOTS.name: |
| 635 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
628 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 636 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
629 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 637 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
630 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 638 reserved_infos.BENCHMARKS.name: |
| 639 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
631 reserved_infos.OWNERS.name: | 640 reserved_infos.OWNERS.name: |
632 'cabb59fe-4bcf-4512-881c-d038c7a80635' | 641 'cabb59fe-4bcf-4512-881c-d038c7a80635' |
633 }, | 642 }, |
634 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', | 643 'guid': '4989617a-14d6-4f80-8f75-dafda2ff13b0', |
635 'name': 'foo', | 644 'name': 'foo', |
636 'unit': 'count' | 645 'unit': 'count' |
637 }, { | 646 }, { |
638 'binBoundaries': [1, [1, 1000, 20]], | 647 'binBoundaries': [1, [1, 1000, 20]], |
639 'diagnostics': { | 648 'diagnostics': { |
640 'buildbot': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 649 reserved_infos.MASTERS.name: |
| 650 'e9c2891d-2b04-413f-8cf4-099827e67626', |
| 651 reserved_infos.BOTS.name: |
| 652 '53fb5448-9f8d-407a-8891-e7233fe1740f', |
641 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: | 653 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name: |
642 '25f0a111-9bb4-4cea-b0c1-af2609623160', | 654 '25f0a111-9bb4-4cea-b0c1-af2609623160', |
643 'telemetry': '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', | 655 reserved_infos.BENCHMARKS.name: |
| 656 '0bc1021b-8107-4db7-bc8c-49d7cf53c5ae', |
644 reserved_infos.OWNERS.name: | 657 reserved_infos.OWNERS.name: |
645 '7c5bd92f-4146-411b-9192-248ffc1be92c' | 658 '7c5bd92f-4146-411b-9192-248ffc1be92c' |
646 }, | 659 }, |
647 'guid': 'bda61ae3-0178-43f8-8aec-3ab78b9a2e18', | 660 'guid': 'bda61ae3-0178-43f8-8aec-3ab78b9a2e18', |
648 'name': 'foo', | 661 'name': 'foo', |
649 'unit': 'count' | 662 'unit': 'count' |
650 }]) | 663 }]) |
651 | 664 |
652 self.assertRaises( | 665 self.assertRaises( |
653 Exception, | 666 Exception, |
654 self.testapp.post, '/add_histograms', {'data': data}) | 667 self.testapp.post, '/add_histograms', {'data': data}) |
655 | 668 |
656 def testFindHistogramLevelSparseDiagnostics(self): | 669 def testFindHistogramLevelSparseDiagnostics(self): |
657 hist = histogram_module.Histogram('hist', 'count') | 670 hist = histogram_module.Histogram('hist', 'count') |
658 histograms = histogram_set.HistogramSet([hist]) | 671 histograms = histogram_set.HistogramSet([hist]) |
659 histograms.AddSharedDiagnostic('foo', histogram_module.GenericSet(['bar'])) | 672 histograms.AddSharedDiagnostic('foo', histogram_module.GenericSet(['bar'])) |
660 histograms.AddSharedDiagnostic( | 673 histograms.AddSharedDiagnostic( |
661 'telemetry', histogram_module.TelemetryInfo()) | 674 reserved_infos.GPUS.name, |
| 675 histogram_module.GenericSet([])) |
662 diagnostics = add_histograms.FindHistogramLevelSparseDiagnostics( | 676 diagnostics = add_histograms.FindHistogramLevelSparseDiagnostics( |
663 hist.guid, histograms) | 677 hist.guid, histograms) |
664 | 678 |
665 self.assertEqual(1, len(diagnostics)) | 679 self.assertEqual(1, len(diagnostics)) |
666 self.assertIsInstance(diagnostics[0], histogram_module.TelemetryInfo) | 680 self.assertIsInstance(diagnostics[0], histogram_module.GenericSet) |
667 | 681 |
668 def testComputeTestPathWithStory(self): | 682 def testComputeTestPathWithStory(self): |
669 hist = histogram_module.Histogram('hist', 'count') | 683 hist = histogram_module.Histogram('hist', 'count') |
670 histograms = histogram_set.HistogramSet([hist]) | 684 histograms = histogram_set.HistogramSet([hist]) |
671 telemetry_info = histogram_module.TelemetryInfo() | 685 histograms.AddSharedDiagnostic( |
672 telemetry_info.AddInfo({ | 686 reserved_infos.MASTERS.name, |
673 'storyDisplayName': 'story', | 687 histogram_module.GenericSet(['master'])) |
674 'benchmarkName': 'benchmark' | 688 histograms.AddSharedDiagnostic( |
675 }) | 689 reserved_infos.BOTS.name, |
676 histograms.AddSharedDiagnostic('telemetry', telemetry_info) | 690 histogram_module.GenericSet(['bot'])) |
677 buildbot_info = histogram_module.BuildbotInfo({ | 691 histograms.AddSharedDiagnostic( |
678 'displayMasterName': 'master', | 692 reserved_infos.BENCHMARKS.name, |
679 'displayBotName': 'bot' | 693 histogram_module.GenericSet(['benchmark'])) |
680 }) | 694 histograms.AddSharedDiagnostic( |
681 histograms.AddSharedDiagnostic('buildbot', buildbot_info) | 695 reserved_infos.STORIES.name, |
| 696 histogram_module.GenericSet(['story'])) |
682 hist = histograms.GetFirstHistogram() | 697 hist = histograms.GetFirstHistogram() |
683 test_path = add_histograms.ComputeTestPath(hist.guid, histograms) | 698 test_path = add_histograms.ComputeTestPath(hist.guid, histograms) |
684 self.assertEqual('master/bot/benchmark/hist/story', test_path) | 699 self.assertEqual('master/bot/benchmark/hist/story', test_path) |
685 | 700 |
686 def testComputeTestPathWithoutStory(self): | 701 def testComputeTestPathWithoutStory(self): |
687 hist = histogram_module.Histogram('hist', 'count') | 702 hist = histogram_module.Histogram('hist', 'count') |
688 histograms = histogram_set.HistogramSet([hist]) | 703 histograms = histogram_set.HistogramSet([hist]) |
689 telemetry_info = histogram_module.TelemetryInfo() | 704 histograms.AddSharedDiagnostic( |
690 telemetry_info.AddInfo({ | 705 reserved_infos.MASTERS.name, |
691 'benchmarkName': 'benchmark' | 706 histogram_module.GenericSet(['master'])) |
692 }) | 707 histograms.AddSharedDiagnostic( |
693 histograms.AddSharedDiagnostic('telemetry', telemetry_info) | 708 reserved_infos.BOTS.name, |
694 buildbot_info = histogram_module.BuildbotInfo({ | 709 histogram_module.GenericSet(['bot'])) |
695 'displayMasterName': 'master', | 710 histograms.AddSharedDiagnostic( |
696 'displayBotName': 'bot' | 711 reserved_infos.BENCHMARKS.name, |
697 }) | 712 histogram_module.GenericSet(['benchmark'])) |
698 histograms.AddSharedDiagnostic('buildbot', buildbot_info) | |
699 hist = histograms.GetFirstHistogram() | 713 hist = histograms.GetFirstHistogram() |
700 test_path = add_histograms.ComputeTestPath(hist.guid, histograms) | 714 test_path = add_histograms.ComputeTestPath(hist.guid, histograms) |
701 self.assertEqual('master/bot/benchmark/hist', test_path) | 715 self.assertEqual('master/bot/benchmark/hist', test_path) |
702 | 716 |
703 def testComputeRevision(self): | 717 def testComputeRevision(self): |
704 hist = histogram_module.Histogram('hist', 'count') | 718 hist = histogram_module.Histogram('hist', 'count') |
705 histograms = histogram_set.HistogramSet([hist]) | 719 histograms = histogram_set.HistogramSet([hist]) |
706 chromium_commit = histogram_module.GenericSet([424242]) | 720 chromium_commit = histogram_module.GenericSet([424242]) |
707 histograms.AddSharedDiagnostic( | 721 histograms.AddSharedDiagnostic( |
708 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name, chromium_commit) | 722 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name, chromium_commit) |
709 self.assertEqual(424242, add_histograms.ComputeRevision(histograms)) | 723 self.assertEqual(424242, add_histograms.ComputeRevision(histograms)) |
710 | 724 |
711 def testComputeRevision_RaisesOnError(self): | 725 def testComputeRevision_RaisesOnError(self): |
712 hist = histogram_module.Histogram('hist', 'count') | 726 hist = histogram_module.Histogram('hist', 'count') |
713 histograms = histogram_set.HistogramSet([hist]) | 727 histograms = histogram_set.HistogramSet([hist]) |
714 chromium_commit = histogram_module.GenericSet([424242, 0]) | 728 chromium_commit = histogram_module.GenericSet([424242, 0]) |
715 histograms.AddSharedDiagnostic( | 729 histograms.AddSharedDiagnostic( |
716 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name, chromium_commit) | 730 reserved_infos.CHROMIUM_COMMIT_POSITIONS.name, chromium_commit) |
717 with self.assertRaises(api_request_handler.BadRequestError): | 731 with self.assertRaises(api_request_handler.BadRequestError): |
718 add_histograms.ComputeRevision(histograms) | 732 add_histograms.ComputeRevision(histograms) |
719 | 733 |
720 def testSparseDiagnosticsAreNotInlined(self): | 734 def testSparseDiagnosticsAreNotInlined(self): |
721 hist = histogram_module.Histogram('hist', 'count') | 735 hist = histogram_module.Histogram('hist', 'count') |
722 histograms = histogram_set.HistogramSet([hist]) | 736 histograms = histogram_set.HistogramSet([hist]) |
723 histograms.AddSharedDiagnostic('foo', histogram_module.BuildbotInfo({ | 737 histograms.AddSharedDiagnostic( |
724 'displayMasterName': 'dmn', | 738 reserved_infos.BENCHMARKS.name, |
725 'displayBotName': 'dbn', | 739 histogram_module.GenericSet(['benchmark'])) |
726 'buildbotMasterName': 'bbmn', | |
727 'buildbotName': 'bbn', | |
728 'buildNumber': 42, | |
729 'logUri': 'uri', | |
730 })) | |
731 add_histograms.InlineDenseSharedDiagnostics(histograms) | 740 add_histograms.InlineDenseSharedDiagnostics(histograms) |
732 self.assertTrue(hist.diagnostics['foo'].has_guid) | 741 self.assertTrue(hist.diagnostics[reserved_infos.BENCHMARKS.name].has_guid) |
733 | 742 |
734 def testDeduplicateAndPut_Same(self): | 743 def testDeduplicateAndPut_Same(self): |
735 d = { | 744 d = { |
736 'buildNumber': 0, | 745 'values': ['master'], |
737 'buildbotMasterName': '', | |
738 'buildbotName': '', | |
739 'displayBotName': 'bot', | |
740 'displayMasterName': 'master', | |
741 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 746 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
742 'logUri': '', | 747 'type': 'GenericSet' |
743 'type': 'BuildbotInfo' | |
744 } | 748 } |
745 test_key = utils.TestKey('Chromium/win7/foo') | 749 test_key = utils.TestKey('Chromium/win7/foo') |
746 entity = histogram.SparseDiagnostic( | 750 entity = histogram.SparseDiagnostic( |
747 data=d, test=test_key, start_revision=1, | 751 data=d, test=test_key, start_revision=1, |
748 end_revision=sys.maxint, id='abc') | 752 end_revision=sys.maxint, id='abc') |
749 entity.put() | 753 entity.put() |
750 d2 = d.copy() | 754 d2 = d.copy() |
751 d2['guid'] = 'def' | 755 d2['guid'] = 'def' |
752 entity2 = histogram.SparseDiagnostic( | 756 entity2 = histogram.SparseDiagnostic( |
753 data=d2, test=test_key, | 757 data=d2, test=test_key, |
754 start_revision=2, end_revision=sys.maxint, id='def') | 758 start_revision=2, end_revision=sys.maxint, id='def') |
755 add_histograms.DeduplicateAndPut([entity2], test_key, 2) | 759 add_histograms.DeduplicateAndPut([entity2], test_key, 2) |
756 sparse = histogram.SparseDiagnostic.query().fetch() | 760 sparse = histogram.SparseDiagnostic.query().fetch() |
757 self.assertEqual(1, len(sparse)) | 761 self.assertEqual(2, len(sparse)) |
758 | 762 |
759 def testDeduplicateAndPut_Different(self): | 763 def testDeduplicateAndPut_Different(self): |
760 d = { | 764 d = { |
761 'buildNumber': 0, | 765 'values': ['master'], |
762 'buildbotMasterName': '', | |
763 'buildbotName': '', | |
764 'displayBotName': 'bot', | |
765 'displayMasterName': 'master', | |
766 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 766 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
767 'logUri': '', | 767 'type': 'GenericSet' |
768 'type': 'BuildbotInfo' | |
769 } | 768 } |
770 test_key = utils.TestKey('Chromium/win7/foo') | 769 test_key = utils.TestKey('Chromium/win7/foo') |
771 entity = histogram.SparseDiagnostic( | 770 entity = histogram.SparseDiagnostic( |
772 data=d, test=test_key, start_revision=1, | 771 data=d, test=test_key, start_revision=1, |
773 end_revision=sys.maxint, id='abc') | 772 end_revision=sys.maxint, id='abc') |
774 entity.put() | 773 entity.put() |
775 d2 = d.copy() | 774 d2 = d.copy() |
776 d2['guid'] = 'def' | 775 d2['guid'] = 'def' |
777 d2['displayBotName'] = 'mac' | 776 d2['displayBotName'] = 'mac' |
778 entity2 = histogram.SparseDiagnostic( | 777 entity2 = histogram.SparseDiagnostic( |
779 data=d2, test=test_key, | 778 data=d2, test=test_key, |
780 start_revision=1, end_revision=sys.maxint, id='def') | 779 start_revision=1, end_revision=sys.maxint, id='def') |
781 add_histograms.DeduplicateAndPut([entity2], test_key, 2) | 780 add_histograms.DeduplicateAndPut([entity2], test_key, 2) |
782 sparse = histogram.SparseDiagnostic.query().fetch() | 781 sparse = histogram.SparseDiagnostic.query().fetch() |
783 self.assertEqual(2, len(sparse)) | 782 self.assertEqual(2, len(sparse)) |
784 | 783 |
785 def testDeduplicateAndPut_New(self): | 784 def testDeduplicateAndPut_New(self): |
786 d = { | 785 d = { |
787 'buildNumber': 0, | 786 'values': ['master'], |
788 'buildbotMasterName': '', | |
789 'buildbotName': '', | |
790 'displayBotName': 'bot', | |
791 'displayMasterName': 'master', | |
792 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', | 787 'guid': 'e9c2891d-2b04-413f-8cf4-099827e67626', |
793 'logUri': '', | 788 'type': 'GenericSet' |
794 'type': 'BuildbotInfo' | |
795 } | 789 } |
796 test_key = utils.TestKey('Chromium/win7/foo') | 790 test_key = utils.TestKey('Chromium/win7/foo') |
797 entity = histogram.SparseDiagnostic( | 791 entity = histogram.SparseDiagnostic( |
798 data=d, test=test_key, start_revision=1, | 792 data=d, test=test_key, start_revision=1, |
799 end_revision=sys.maxint, id='abc') | 793 end_revision=sys.maxint, id='abc') |
800 entity.put() | 794 entity.put() |
801 add_histograms.DeduplicateAndPut([entity], test_key, 1) | 795 add_histograms.DeduplicateAndPut([entity], test_key, 1) |
802 sparse = histogram.SparseDiagnostic.query().fetch() | 796 sparse = histogram.SparseDiagnostic.query().fetch() |
803 self.assertEqual(1, len(sparse)) | 797 self.assertEqual(1, len(sparse)) |
OLD | NEW |