@@ -85,7 +85,9 @@ public void loginPageThenDefaultLoginPageIsRendered() throws Exception {
85
85
String csrfAttributeName = HttpSessionCsrfTokenRepository .class .getName ().concat (".CSRF_TOKEN" );
86
86
// @formatter:off
87
87
this .mvc .perform (get ("/login" ).sessionAttr (csrfAttributeName , csrfToken ))
88
- .andExpect (content ().string ("<!DOCTYPE html>\n "
88
+ .andExpect ((result ) -> {
89
+ CsrfToken token = (CsrfToken ) result .getRequest ().getAttribute (CsrfToken .class .getName ());
90
+ assertThat (result .getResponse ().getContentAsString ()).isEqualTo ("<!DOCTYPE html>\n "
89
91
+ "<html lang=\" en\" >\n "
90
92
+ " <head>\n "
91
93
+ " <meta charset=\" utf-8\" >\n "
@@ -108,11 +110,12 @@ public void loginPageThenDefaultLoginPageIsRendered() throws Exception {
108
110
+ " <label for=\" password\" class=\" sr-only\" >Password</label>\n "
109
111
+ " <input type=\" password\" id=\" password\" name=\" password\" class=\" form-control\" placeholder=\" Password\" required>\n "
110
112
+ " </p>\n "
111
- + "<input name=\" " + csrfToken .getParameterName () + "\" type=\" hidden\" value=\" " + csrfToken .getToken () + "\" />\n "
113
+ + "<input name=\" " + token .getParameterName () + "\" type=\" hidden\" value=\" " + token .getToken () + "\" />\n "
112
114
+ " <button class=\" btn btn-lg btn-primary btn-block\" type=\" submit\" >Sign in</button>\n "
113
115
+ " </form>\n "
114
116
+ "</div>\n "
115
- + "</body></html>" ));
117
+ + "</body></html>" );
118
+ });
116
119
// @formatter:on
117
120
}
118
121
@@ -131,7 +134,9 @@ public void loginPageWhenErrorThenDefaultLoginPageWithError() throws Exception {
131
134
// @formatter:off
132
135
this .mvc .perform (get ("/login?error" ).session ((MockHttpSession ) mvcResult .getRequest ().getSession ())
133
136
.sessionAttr (csrfAttributeName , csrfToken ))
134
- .andExpect (content ().string ("<!DOCTYPE html>\n "
137
+ .andExpect ((result ) -> {
138
+ CsrfToken token = (CsrfToken ) result .getRequest ().getAttribute (CsrfToken .class .getName ());
139
+ assertThat (result .getResponse ().getContentAsString ()).isEqualTo ("<!DOCTYPE html>\n "
135
140
+ "<html lang=\" en\" >\n "
136
141
+ " <head>\n "
137
142
+ " <meta charset=\" utf-8\" >\n "
@@ -153,11 +158,12 @@ public void loginPageWhenErrorThenDefaultLoginPageWithError() throws Exception {
153
158
+ " <label for=\" password\" class=\" sr-only\" >Password</label>\n "
154
159
+ " <input type=\" password\" id=\" password\" name=\" password\" class=\" form-control\" placeholder=\" Password\" required>\n "
155
160
+ " </p>\n "
156
- + "<input name=\" " + csrfToken .getParameterName () + "\" type=\" hidden\" value=\" " + csrfToken .getToken () + "\" />\n "
161
+ + "<input name=\" " + token .getParameterName () + "\" type=\" hidden\" value=\" " + token .getToken () + "\" />\n "
157
162
+ " <button class=\" btn btn-lg btn-primary btn-block\" type=\" submit\" >Sign in</button>\n "
158
163
+ " </form>\n "
159
164
+ "</div>\n "
160
- + "</body></html>" ));
165
+ + "</body></html>" );
166
+ });
161
167
// @formatter:on
162
168
}
163
169
@@ -180,7 +186,9 @@ public void loginPageWhenLoggedOutThenDefaultLoginPageWithLogoutMessage() throws
180
186
String csrfAttributeName = HttpSessionCsrfTokenRepository .class .getName ().concat (".CSRF_TOKEN" );
181
187
// @formatter:off
182
188
this .mvc .perform (get ("/login?logout" ).sessionAttr (csrfAttributeName , csrfToken ))
183
- .andExpect (content ().string ("<!DOCTYPE html>\n "
189
+ .andExpect ((result ) -> {
190
+ CsrfToken token = (CsrfToken ) result .getRequest ().getAttribute (CsrfToken .class .getName ());
191
+ assertThat (result .getResponse ().getContentAsString ()).isEqualTo ("<!DOCTYPE html>\n "
184
192
+ "<html lang=\" en\" >\n "
185
193
+ " <head>\n "
186
194
+ " <meta charset=\" utf-8\" >\n "
@@ -203,11 +211,12 @@ public void loginPageWhenLoggedOutThenDefaultLoginPageWithLogoutMessage() throws
203
211
+ " <label for=\" password\" class=\" sr-only\" >Password</label>\n "
204
212
+ " <input type=\" password\" id=\" password\" name=\" password\" class=\" form-control\" placeholder=\" Password\" required>\n "
205
213
+ " </p>\n "
206
- + "<input name=\" " + csrfToken .getParameterName () + "\" type=\" hidden\" value=\" " + csrfToken .getToken () + "\" />\n "
214
+ + "<input name=\" " + token .getParameterName () + "\" type=\" hidden\" value=\" " + token .getToken () + "\" />\n "
207
215
+ " <button class=\" btn btn-lg btn-primary btn-block\" type=\" submit\" >Sign in</button>\n "
208
216
+ " </form>\n "
209
217
+ "</div>\n "
210
- + "</body></html>" ));
218
+ + "</body></html>" );
219
+ });
211
220
// @formatter:on
212
221
}
213
222
@@ -230,7 +239,9 @@ public void loginPageWhenRememberConfigureThenDefaultLoginPageWithRememberMeChec
230
239
String csrfAttributeName = HttpSessionCsrfTokenRepository .class .getName ().concat (".CSRF_TOKEN" );
231
240
// @formatter:off
232
241
this .mvc .perform (get ("/login" ).sessionAttr (csrfAttributeName , csrfToken ))
233
- .andExpect (content ().string ("<!DOCTYPE html>\n "
242
+ .andExpect ((result ) -> {
243
+ CsrfToken token = (CsrfToken ) result .getRequest ().getAttribute (CsrfToken .class .getName ());
244
+ assertThat (result .getResponse ().getContentAsString ()).isEqualTo ("<!DOCTYPE html>\n "
234
245
+ "<html lang=\" en\" >\n "
235
246
+ " <head>\n "
236
247
+ " <meta charset=\" utf-8\" >\n "
@@ -254,11 +265,12 @@ public void loginPageWhenRememberConfigureThenDefaultLoginPageWithRememberMeChec
254
265
+ " <input type=\" password\" id=\" password\" name=\" password\" class=\" form-control\" placeholder=\" Password\" required>\n "
255
266
+ " </p>\n "
256
267
+ "<p><input type='checkbox' name='remember-me'/> Remember me on this computer.</p>\n "
257
- + "<input name=\" " + csrfToken .getParameterName () + "\" type=\" hidden\" value=\" " + csrfToken .getToken () + "\" />\n "
268
+ + "<input name=\" " + token .getParameterName () + "\" type=\" hidden\" value=\" " + token .getToken () + "\" />\n "
258
269
+ " <button class=\" btn btn-lg btn-primary btn-block\" type=\" submit\" >Sign in</button>\n "
259
270
+ " </form>\n "
260
271
+ "</div>\n "
261
- + "</body></html>" ));
272
+ + "</body></html>" );
273
+ });
262
274
// @formatter:on
263
275
}
264
276
0 commit comments