add test case for manipulating data in outgoing event of wall

ujjwalguptaofficial · May 17, 2020 · df47eb8 · df47eb8
1 parent ba3d62b
commit df47eb8
Show file tree

Hide file tree

Showing 7 changed files with 28 additions and 6 deletions.
diff --git a/dist/fort.js b/dist/fort.js
diff --git a/dist/fort.js.map b/dist/fort.js.map
diff --git a/src/handlers/request_handler_helper.ts b/src/handlers/request_handler_helper.ts
@@ -19,7 +19,7 @@ export class RequestHandlerHelper {
 
     protected wallInstances: Wall[] = [];
 
-    protected controllerResult_: HttpResult | HttpFormatResult;
+    protected controllerResult_: HttpResult | HttpFormatResult = {} as any;
 
     protected runWallOutgoing() {
         const outgoingResults: Array<Promise<any>> = [];

diff --git a/tests/general/controllers/random_controller.ts b/tests/general/controllers/random_controller.ts
@@ -3,7 +3,7 @@ import * as path from "path";
 
 
 export class RandomController extends Controller {
-    
+
     @DefaultWorker()
     async format() {
         const result = {
@@ -62,4 +62,10 @@ export class RandomController extends Controller {
     async null() {
 
     }
+
+    @Worker()
+    @Route("/status-override")
+    async statusCodeOverrideTestUsingWall() {
+        return textResult("ok", 202);
+    }
 }
diff --git a/tests/general/test/random.js b/tests/general/test/random.js
@@ -117,4 +117,14 @@ describe("/random", () => {
             done();
         })
     })
+
+    it("/status-override using wall", (done) => {
+        request.get('/random/status-override').end((err, res) => {
+            expect(err).to.be.null;
+            expect(res).to.have.status(200);
+            expect(res).to.have.header('content-type', 'text/plain');
+            expect(res.text).to.be.equal("overrided");
+            done();
+        })
+    })
 })
diff --git a/tests/general/test/wall.js b/tests/general/test/wall.js
@@ -52,7 +52,7 @@ describe("wall test", () => {
             expect(res).to.have.header('custom-header-from-incoming-wall', '*');
             expect(res).to.have.header('injection-result', 'wall constructor onIncoming called on outgoing called');
             expect(res).to.have.header('wall-without-outgoing-wall', '*');
-            expect(res.body).haveOwnProperty('reqCount').equal(108);
+            expect(res.body).haveOwnProperty('reqCount').equal(109);
             done();
         })
     })

diff --git a/tests/general/walls/request_logger.ts b/tests/general/walls/request_logger.ts
@@ -1,4 +1,4 @@
-import { Wall, textResult, Assign } from "fortjs";
+import { Wall, textResult, HttpResult, Assign } from "fortjs";
 let reqCount = 0;
 export class RequestLogger extends Wall {
 
@@ -36,9 +36,14 @@ export class RequestLogger extends Wall {
         }
     }
 
-    async onOutgoing(result, @Assign('on outgoing called') value: string) {
+    async onOutgoing(result: HttpResult, @Assign('on outgoing called') value: string) {
         this.logger.log('executing request logger');
         this.response.setHeader('Custom-Header-From-Outgoing-Wall', '*');
         this.response.setHeader('injection-result', this.injectionValue + ` ${value}`);
+
+        if (result.statusCode === 202) {
+            result.statusCode = 200;
+            result.responseData = "overrided";
+        }
     }
 }