test: asObject for getParamsAST

kubb-labs · Oct 18, 2023 · 2da4c7a · 2da4c7a
1 parent a435b1e
commit 2da4c7a
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 4 deletions.
diff --git a/packages/swagger/src/utils/getParams.test.ts b/packages/swagger/src/utils/getParams.test.ts
@@ -49,4 +49,25 @@ describe('getParams', () => {
       ).toString(),
     ).toBe('name: Pet["name"], age: Pet["age"]')
   })
+
+  test('if operation returns a string with typed parameters and as an object', () => {
+    expect(
+      getParams(
+        {
+          name: 'Pet',
+          schema: {
+            properties: {
+              name: {
+                type: 'string',
+              },
+              age: {
+                type: 'number',
+              },
+            },
+          },
+        },
+        { asObject: true },
+      ).toString(),
+    ).toBe('{ name, age }: Pet')
+  })
 })
diff --git a/packages/swagger/src/utils/getParams.ts b/packages/swagger/src/utils/getParams.ts
@@ -16,12 +16,12 @@ export function getASTParams(
   if (asObject) {
     return [
       {
-        name: `{ ${getASTParams(operationSchema, { typed: true })
+        name: `{ ${getASTParams(operationSchema)
           .map((item) => item.name)
           .join(', ')} }`,
         type: operationSchema?.name,
         enabled: !!operationSchema?.name,
-        required: !!operationSchema?.schema.required?.length,
+        required: true,
       },
     ]
   }
@@ -41,9 +41,9 @@ type GetParamsResult = {
 // TODO convert to class together with `createFunctionParams` and `getASTParams`
 export function getParams(
   operationSchema: OperationSchema | undefined,
-  { typed = false, override }: { typed?: boolean; override?: (data: FunctionParamsAST) => FunctionParamsAST } = {},
+  { typed = false, override, asObject = false }: { typed?: boolean; asObject?: boolean; override?: (data: FunctionParamsAST) => FunctionParamsAST } = {},
 ): GetParamsResult {
-  const ast = getASTParams(operationSchema, { typed, override })
+  const ast = getASTParams(operationSchema, { typed, override, asObject })
   const functionParams = new FunctionParams()
   functionParams.add(ast)