def _verify_full_response(self, request, expected_response):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
self.assertProtoEquals(expected_response, response)
def _verify_handler(self, request, expected_session_group_names,
expected_total_size):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
self.assertEqual(expected_session_group_names,
[sg.name for sg in response.session_groups])
self.assertEqual(expected_total_size, response.total_size)
def _run_handler(self, request):
request_proto = api_pb2.ListSessionGroupsRequest()
text_format.Merge(request, request_proto)
handler = list_session_groups.Handler(
backend_context.Context(self._mock_tb_context), request_proto)
response = handler.run()
# Sort the metric values repeated field in each session group to
# canonicalize the response.
for group in response.session_groups:
group.metric_values.sort(key=operator.attrgetter("name.tag"))
return response
def list_session_groups_route(self, request):
try:
_verify_request_is_post(request, '/session_groups')
request_proto = json_format.Parse(request.data,
api_pb2.ListSessionGroupsRequest())
return http_util.Respond(
request,
json_format.MessageToJson(
list_session_groups.Handler(self._context, request_proto).run(),
including_default_value_fields=True,
),
'application/json')
except error.HParamsError as e:
logger.error('HParams error: %s' % e)
raise werkzeug.exceptions.BadRequest(description=str(e))
def list_session_groups_route(self, request):
try:
# args.get() returns the request unquoted.
request_proto = request.args.get('request')
if request_proto is None:
raise error.HParamsError(
'/session_groups must have a \'request\' arg.')
request_proto = json_format.Parse(
request_proto, api_pb2.ListSessionGroupsRequest())
return http_util.Respond(
request,
json_format.MessageToJson(
list_session_groups.Handler(self._context,
request_proto).run()),
'application/json')
except error.HParamsError as e:
raise werkzeug.exceptions.BadRequest(description=str(e))