@@ -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