Skip to content

Commit 4d65d62

Browse files
authored
Update TestExchange.java
Split assertRedirect up in 2 methods to allow custom messages
1 parent b330d27 commit 4d65d62

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

src/main/java/org/javawebstack/httpserver/test/TestExchange.java

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,23 +77,39 @@ public TestExchange assertCreated(String message) {
7777
Assertions.assertEquals(201, mockRes.getStatus(), message);
7878
return this;
7979
}
80+
8081
public TestExchange assertNoContent() {
8182
Assertions.assertEquals(204, mockRes.getStatus());
8283
return this;
8384
}
85+
8486
public TestExchange assertNoContent(String message) {
8587
Assertions.assertEquals(204, mockRes.getStatus(), message);
8688
return this;
8789
}
90+
8891
public TestExchange assertRedirect() {
8992
Assertions.assertTrue(isRedirect());
9093
return this;
9194
}
92-
public TestExchange assertRedirect(String url) {
93-
Assertions.assertTrue(isRedirect());
95+
96+
public TestExchange assertRedirect(String message) {
97+
Assertions.assertTrue(isRedirect(), message);
98+
return this;
99+
}
100+
101+
public TestExchange assertRedirectTo(String url) {
102+
assertRedirect();
94103
Assertions.assertEquals(url, mockRes.getHeader("Location"));
95104
return this;
96105
}
106+
107+
public TestExchange assertRedirectTo(String url, String message) {
108+
assertRedirect(message);
109+
Assertions.assertEquals(url, mockRes.getHeader("Location"), message);
110+
return this;
111+
}
112+
97113
public TestExchange assertNotFound(){
98114
Assertions.assertEquals(404, mockRes.getStatus());
99115
return this;

0 commit comments

Comments
 (0)