New tests for writing responses directly

fabiob · fabiob · commit 77f71b28e28f · 2017-09-23T16:58:02.000-03:00
diff --git a/test/functional/action-params.spec.ts b/test/functional/action-params.spec.ts
@@ -6,6 +6,7 @@ import {assertRequest} from "./test-utils";
 import {User} from "../fakes/global-options/User";
 import {Controller} from "../../src/decorator/Controller";
 import {Get} from "../../src/decorator/Get";
+import {Ctx} from "../../src/decorator/Ctx";
 import {Req} from "../../src/decorator/Req";
 import {Res} from "../../src/decorator/Res";
 import {Param} from "../../src/decorator/Param";
@@ -88,6 +89,24 @@ describe("action parameters", () => {
                 return "<html><body>hello</body></html>";
             }
 
+            @Get("/users/direct")
+            getUsersDirectExpress(@Res() response: any): any {
+                if (typeof response.send === "function")
+                    response.status(201).contentType("custom/x-sample").send("hi, I was written directly to the response");
+                else {
+                    response.status = 201;
+                    response.type = "custom/x-sample";
+                    response.body = "hi, I was written directly to the response";
+                }
+            }
+
+            @Get("/users/direct/ctx")
+            getUsersDirectKoa(@Ctx() ctx: any): any {
+                ctx.response.status = 201;
+                ctx.response.type = "custom/x-sample";
+                ctx.response.body = "hi, I was written directly to the response using Koa Ctx";
+            }
+
             @Get("/users/:userId")
             getUser(@Param("userId") userId: number) {
                 paramUserId = userId;
@@ -346,6 +365,22 @@ describe("action parameters", () => {
         });
     });
 
+    describe("writing directly to the response using @Res should work", () => {
+        assertRequest([3001, 3002], "get", "users/direct", response => {
+            expect(response).to.be.status(201);
+            expect(response.body).to.be.equal("hi, I was written directly to the response");
+            expect(response).to.have.header("content-type", "custom/x-sample; charset=utf-8");
+        });
+    });
+
+    describe("writing directly to the response using @Ctx should work", () => {
+        assertRequest([3002], "get", "users/direct/ctx", response => {
+            expect(response).to.be.status(201);
+            expect(response.body).to.be.equal("hi, I was written directly to the response using Koa Ctx");
+            expect(response).to.have.header("content-type", "custom/x-sample; charset=utf-8");
+        });
+    });
+
     describe("@Param should give a param from route", () => {
         assertRequest([3001, 3002], "get", "users/1", response => {
             expect(paramUserId).to.be.equal(1);
