@@ -112,7 +112,7 @@ void saveTokenSecure() {
112112 @ Test
113113 void saveTokenSecureFlagTrue () {
114114 this .request .setSecure (false );
115- this .repository .setSecure ( Boolean .TRUE );
115+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .TRUE ) );
116116 CsrfToken token = this .repository .generateToken (this .request );
117117 this .repository .saveToken (token , this .request , this .response );
118118 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -132,7 +132,7 @@ void saveTokenSecureFlagTrueUsingCustomizer() {
132132 @ Test
133133 void saveTokenSecureFlagFalse () {
134134 this .request .setSecure (true );
135- this .repository .setSecure ( Boolean .FALSE );
135+ this .repository .setCookieCustomizer (( cookie )-> cookie . secure ( Boolean .FALSE ) );
136136 CsrfToken token = this .repository .generateToken (this .request );
137137 this .repository .saveToken (token , this .request , this .response );
138138 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -163,7 +163,7 @@ void saveTokenNull() {
163163
164164 @ Test
165165 void saveTokenHttpOnlyTrue () {
166- this .repository .setCookieHttpOnly ( true );
166+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( true ) );
167167 CsrfToken token = this .repository .generateToken (this .request );
168168 this .repository .saveToken (token , this .request , this .response );
169169 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -181,7 +181,7 @@ void saveTokenHttpOnlyTrueUsingCustomizer() {
181181
182182 @ Test
183183 void saveTokenHttpOnlyFalse () {
184- this .repository .setCookieHttpOnly ( false );
184+ this .repository .setCookieCustomizer (( cookie ) -> cookie . httpOnly ( false ) );
185185 CsrfToken token = this .repository .generateToken (this .request );
186186 this .repository .saveToken (token , this .request , this .response );
187187 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -239,7 +239,7 @@ void saveTokenNullCustomPath() {
239239 @ Test
240240 void saveTokenWithCookieDomain () {
241241 String domainName = "example.com" ;
242- this .repository .setCookieDomain ( domainName );
242+ this .repository .setCookieCustomizer (( cookie ) -> cookie . domain ( domainName ) );
243243 CsrfToken token = this .repository .generateToken (this .request );
244244 this .repository .saveToken (token , this .request , this .response );
245245 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -259,7 +259,7 @@ void saveTokenWithCookieDomainUsingCustomizer() {
259259 @ Test
260260 void saveTokenWithCookieMaxAge () {
261261 int maxAge = 1200 ;
262- this .repository .setCookieMaxAge ( maxAge );
262+ this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( maxAge ) );
263263 CsrfToken token = this .repository .generateToken (this .request );
264264 this .repository .saveToken (token , this .request , this .response );
265265 Cookie tokenCookie = this .response .getCookie (CookieCsrfTokenRepository .DEFAULT_CSRF_COOKIE_NAME );
@@ -506,7 +506,7 @@ void setHeaderNameNullIllegalArgumentException() {
506506
507507 @ Test
508508 void setCookieMaxAgeZeroIllegalArgumentException () {
509- assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieMaxAge ( 0 ));
509+ assertThatIllegalArgumentException ().isThrownBy (() -> this .repository .setCookieCustomizer (( cookie ) -> cookie . maxAge ( 0 ) ));
510510 }
511511
512512}
0 commit comments