refactor(): Refactor Controller class visitor tests

nestjs · Feb 15, 2021 · cd75f5d · cd75f5d
1 parent 4564117
commit cd75f5d
Showing 1 changed file with 31 additions and 21 deletions.
diff --git a/test/plugin/controller-class-visitor.spec.ts b/test/plugin/controller-class-visitor.spec.ts
@@ -5,29 +5,39 @@ import {
   appControllerTextTranspiled
 } from './fixtures/app.controller';
 
+const compilerOptions: ts.CompilerOptions = {
+  module: ts.ModuleKind.CommonJS,
+  target: ts.ScriptTarget.ESNext,
+  newLine: ts.NewLineKind.LineFeed,
+  noEmitHelpers: true
+}
+
+const transpileModule = (filename, controllerText, compilerOptions, swaggerDocumentOptions = {}) => {
+  const fakeProgram = ts.createProgram([filename], compilerOptions);
+
+  return ts.transpileModule(controllerText, {
+    compilerOptions,
+    fileName: filename,
+    transformers: {
+      before: [
+        before(
+          {...swaggerDocumentOptions, introspectComments: true },
+          fakeProgram
+        )
+      ]
+    }
+  })
+}
+
 describe('Controller methods', () => {
-  it('should add response based on the return value', () => {
-    const options: ts.CompilerOptions = {
-      module: ts.ModuleKind.CommonJS,
-      target: ts.ScriptTarget.ESNext,
-      newLine: ts.NewLineKind.LineFeed,
-      noEmitHelpers: true
-    };
-    const filename = 'app.controller.ts';
-    const fakeProgram = ts.createProgram([filename], options);
+  it('Should generate summary property', () => {
+    const result = transpileModule(
+      'app.controller.ts',
+      appControllerText,
+      compilerOptions,
+      {controllerKeyOfComment: 'summary'}
+    );
 
-    const result = ts.transpileModule(appControllerText, {
-      compilerOptions: options,
-      fileName: filename,
-      transformers: {
-        before: [
-          before(
-            { controllerKeyOfComment: 'summary', introspectComments: true },
-            fakeProgram
-          )
-        ]
-      }
-    });
     expect(result.outputText).toEqual(appControllerTextTranspiled);
   });
 });