@@ -112,7 +112,7 @@ void saveTokenSecure() {
112
112
@ Test
113
113
void saveTokenSecureFlagTrue () {
114
114
this .request .setSecure (false );
115
- this .repository .setSecure ( Boolean .TRUE );
115
+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .TRUE ) );
116
116
CsrfToken token = this .repository .generateToken (this .request );
117
117
this .repository .saveToken (token , this .request , this .response );
118
118
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -132,7 +132,7 @@ void saveTokenSecureFlagTrueUsingCustomizer() {
132
132
@ Test
133
133
void saveTokenSecureFlagFalse () {
134
134
this .request .setSecure (true );
135
- this .repository .setSecure ( Boolean .FALSE );
135
+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .FALSE ) );
136
136
CsrfToken token = this .repository .generateToken (this .request );
137
137
this .repository .saveToken (token , this .request , this .response );
138
138
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -163,7 +163,7 @@ void saveTokenNull() {
163
163
164
164
@ Test
165
165
void saveTokenHttpOnlyTrue () {
166
- this .repository .setCookieHttpOnly ( true );
166
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( true ) );
167
167
CsrfToken token = this .repository .generateToken (this .request );
168
168
this .repository .saveToken (token , this .request , this .response );
169
169
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -181,7 +181,7 @@ void saveTokenHttpOnlyTrueUsingCustomizer() {
181
181
182
182
@ Test
183
183
void saveTokenHttpOnlyFalse () {
184
- this .repository .setCookieHttpOnly ( false );
184
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( false ) );
185
185
CsrfToken token = this .repository .generateToken (this .request );
186
186
this .repository .saveToken (token , this .request , this .response );
187
187
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -239,7 +239,7 @@ void saveTokenNullCustomPath() {
239
239
@ Test
240
240
void saveTokenWithCookieDomain () {
241
241
String domainName = "example.com" ;
242
- this .repository .setCookieDomain ( domainName );
242
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . domain ( domainName ) );
243
243
CsrfToken token = this .repository .generateToken (this .request );
244
244
this .repository .saveToken (token , this .request , this .response );
245
245
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -259,7 +259,7 @@ void saveTokenWithCookieDomainUsingCustomizer() {
259
259
@ Test
260
260
void saveTokenWithCookieMaxAge () {
261
261
int maxAge = 1200 ;
262
- this .repository .setCookieMaxAge ( maxAge );
262
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( maxAge ) );
263
263
CsrfToken token = this .repository .generateToken (this .request );
264
264
this .repository .saveToken (token , this .request , this .response );
265
265
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -506,7 +506,7 @@ void setHeaderNameNullIllegalArgumentException() {
506
506
507
507
@ Test
508
508
void setCookieMaxAgeZeroIllegalArgumentException () {
509
- assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieMaxAge ( 0 ));
509
+ assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( 0 ) ));
510
510
}
511
511
512
512
}
0 commit comments