def _RetrieveOutputJson(isolate_hash, filename):
output_files = isolate_service.Retrieve(isolate_hash)['files']
if filename not in output_files:
raise ReadValueError("The test didn't produce %s." % filename)
output_json_isolate_hash = output_files[filename]['h']
return json.loads(isolate_service.Retrieve(output_json_isolate_hash))
def _Poll(self):
test_output = isolate_service.Retrieve(self._isolate_hash)
chartjson_isolate_hash = test_output['files']['chartjson-output.json'][
'h']
chartjson = json.loads(
isolate_service.Retrieve(chartjson_isolate_hash))
chart = chartjson['charts'][self._metric][self._test]
if chart['type'] == 'list_of_scalar_values':
result_values = tuple(chart['values'])
elif chart['type'] == 'histogram':
result_values = _ResultValuesFromHistogram(chart['buckets'])
elif chart['type'] == 'scalar':
result_values = (chart['value'], )
self._Complete(result_values=result_values)
def _Poll(self):
result_values = []
for isolate_hash in self._isolate_hashes:
output = isolate_service.Retrieve(isolate_hash)
graphjson_isolate_hash = output['files']['chartjson-output.json'][
'h']
graphjson = json.loads(
isolate_service.Retrieve(graphjson_isolate_hash))
result_value = float(
graphjson[self._chart]['traces'][self._trace][0])
result_values.append(result_value)
self._Complete(result_values=tuple(result_values))
def testRetrieveContent(self, request_json, _):
request_json.return_value = {
'content': base64.b64encode(zlib.compress('file contents'))
}
isolate_contents = isolate_service.Retrieve(_ISOLATED_HASH)
self.assertEqual(isolate_contents, 'file contents')
url = 'https://isolateserver.appspot.com/_ah/api/isolateservice/v1/retrieve'
body = {
'namespace': {
'namespace': 'default-gzip'
},
'digest': _ISOLATED_HASH,
}
request_json.assert_called_once_with(url, 'POST', body)
def testRetrieveUrl(self, request_json, request):
request_json.return_value = {
'url':
'https://isolateserver.storage.googleapis.com/default-gzip/' +
_FILE_HASH
}
request.return_value = zlib.compress('file contents')
file_contents = isolate_service.Retrieve(_FILE_HASH)
self.assertEqual(file_contents, 'file contents')
url = 'https://isolateserver.appspot.com/_ah/api/isolateservice/v1/retrieve'
body = {
'namespace': {
'namespace': 'default-gzip'
},
'digest': _FILE_HASH
}
request_json.assert_called_once_with(url, 'POST', body)
request.assert_called_once_with(
'https://isolateserver.storage.googleapis.com/default-gzip/' +
_FILE_HASH, 'GET')
def testRetrieveUnknownFormat(self, request_json, _):
request_json.return_value = {}
with self.assertRaises(NotImplementedError):
isolate_service.Retrieve('digest')