This repository was archived by the owner on Mar 3, 2026. It is now read-only.
forked from jooby-project/jooby
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDefaultErrHandlerTest.java
More file actions
124 lines (99 loc) · 4.35 KB
/
DefaultErrHandlerTest.java
File metadata and controls
124 lines (99 loc) · 4.35 KB
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
package org.jooby;
import com.google.common.collect.ImmutableList;
import com.typesafe.config.Config;
import static org.easymock.EasyMock.expect;
import org.jooby.test.MockUnit;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.util.Map;
@RunWith(PowerMockRunner.class)
@PrepareForTest({Err.DefHandler.class, LoggerFactory.class})
public class DefaultErrHandlerTest {
@SuppressWarnings({"unchecked"})
@Test
public void handleNoErrMessage() throws Exception {
Err ex = new Err(500);
StringWriter writer = new StringWriter();
ex.printStackTrace(new PrintWriter(writer));
String[] stacktrace = writer.toString().replace("\r", "").split("\\n");
new MockUnit(Request.class, Response.class, Route.class, Config.class, Env.class)
.expect(handleErr(ex,true))
.run(unit -> {
Request req = unit.get(Request.class);
Response rsp = unit.get(Response.class);
new Err.DefHandler().handle(req, rsp, ex);
}, unit -> {
Result result = unit.captured(Result.class).iterator().next();
View view = (View) result.ifGet(ImmutableList.of(MediaType.html)).get();
assertEquals("err", view.name());
checkErr(stacktrace, "Server Error(500)", (Map<String, Object>) view.model()
.get("err"));
Object hash = result.ifGet(MediaType.ALL).get();
assertEquals(4, ((Map<String, Object>) hash).size());
});
}
private MockUnit.Block handleErr(Throwable ex, boolean stacktrace) {
return unit -> {
Logger log = unit.mock(Logger.class);
log.error("execution of: {}{} resulted in exception\nRoute:\n{}\n\nStacktrace:", "GET",
"/path", "route", ex);
unit.mockStatic(LoggerFactory.class);
expect(LoggerFactory.getLogger(Err.class)).andReturn(log);
Route route = unit.get(Route.class);
expect(route.print(6)).andReturn("route");
Config conf = unit.get(Config.class);
expect(conf.getBoolean("err.stacktrace")).andReturn(stacktrace);
Env env = unit.get(Env.class);
expect(env.name()).andReturn("dev");
Request req = unit.get(Request.class);
expect(req.require(Config.class)).andReturn(conf);
expect(req.require(Env.class)).andReturn(env);
expect(req.path()).andReturn("/path");
expect(req.method()).andReturn("GET");
expect(req.route()).andReturn(route);
Response rsp = unit.get(Response.class);
rsp.send(unit.capture(Result.class));
};
}
@SuppressWarnings({"unchecked"})
@Test
public void handleWithErrMessage() throws Exception {
Err ex = new Err(500, "Something something dark");
StringWriter writer = new StringWriter();
ex.printStackTrace(new PrintWriter(writer));
String[] stacktrace = writer.toString().replace("\r", "").split("\\n");
new MockUnit(Request.class, Response.class, Route.class, Env.class, Config.class)
.expect(handleErr(ex, true))
.run(unit -> {
Request req = unit.get(Request.class);
Response rsp = unit.get(Response.class);
new Err.DefHandler().handle(req, rsp, ex);
},
unit -> {
Result result = unit.captured(Result.class).iterator().next();
View view = (View) result.ifGet(ImmutableList.of(MediaType.html)).get();
assertEquals("err", view.name());
checkErr(stacktrace, "Server Error(500): Something something dark",
(Map<String, Object>) view.model()
.get("err"));
Object hash = result.ifGet(MediaType.ALL).get();
assertEquals(4, ((Map<String, Object>) hash).size());
});
}
private void checkErr(final String[] stacktrace, final String message,
final Map<String, Object> err) {
assertEquals(message, err.remove("message"));
assertEquals("Server Error", err.remove("reason"));
assertEquals(500, err.remove("status"));
assertArrayEquals(stacktrace, (String[]) err.remove("stacktrace"));
assertEquals(err.toString(), 0, err.size());
}
}