@@ -108,7 +108,7 @@ void saveTokenSecure() {
108
108
@ Test
109
109
void saveTokenSecureFlagTrue () {
110
110
this .request .setSecure (false );
111
- this .repository .setSecure ( Boolean .TRUE );
111
+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .TRUE ) );
112
112
CsrfToken token = this .repository .generateToken (this .request );
113
113
this .repository .saveToken (token , this .request , this .response );
114
114
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -128,7 +128,7 @@ void saveTokenSecureFlagTrueUsingCustomizer() {
128
128
@ Test
129
129
void saveTokenSecureFlagFalse () {
130
130
this .request .setSecure (true );
131
- this .repository .setSecure ( Boolean .FALSE );
131
+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .FALSE ) );
132
132
CsrfToken token = this .repository .generateToken (this .request );
133
133
this .repository .saveToken (token , this .request , this .response );
134
134
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -159,7 +159,7 @@ void saveTokenNull() {
159
159
160
160
@ Test
161
161
void saveTokenHttpOnlyTrue () {
162
- this .repository .setCookieHttpOnly ( true );
162
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( true ) );
163
163
CsrfToken token = this .repository .generateToken (this .request );
164
164
this .repository .saveToken (token , this .request , this .response );
165
165
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -177,7 +177,7 @@ void saveTokenHttpOnlyTrueUsingCustomizer() {
177
177
178
178
@ Test
179
179
void saveTokenHttpOnlyFalse () {
180
- this .repository .setCookieHttpOnly ( false );
180
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( false ) );
181
181
CsrfToken token = this .repository .generateToken (this .request );
182
182
this .repository .saveToken (token , this .request , this .response );
183
183
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -235,7 +235,7 @@ void saveTokenNullCustomPath() {
235
235
@ Test
236
236
void saveTokenWithCookieDomain () {
237
237
String domainName = "example.com" ;
238
- this .repository .setCookieDomain ( domainName );
238
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . domain ( domainName ) );
239
239
CsrfToken token = this .repository .generateToken (this .request );
240
240
this .repository .saveToken (token , this .request , this .response );
241
241
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -255,7 +255,7 @@ void saveTokenWithCookieDomainUsingCustomizer() {
255
255
@ Test
256
256
void saveTokenWithCookieMaxAge () {
257
257
int maxAge = 1200 ;
258
- this .repository .setCookieMaxAge ( maxAge );
258
+ this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( maxAge ) );
259
259
CsrfToken token = this .repository .generateToken (this .request );
260
260
this .repository .saveToken (token , this .request , this .response );
261
261
Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -464,7 +464,7 @@ void setHeaderNameNullIllegalArgumentException() {
464
464
465
465
@ Test
466
466
void setCookieMaxAgeZeroIllegalArgumentException () {
467
- assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieMaxAge ( 0 ));
467
+ assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( 0 ) ));
468
468
}
469
469
470
470
}
0 commit comments