change undo behavior

Author: HantingZhang2

features/step_definitions/request_steps.ts

@@ -143,7 +143,8 @@ When("the request is sent", async function (this: World) {
           this.apiVersion,
           undoAction,
           this.operationId,
-          this.response
+          this.response,
+          this.opts,
         )
       );
     }

features/support/given.ts

@@ -121,7 +121,7 @@ for (const apiVersion of Versions) {
       // register undo method
       if (undoAction.undo.type == "unsafe") {
         this.undo.push(
-          buildUndoFor(apiVersion, undoAction, operationName, result)
+          buildUndoFor(apiVersion, undoAction, operationName, result, opts)
         );
       }
 

features/support/undo.ts

@@ -10,6 +10,7 @@ logger.setLevel(process.env.DEBUG ? logger.levels.DEBUG : logger.levels.INFO);
 
 interface iOperationParameter {
   name: string;
+  origin: string;
   source: string;
   template: string;
 }
@@ -42,7 +43,8 @@ function buildUndoFor(
   apiVersion: string,
   operationUndo: iUndoAction,
   operationOrig: string,
-  response: any
+  response: any,
+  request: any
 ): { (): void } {
   return async function () {
     var apiName = operationUndo.tag.replace(/\s/g, "");
@@ -85,10 +87,18 @@ function buildUndoFor(
     // perform operation
     const opts: { [key: string]: any } = {};
     for (const p of operationUndo.undo.parameters) {
+      var dataSource: { [key: string]: any; };
+      if (p.origin == undefined) {
+         dataSource = response;
+      } else if (p.origin === "request") {
+         dataSource =request.body;
+      } else {
+        dataSource = response;
+      }
       if (p.source !== undefined) {
-        opts[p.name.toAttributeName()] = pathLookup(response, p.source);
+        opts[p.name.toAttributeName()] = pathLookup(dataSource, p.source);
       } else if (p.template !== undefined) {
-        const data = JSON.parse(p.template.templated(response));
+        const data = JSON.parse(p.template.templated(dataSource));
         const param: { [key: string]: any } = {};
         for (const [key, value] of Object.entries(data)) {
           param[key.toAttributeName()] = value;