def test_json(self): input_dict = dict(servers=dict(a=(2, 3))) expected_json = '{"servers":{"a":[2,3]}}' serializer = wsgi.JSONDictSerializer() result = serializer.serialize(input_dict) result = result.replace('\n', '').replace(' ', '') self.assertEqual(result, expected_json)
def test_json_with_unicode(self): input_dict = dict(servers=dict(a=(2, u'\u7f51\u7edc'))) expected_json = '{"servers":{"a":[2,"\\u7f51\\u7edc"]}}' serializer = wsgi.JSONDictSerializer() result = serializer.serialize(input_dict) result = result.replace('\n', '').replace(' ', '') self.assertEqual(result, expected_json)
def create_resource(version, controller_dict): """ Generic function for creating a wsgi resource The function takes as input: - desired version - controller and metadata dictionary e.g.: {'1.0': [ctrl_v10, meta_v10, xml_ns], '1.1': [ctrl_v11, meta_v11, xml_ns]} """ # the first element of the iterable is expected to be the controller controller = controller_dict[version][0] # the second element should be the metadata metadata = controller_dict[version][1] # and the third element the xml namespace xmlns = controller_dict[version][2] # and also the function for building the fault body fault_body_function = faults.fault_body_function(version) headers_serializers = { '1.0': HeaderSerializer10(), '1.1': HeaderSerializer11() } xml_serializer = wsgi.XMLDictSerializer(metadata, xmlns) json_serializer = wsgi.JSONDictSerializer() xml_deserializer = wsgi.XMLDeserializer(metadata) json_deserializer = wsgi.JSONDeserializer() body_serializers = { 'application/xml': xml_serializer, 'application/json': json_serializer, } body_deserializers = { 'application/xml': xml_deserializer, 'application/json': json_deserializer, } serializer = wsgi.ResponseSerializer(body_serializers, headers_serializers[version]) deserializer = wsgi.RequestDeserializer(body_deserializers) return wsgi.Resource(controller, fault_body_function, deserializer, serializer)
def Resource(controller, faults=None, deserializers=None, serializers=None): """Represents an API entity resource and the associated serialization and deserialization logic """ xml_deserializer = wsgi.XMLDeserializer(attributes.get_attr_metadata()) default_deserializers = {'application/xml': xml_deserializer, 'application/json': wsgi.JSONDeserializer()} xml_serializer = wsgi.XMLDictSerializer(attributes.get_attr_metadata()) default_serializers = {'application/xml': xml_serializer, 'application/json': wsgi.JSONDictSerializer()} format_types = {'xml': 'application/xml', 'json': 'application/json'} action_status = dict(create=201, delete=204) default_deserializers.update(deserializers or {}) default_serializers.update(serializers or {}) deserializers = default_deserializers serializers = default_serializers faults = faults or {} @webob.dec.wsgify(RequestClass=Request) def resource(request): route_args = request.environ.get('wsgiorg.routing_args') if route_args: args = route_args[1].copy() else: args = {} # NOTE(jkoelker) by now the controller is already found, remove # it from the args if it is in the matchdict args.pop('controller', None) fmt = args.pop('format', None) action = args.pop('action', None) content_type = format_types.get(fmt, request.best_match_content_type()) deserializer = deserializers.get(content_type) serializer = serializers.get(content_type) try: if request.body: args['body'] = deserializer.deserialize(request.body)['body'] method = getattr(controller, action) result = method(request=request, **args) except (exceptions.QuantumException, netaddr.AddrFormatError) as e: LOG.exception(_('%s failed'), action) body = serializer.serialize({'QuantumError': e}) kwargs = {'body': body, 'content_type': content_type} for fault in faults: if isinstance(e, fault): raise faults[fault](**kwargs) raise webob.exc.HTTPInternalServerError(**kwargs) except webob.exc.HTTPException as e: LOG.exception(_('%s failed'), action) e.body = serializer.serialize({'QuantumError': e}) e.content_type = content_type raise except Exception as e: # NOTE(jkoelker) Everyting else is 500 LOG.exception(_('%s failed'), action) # Do not expose details of 500 error to clients. msg = _('Request Failed: internal server error while ' 'processing your request.') body = serializer.serialize({'QuantumError': msg}) kwargs = {'body': body, 'content_type': content_type} raise webob.exc.HTTPInternalServerError(**kwargs) status = action_status.get(action, 200) body = serializer.serialize(result) # NOTE(jkoelker) Comply with RFC2616 section 9.7 if status == 204: content_type = '' body = None return webob.Response(request=request, status=status, content_type=content_type, body=body) return resource