|
@@ -44,6 +44,7 @@ public class ReactorContextTestExecutionListener
|
|
extends DelegatingTestExecutionListener {
|
|
extends DelegatingTestExecutionListener {
|
|
|
|
|
|
private static final String HOOKS_CLASS_NAME = "reactor.core.publisher.Hooks";
|
|
private static final String HOOKS_CLASS_NAME = "reactor.core.publisher.Hooks";
|
|
|
|
+ private static final String CONTEXT_OPERATOR_KEY = SecurityContext.class.getName();
|
|
|
|
|
|
public ReactorContextTestExecutionListener() {
|
|
public ReactorContextTestExecutionListener() {
|
|
super(createDelegate());
|
|
super(createDelegate());
|
|
@@ -59,12 +60,12 @@ public class ReactorContextTestExecutionListener
|
|
@Override
|
|
@Override
|
|
public void beforeTestMethod(TestContext testContext) throws Exception {
|
|
public void beforeTestMethod(TestContext testContext) throws Exception {
|
|
SecurityContext securityContext = TestSecurityContextHolder.getContext();
|
|
SecurityContext securityContext = TestSecurityContextHolder.getContext();
|
|
- Hooks.onLastOperator(Operators.lift((s, sub) -> new SecuritySubContext<>(sub, securityContext)));
|
|
|
|
|
|
+ Hooks.onLastOperator(CONTEXT_OPERATOR_KEY, Operators.lift((s, sub) -> new SecuritySubContext<>(sub, securityContext)));
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
@Override
|
|
public void afterTestMethod(TestContext testContext) throws Exception {
|
|
public void afterTestMethod(TestContext testContext) throws Exception {
|
|
- Hooks.resetOnLastOperator();
|
|
|
|
|
|
+ Hooks.resetOnLastOperator(CONTEXT_OPERATOR_KEY);
|
|
}
|
|
}
|
|
|
|
|
|
private static class SecuritySubContext<T> implements CoreSubscriber<T> {
|
|
private static class SecuritySubContext<T> implements CoreSubscriber<T> {
|