root
/
spring-framework
mirror of https://github.com/spring-projects/spring-framework.git
1
0
Fork 0
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Polishing in tests of ThreadLocalAccessor implementations 

See gh-32296
This commit is contained in:
rstoyanchev 2024-05-10 16:49:11 +01:00
parent 5cb4985234
commit 3ada9a0c79
2 changed files with 70 additions and 73 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

73
spring-context/src/test/java/org/springframework/context/i18n/LocaleContextThreadLocalAccessorTests.java
View File

@ -17,72 +17,71 @@
package org.springframework.context.i18n;
import java.util.Locale;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import java.util.stream.Stream;
import io.micrometer.context.ContextRegistry;
import io.micrometer.context.ContextSnapshot;
import io.micrometer.context.ContextSnapshotFactory;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import org.springframework.lang.Nullable;
import static org.assertj.core.api.Assertions.assertThat;
/**
* Tests for {@link LocaleContextThreadLocalAccessor}.
*
* @author Tadaya Tsuyukubo
* @author Rossen Stoyanchev
*/
class LocaleContextThreadLocalAccessorTests {
private final ContextRegistry registry = new ContextRegistry()
.registerThreadLocalAccessor(new LocaleContextThreadLocalAccessor());
private final ContextRegistry registry =
new ContextRegistry().registerThreadLocalAccessor(new LocaleContextThreadLocalAccessor());
@AfterEach
void cleanUp() {
LocaleContextHolder.resetLocaleContext();
private static Stream<Arguments> propagation() {
LocaleContext previousContext = new SimpleLocaleContext(Locale.ENGLISH);
LocaleContext currentContext = new SimpleLocaleContext(Locale.ENGLISH);
return Stream.of(Arguments.of(null, currentContext), Arguments.of(previousContext, currentContext));
}
@ParameterizedTest
@MethodSource
@SuppressWarnings("try")
void propagation(@Nullable LocaleContext previous, LocaleContext current) throws Exception {
LocaleContextHolder.setLocaleContext(current);
ContextSnapshot snapshot = ContextSnapshotFactory.builder()
.contextRegistry(this.registry)
.clearMissing(true)
.build()
.captureAll();
@SuppressWarnings({ "try", "unused" })
void propagation(LocaleContext previousContext, LocaleContext currentContext) throws Exception {
AtomicReference<LocaleContext> previousHolder = new AtomicReference<>();
AtomicReference<LocaleContext> currentHolder = new AtomicReference<>();
CountDownLatch latch = new CountDownLatch(1);
new Thread(() -> {
LocaleContextHolder.setLocaleContext(previous);
ContextSnapshot snapshot = createContextSnapshotFor(currentContext);
AtomicReference<LocaleContext> contextInScope = new AtomicReference<>();
AtomicReference<LocaleContext> contextAfterScope = new AtomicReference<>();
Thread thread = new Thread(() -> {
LocaleContextHolder.setLocaleContext(previousContext);
try (ContextSnapshot.Scope scope = snapshot.setThreadLocals()) {
currentHolder.set(LocaleContextHolder.getLocaleContext());
contextInScope.set(LocaleContextHolder.getLocaleContext());
}
previousHolder.set(LocaleContextHolder.getLocaleContext());
latch.countDown();
}).start();
contextAfterScope.set(LocaleContextHolder.getLocaleContext());
});
latch.await(1, TimeUnit.SECONDS);
assertThat(previousHolder).hasValueSatisfying(value -> assertThat(value).isSameAs(previous));
assertThat(currentHolder).hasValueSatisfying(value -> assertThat(value).isSameAs(current));
thread.start();
thread.join(1000);
assertThat(contextAfterScope).hasValueSatisfying(value -> assertThat(value).isSameAs(previousContext));
assertThat(contextInScope).hasValueSatisfying(value -> assertThat(value).isSameAs(currentContext));
}
private static Stream<Arguments> propagation() {
LocaleContext previous = new SimpleLocaleContext(Locale.ENGLISH);
LocaleContext current = new SimpleLocaleContext(Locale.ENGLISH);
return Stream.of(
Arguments.of(null, current),
Arguments.of(previous, current)
);
private ContextSnapshot createContextSnapshotFor(LocaleContext context) {
LocaleContextHolder.setLocaleContext(context);
try {
return ContextSnapshotFactory.builder()
.contextRegistry(this.registry).clearMissing(true).build()
.captureAll();
}
finally {
LocaleContextHolder.resetLocaleContext();
}
}
}

70
spring-web/src/test/java/org/springframework/web/context/request/RequestAttributesThreadLocalAccessorTests.java
View File

@ -16,8 +16,6 @@
package org.springframework.web.context.request;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import java.util.stream.Stream;
@ -25,13 +23,10 @@ import io.micrometer.context.ContextRegistry;
import io.micrometer.context.ContextSnapshot;
import io.micrometer.context.ContextSnapshot.Scope;
import io.micrometer.context.ContextSnapshotFactory;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import org.springframework.lang.Nullable;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock;
@ -39,52 +34,55 @@ import static org.mockito.Mockito.mock;
* Tests for {@link RequestAttributesThreadLocalAccessor}.
*
* @author Tadaya Tsuyukubo
* @author Rossen Stoyanchev
*/
class RequestAttributesThreadLocalAccessorTests {
private final ContextRegistry registry = new ContextRegistry()
.registerThreadLocalAccessor(new RequestAttributesThreadLocalAccessor());
private final ContextRegistry registry =
new ContextRegistry().registerThreadLocalAccessor(new RequestAttributesThreadLocalAccessor());
@AfterEach
void cleanUp() {
RequestContextHolder.resetRequestAttributes();
private static Stream<Arguments> propagation() {
RequestAttributes previous = mock(RequestAttributes.class);
RequestAttributes current = mock(RequestAttributes.class);
return Stream.of(Arguments.of(null, current), Arguments.of(previous, current));
}
@ParameterizedTest
@MethodSource
@SuppressWarnings({ "try", "unused" })
void propagation(@Nullable RequestAttributes previous, RequestAttributes current) throws Exception {
RequestContextHolder.setRequestAttributes(current);
ContextSnapshot snapshot = ContextSnapshotFactory.builder()
.contextRegistry(this.registry)
.clearMissing(true)
.build()
.captureAll();
void propagation(RequestAttributes previousRequest, RequestAttributes currentRequest) throws Exception {
AtomicReference<RequestAttributes> previousHolder = new AtomicReference<>();
AtomicReference<RequestAttributes> currentHolder = new AtomicReference<>();
CountDownLatch latch = new CountDownLatch(1);
new Thread(() -> {
RequestContextHolder.setRequestAttributes(previous);
ContextSnapshot snapshot = getSnapshotFor(currentRequest);
AtomicReference<RequestAttributes> requestInScope = new AtomicReference<>();
AtomicReference<RequestAttributes> requestAfterScope = new AtomicReference<>();
Thread thread = new Thread(() -> {
RequestContextHolder.setRequestAttributes(previousRequest);
try (Scope scope = snapshot.setThreadLocals()) {
currentHolder.set(RequestContextHolder.getRequestAttributes());
requestInScope.set(RequestContextHolder.getRequestAttributes());
}
previousHolder.set(RequestContextHolder.getRequestAttributes());
latch.countDown();
}).start();
requestAfterScope.set(RequestContextHolder.getRequestAttributes());
});
latch.await(1, TimeUnit.SECONDS);
assertThat(previousHolder).hasValueSatisfying(value -> assertThat(value).isSameAs(previous));
assertThat(currentHolder).hasValueSatisfying(value -> assertThat(value).isSameAs(current));
thread.start();
thread.join(1000);
assertThat(requestInScope).hasValueSatisfying(value -> assertThat(value).isSameAs(currentRequest));
assertThat(requestAfterScope).hasValueSatisfying(value -> assertThat(value).isSameAs(previousRequest));
}
private static Stream<Arguments> propagation() {
RequestAttributes previous = mock(RequestAttributes.class);
RequestAttributes current = mock(RequestAttributes.class);
return Stream.of(
Arguments.of(null, current),
Arguments.of(previous, current)
);
private ContextSnapshot getSnapshotFor(RequestAttributes request) {
RequestContextHolder.setRequestAttributes(request);
try {
return ContextSnapshotFactory.builder()
.contextRegistry(this.registry).clearMissing(true).build()
.captureAll();
}
finally {
RequestContextHolder.resetRequestAttributes();
}
}
}