feat(ui): Script > Post Request > add method to get response status c…

…ode (#264)
flawiddsouza · Sep 30, 2024 · b8ee500 · b8ee500
1 parent b51a080
commit b8ee500
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/packages/ui/src/components/SnippetDropdown.vue b/packages/ui/src/components/SnippetDropdown.vue
@@ -254,6 +254,11 @@ export default defineComponent ({
                 type: 'option',
                 label: 'Returns the response body as a parsed JSON object',
                 value: 'rf.response.getBodyJSON()'
+            },
+            {
+                type: 'option',
+                label: 'Returns the status code of the response',
+                value: 'rf.response.getStatusCode()'
             }
         ]
 

diff --git a/packages/ui/src/constants.ts b/packages/ui/src/constants.ts
@@ -225,6 +225,11 @@ export default {
                     label: 'rf.response.getBodyJSON()',
                     type: 'function',
                     info: 'Returns the response body as a parsed JSON object'
+                },
+                {
+                    label: 'rf.response.getStatusCode()',
+                    type: 'function',
+                    info: 'Returns the status code of the response',
                 }
             ],
         },

diff --git a/packages/ui/src/helpers.ts b/packages/ui/src/helpers.ts
@@ -1525,7 +1525,8 @@ export function scriptConversion(scriptToConvert: string, scriptType: 'postmanTo
         postmanToRestfox: {
             'pm.environment.set': 'rf.setEnvVar',
             'pm.environment.get': 'rf.getEnvVar',
-            'pm.response.json()': 'rf.response.getBodyJSON()'
+            'pm.response.json()': 'rf.response.getBodyJSON()',
+            'pm.response.code': 'rf.response.getStatusCode()'
         },
         restfoxToPostman: {
             'rf.setEnvVar': 'pm.environment.set',

diff --git a/packages/ui/src/plugin.ts b/packages/ui/src/plugin.ts
@@ -173,6 +173,7 @@ export async function createRequestContextForPlugin(cacheId: string, request: Co
 export function createResponseContextForPlugin(response: RequestFinalResponse, environment: any, setEnvironmentVariable: (name: string, value: string) => void, testResults: PluginTestResult[]): { expose: PluginExpose } {
     let bufferCopy = response.buffer.slice(0)
     const headers = response.headers
+    const statusCode = response.status
 
     const generalContextMethods = {
         ...generalContextMethodsBase,
@@ -216,6 +217,9 @@ export function createResponseContextForPlugin(response: RequestFinalResponse, e
                 const header = headers.find((header: RequestInitialResponseHeader) => header[0].toLowerCase() == headerName.toLowerCase())
                 return header ? header[1] : undefined
             },
+            getStatusCode() {
+                return statusCode
+            },
             // deprecated but won't be removed
             getEnvironmentVariable: generalContextMethods.getEnvVar,
             setEnvironmentVariable: generalContextMethods.setEnvVar,