Cwikius-Spring
/
spring-security
mirror of https://github.com/spring-projects/spring-security.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polish CsrfWebFilterTests 

Issue gh-9113
This commit is contained in:
Josh Cummings 2021-06-04 16:37:25 -06:00
parent 0c8b6df82a
commit ca76c54471
No known key found for this signature in database
GPG Key ID: 49EF60DD7FF83443
1 changed files with 7 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
web/src/test/java/org/springframework/security/web/server/csrf/CsrfWebFilterTests.java
View File

@ -16,9 +16,6 @@
package org.springframework.security.web.server.csrf;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicReference;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
@ -233,21 +230,16 @@ public class CsrfWebFilterTests {
// gh-9113
@Test
public void filterWhenSubscribingCsrfTokenMultipleTimesThenGenerateOnlyOnce() {
PublisherProbe<CsrfToken> chainResult = PublisherProbe.empty();
this.csrfFilter.setCsrfTokenRepository(this.repository);
given(this.repository.loadToken(any())).willReturn(Mono.empty());
AtomicInteger count = new AtomicInteger();
given(this.repository.generateToken(any())).willReturn(Mono.fromCallable(() -> {
count.incrementAndGet();
return this.token;
}));
given(this.repository.saveToken(any(), any())).willReturn(Mono.empty());
AtomicReference<Mono<CsrfToken>> tokenFromExchange = new AtomicReference<>();
given(this.chain.filter(any())).willReturn(
Mono.fromRunnable(() -> tokenFromExchange.set(this.get.getAttribute(CsrfToken.class.getName()))));
given(this.repository.generateToken(any())).willReturn(chainResult.mono());
given(this.chain.filter(any())).willReturn(Mono.empty());
this.csrfFilter.filter(this.get, this.chain).block();
tokenFromExchange.get().block();
tokenFromExchange.get().block();
assertThat(count).hasValue(1);
Mono<CsrfToken> result = this.get.getAttribute(CsrfToken.class.getName());
result.block();
result.block();
assertThat(chainResult.subscribeCount()).isEqualTo(1);
}
@RestController