-
-
Notifications
You must be signed in to change notification settings - Fork 281
/
setResponseHeaders.spec.ts
129 lines (105 loc) · 3.27 KB
/
setResponseHeaders.spec.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
import {PlatformTest} from "@tsed/common";
import {Get, JsonEntityStore, Redirect, Returns} from "@tsed/schema";
import {setResponseHeaders} from "./setResponseHeaders";
import {createServerlessContext} from "../../test/utils/createServerlessContext";
describe("setResponseHeaders", () => {
beforeEach(() => PlatformTest.create());
afterEach(() => PlatformTest.reset());
it("should set headers, status and contentType", async () => {
class Test {
@Get("/")
@Returns(200).Header("x-header", "test")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
jest.spyOn(ctx.response, "setHeaders");
jest.spyOn(ctx.response, "status");
// WHEN
await setResponseHeaders(ctx);
// THEN
expect(ctx.response.setHeaders).toHaveBeenCalledWith({"x-header": "test"});
expect(ctx.response.status).toHaveBeenCalledWith(200);
});
it("should not set headers automatically", async () => {
class Test {
@Get("/")
@Returns(200).Header("x-header")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
jest.spyOn(ctx.response, "setHeaders");
jest.spyOn(ctx.response, "status");
// WHEN
await setResponseHeaders(ctx);
// THEN
expect(ctx.response.setHeaders).toHaveBeenCalledWith({});
expect(ctx.response.status).toHaveBeenCalledWith(200);
});
it("should redirect with 301", async () => {
class Test {
@Get("/")
@Redirect(301, "/path")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
jest.spyOn(ctx.response, "redirect");
// WHEN
await setResponseHeaders(ctx as any);
// THEN
expect(ctx.response.redirect).toHaveBeenCalledWith(301, "/path");
});
it("should redirect with default", async () => {
class Test {
@Get("/")
@Redirect(302, "/path")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
jest.spyOn(ctx.response, "redirect");
// WHEN
await setResponseHeaders(ctx as any);
// THEN
expect(ctx.response.redirect).toHaveBeenCalledWith(302, "/path");
});
it("should redirect with assigned value", async () => {
class Test {
@Get("/")
@Redirect(302, "/path")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
jest.spyOn(ctx.response, "redirect");
ctx.response.setHeaders({Location: "/path/other"});
// WHEN
await setResponseHeaders(ctx as any);
// THEN
expect(ctx.response.redirect).toHaveBeenCalledWith(302, "/path/other");
});
it("should do nothing when headers is already sent", async () => {
class Test {
@Get("/")
@Returns(200).Header("x-header", "test")
test() {}
}
const ctx = createServerlessContext({
endpoint: JsonEntityStore.fromMethod(Test, "test")
});
await ctx.destroy();
jest.spyOn(ctx.response, "setHeaders");
jest.spyOn(ctx.response, "isDone");
// WHEN
await setResponseHeaders(ctx);
// THEN
return expect(ctx.response.isDone).toHaveBeenCalled();
});
});