Add response example

nestjs · Mar 11, 2024 · 713c48c · 713c48c
1 parent feca721
commit 713c48c
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 11 deletions.
diff --git a/lib/decorators/api-response.decorator.ts b/lib/decorators/api-response.decorator.ts
@@ -14,6 +14,7 @@ export interface ApiResponseMetadata
   type?: Type<unknown> | Function | [Function] | string;
   isArray?: boolean;
   description?: string;
+  example?: any;
 }
 
 export interface ApiResponseSchemaHost

diff --git a/lib/services/response-object-mapper.ts b/lib/services/response-object-mapper.ts
@@ -1,5 +1,5 @@
 import { omit } from 'lodash';
-import { ApiResponseSchemaHost } from '../decorators';
+import { ApiResponseMetadata, ApiResponseSchemaHost } from '../decorators';
 import { getSchemaPath } from '../utils';
 import { MimetypeContentWrapper } from './mimetype-content-wrapper';
 
@@ -19,7 +19,8 @@ export class ResponseObjectMapper {
           items: {
             $ref: getSchemaPath(name)
           }
-        }
+        },
+        ...(response.example ? { example: response.example } : {})
       })
     };
   }
@@ -30,20 +31,25 @@ export class ResponseObjectMapper {
       ...this.mimetypeContentWrapper.wrap(produces, {
         schema: {
           $ref: getSchemaPath(name)
-        }
+        },
+        ...(response.example ? { example: response.example } : {})
       })
     };
   }
 
-  wrapSchemaWithContent(response: ApiResponseSchemaHost, produces: string[]) {
-    if (!response.schema) {
+  wrapSchemaWithContent(
+    response: ApiResponseSchemaHost & ApiResponseMetadata,
+    produces: string[]
+  ) {
+    if (!response.schema && !response.example) {
       return response;
     }
     const content = this.mimetypeContentWrapper.wrap(produces, {
-      schema: response.schema
+      ...(response.schema ? { schema: response.schema } : {}),
+      ...(response.example ? { example: response.example } : {})
     });
     return {
-      ...omit(response, 'schema'),
+      ...omit(response, ['schema', 'example']),
       ...content
     };
   }

diff --git a/test/explorer/swagger-explorer.spec.ts b/test/explorer/swagger-explorer.spec.ts
@@ -97,7 +97,11 @@ describe('SwaggerExplorer', () => {
       @ApiOperation({ summary: 'Create foo' })
       @ApiCreatedResponse({
         type: Foo,
-        description: 'Newly created Foo object'
+        description: 'Newly created Foo object',
+        example: {
+          id: 'foo',
+          name: 'Foo'
+        }
       })
       create(
         @Body() createFoo: CreateFoo,
@@ -138,7 +142,11 @@ describe('SwaggerExplorer', () => {
       @Post('foos')
       @ApiCreatedResponse({
         type: Foo,
-        description: 'Newly created Foo object'
+        description: 'Newly created Foo object',
+        example: {
+          id: 'foo',
+          name: 'Foo'
+        }
       })
       create(
         @Body() createFoo: CreateFoo,
@@ -158,7 +166,11 @@ describe('SwaggerExplorer', () => {
       static [METADATA_FACTORY_NAME]() {
         return {
           create: {
-            summary: 'Create foo'
+            summary: 'Create foo',
+            example: {
+              id: 'foo',
+              name: 'Foo'
+            }
           },
           find: {
             summary: 'List all Foos',
@@ -301,7 +313,8 @@ describe('SwaggerExplorer', () => {
       ).toEqual({
         schema: {
           $ref: '#/components/schemas/Foo'
-        }
+        },
+        example: { id: 'foo', name: 'Foo' }
       });
 
       // GET