fastify
diff --git a/‎lib/util/resolve-schema-reference.js
Lines changed: 1 addition & 1 deletion b/‎lib/util/resolve-schema-reference.js
Lines changed: 1 addition & 1 deletion
diff --git a/‎test/spec/openapi/schema.js
Lines changed: 57 additions & 0 deletions b/‎test/spec/openapi/schema.js
Lines changed: 57 additions & 0 deletions
@@ -10,7 +10,7 @@ function resolveSchemaReference (rawSchema, ref) {
     return undefined
   }
 
-  return resolvedReference.definitions[schemaId]
+  return resolvedReference.definitions?.[schemaId]
 }
 
 module.exports = {
 
@@ -1022,6 +1022,63 @@ test('add default properties for url params when missing schema.params', async t
   })
 })
 
+test('support custom transforms which returns $ref in the response', async t => {
+  const customObject = {}
+  const opt = {
+    schema: {
+      response: {
+        200: customObject
+      }
+    }
+  }
+
+  const fastify = Fastify()
+  await fastify.register(fastifySwagger, {
+    openapi: true,
+    transform: ({ schema, ...rest }) => {
+      schema.response['200'] = {
+        $ref: '#/components/schemas/CustomObject'
+      }
+      return {
+        schema,
+        ...rest
+      }
+    },
+    transformObject: ({ openapiObject }) => {
+      openapiObject.components.schemas.CustomObject = {
+        type: 'object',
+        properties: {
+          hello: {
+            type: 'string'
+          }
+        }
+      }
+      return openapiObject
+    }
+  })
+  fastify.post('/', opt, () => { })
+  await fastify.ready()
+
+  const swaggerObject = fastify.swagger()
+
+  const swaggerPath = swaggerObject.paths['/'].post
+  t.has(swaggerPath.responses['200'].content['application/json'].schema, {
+    $ref: '#/components/schemas/CustomObject'
+  })
+
+  // validate seems to mutate the swaggerPath object
+  const api = await Swagger.validate(swaggerObject)
+  const definedPath = api.paths['/'].post
+  t.same(definedPath.responses['200'].content['application/json'].schema, {
+    type: 'object',
+    properties: {
+      hello: {
+        type: 'string'
+      }
+    }
+  })
+})
+
 test('avoid overwriti
42C1
ng params when schema.params is provided', async t => {
   const opt = {
     schema: {
Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ function resolveSchemaReference (rawSchema, ref) {`
`10`	`10`	`return undefined`
`11`	`11`	`}`
`12`	`12`
`13`		`- return resolvedReference.definitions[schemaId]`
	`13`	`+ return resolvedReference.definitions?.[schemaId]`
`14`	`14`	`}`
`15`	`15`
`16`	`16`	`module.exports = {`