Example 1 with PassThroughFilterChain
use of org.springframework.web.testfixture.servlet.PassThroughFilterChain in project spring-framework by spring-projects.
the class OpenEntityManagerInViewTests method testOpenEntityManagerInViewFilterAsyncScenario.
@Test
public void testOpenEntityManagerInViewFilterAsyncScenario() throws Exception {
given(manager.isOpen()).willReturn(true);
final EntityManagerFactory factory2 = mock(EntityManagerFactory.class);
final EntityManager manager2 = mock(EntityManager.class);
given(factory2.createEntityManager()).willReturn(manager2);
given(manager2.isOpen()).willReturn(true);
MockServletContext sc = new MockServletContext();
StaticWebApplicationContext wac = new StaticWebApplicationContext();
wac.setServletContext(sc);
wac.getDefaultListableBeanFactory().registerSingleton("entityManagerFactory", factory);
wac.getDefaultListableBeanFactory().registerSingleton("myEntityManagerFactory", factory2);
wac.refresh();
sc.setAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
MockFilterConfig filterConfig = new MockFilterConfig(wac.getServletContext(), "filter");
MockFilterConfig filterConfig2 = new MockFilterConfig(wac.getServletContext(), "filter2");
filterConfig2.addInitParameter("entityManagerFactoryBeanName", "myEntityManagerFactory");
final OpenEntityManagerInViewFilter filter = new OpenEntityManagerInViewFilter();
filter.init(filterConfig);
final OpenEntityManagerInViewFilter filter2 = new OpenEntityManagerInViewFilter();
filter2.init(filterConfig2);
final AtomicInteger count = new AtomicInteger();
final FilterChain filterChain = (servletRequest, servletResponse) -> {
assertThat(TransactionSynchronizationManager.hasResource(factory)).isTrue();
servletRequest.setAttribute("invoked", Boolean.TRUE);
count.incrementAndGet();
};
final AtomicInteger count2 = new AtomicInteger();
final FilterChain filterChain2 = (servletRequest, servletResponse) -> {
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isTrue();
filter.doFilter(servletRequest, servletResponse, filterChain);
count2.incrementAndGet();
};
FilterChain filterChain3 = new PassThroughFilterChain(filter2, filterChain2);
AsyncWebRequest asyncWebRequest = mock(AsyncWebRequest.class);
given(asyncWebRequest.isAsyncStarted()).willReturn(true);
WebAsyncManager asyncManager = WebAsyncUtils.getAsyncManager(this.request);
asyncManager.setTaskExecutor(this.taskExecutor);
asyncManager.setAsyncWebRequest(asyncWebRequest);
asyncManager.startCallableProcessing((Callable<String>) () -> "anything");
this.taskExecutor.await();
assertThat(asyncManager.getConcurrentResult()).as("Concurrent result ").isEqualTo("anything");
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
filter2.doFilter(this.request, this.response, filterChain3);
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
assertThat(count.get()).isEqualTo(1);
assertThat(count2.get()).isEqualTo(1);
assertThat(request.getAttribute("invoked")).isNotNull();
verify(asyncWebRequest, times(2)).addCompletionHandler(any(Runnable.class));
verify(asyncWebRequest).addTimeoutHandler(any(Runnable.class));
verify(asyncWebRequest, times(2)).addCompletionHandler(any(Runnable.class));
verify(asyncWebRequest).startAsync();
// Async dispatch after concurrent handling produces result ...
reset(asyncWebRequest);
given(asyncWebRequest.isAsyncStarted()).willReturn(false);
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
filter.doFilter(this.request, this.response, filterChain3);
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
assertThat(count.get()).isEqualTo(2);
assertThat(count2.get()).isEqualTo(2);
verify(this.manager).close();
verify(manager2).close();
wac.close();
}
Also used :
ArgumentMatchers.any(org.mockito.ArgumentMatchers.any)
BeforeEach(org.junit.jupiter.api.BeforeEach)
Assertions.assertThat(org.assertj.core.api.Assertions.assertThat)
AsyncWebRequest(org.springframework.web.context.request.async.AsyncWebRequest)
StaticWebApplicationContext(org.springframework.web.context.support.StaticWebApplicationContext)
Callable(java.util.concurrent.Callable)
WebAsyncManager(org.springframework.web.context.request.async.WebAsyncManager)
TransactionSynchronizationManager(org.springframework.transaction.support.TransactionSynchronizationManager)
ServletWebRequest(org.springframework.web.context.request.ServletWebRequest)
MockHttpServletResponse(org.springframework.web.testfixture.servlet.MockHttpServletResponse)
AtomicInteger(java.util.concurrent.atomic.AtomicInteger)
BDDMockito.given(org.mockito.BDDMockito.given)
SimpleAsyncTaskExecutor(org.springframework.core.task.SimpleAsyncTaskExecutor)
AsyncEvent(jakarta.servlet.AsyncEvent)
StandardServletAsyncWebRequest(org.springframework.web.context.request.async.StandardServletAsyncWebRequest)
WebAsyncUtils(org.springframework.web.context.request.async.WebAsyncUtils)
MockHttpServletRequest(org.springframework.web.testfixture.servlet.MockHttpServletRequest)
FilterChain(jakarta.servlet.FilterChain)
WebApplicationContext(org.springframework.web.context.WebApplicationContext)
MockFilterConfig(org.springframework.web.testfixture.servlet.MockFilterConfig)
Mockito.times(org.mockito.Mockito.times)
EntityManagerFactory(jakarta.persistence.EntityManagerFactory)
Mockito.verify(org.mockito.Mockito.verify)
TimeUnit(java.util.concurrent.TimeUnit)
Test(org.junit.jupiter.api.Test)
CountDownLatch(java.util.concurrent.CountDownLatch)
AfterEach(org.junit.jupiter.api.AfterEach)
AsyncListener(jakarta.servlet.AsyncListener)
EntityManager(jakarta.persistence.EntityManager)
MockAsyncContext(org.springframework.web.testfixture.servlet.MockAsyncContext)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
Mockito.reset(org.mockito.Mockito.reset)
MockServletContext(org.springframework.web.testfixture.servlet.MockServletContext)
Mockito.mock(org.mockito.Mockito.mock)
WebAsyncManager(org.springframework.web.context.request.async.WebAsyncManager)
FilterChain(jakarta.servlet.FilterChain)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
StaticWebApplicationContext(org.springframework.web.context.support.StaticWebApplicationContext)
AsyncWebRequest(org.springframework.web.context.request.async.AsyncWebRequest)
StandardServletAsyncWebRequest(org.springframework.web.context.request.async.StandardServletAsyncWebRequest)
MockServletContext(org.springframework.web.testfixture.servlet.MockServletContext)
MockFilterConfig(org.springframework.web.testfixture.servlet.MockFilterConfig)
EntityManager(jakarta.persistence.EntityManager)
AtomicInteger(java.util.concurrent.atomic.AtomicInteger)
EntityManagerFactory(jakarta.persistence.EntityManagerFactory)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
Test(org.junit.jupiter.api.Test)
Example 2 with PassThroughFilterChain
use of org.springframework.web.testfixture.servlet.PassThroughFilterChain in project spring-framework by spring-projects.
the class OpenEntityManagerInViewTests method testOpenEntityManagerInViewFilter.
@Test
public void testOpenEntityManagerInViewFilter() throws Exception {
given(manager.isOpen()).willReturn(true);
final EntityManagerFactory factory2 = mock(EntityManagerFactory.class);
final EntityManager manager2 = mock(EntityManager.class);
given(factory2.createEntityManager()).willReturn(manager2);
given(manager2.isOpen()).willReturn(true);
MockServletContext sc = new MockServletContext();
StaticWebApplicationContext wac = new StaticWebApplicationContext();
wac.setServletContext(sc);
wac.getDefaultListableBeanFactory().registerSingleton("entityManagerFactory", factory);
wac.getDefaultListableBeanFactory().registerSingleton("myEntityManagerFactory", factory2);
wac.refresh();
sc.setAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE, wac);
MockHttpServletRequest request = new MockHttpServletRequest(sc);
MockHttpServletResponse response = new MockHttpServletResponse();
MockFilterConfig filterConfig = new MockFilterConfig(wac.getServletContext(), "filter");
MockFilterConfig filterConfig2 = new MockFilterConfig(wac.getServletContext(), "filter2");
filterConfig2.addInitParameter("entityManagerFactoryBeanName", "myEntityManagerFactory");
final OpenEntityManagerInViewFilter filter = new OpenEntityManagerInViewFilter();
filter.init(filterConfig);
final OpenEntityManagerInViewFilter filter2 = new OpenEntityManagerInViewFilter();
filter2.init(filterConfig2);
final FilterChain filterChain = (servletRequest, servletResponse) -> {
assertThat(TransactionSynchronizationManager.hasResource(factory)).isTrue();
servletRequest.setAttribute("invoked", Boolean.TRUE);
};
final FilterChain filterChain2 = (servletRequest, servletResponse) -> {
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isTrue();
filter.doFilter(servletRequest, servletResponse, filterChain);
};
FilterChain filterChain3 = new PassThroughFilterChain(filter2, filterChain2);
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
filter2.doFilter(request, response, filterChain3);
assertThat(TransactionSynchronizationManager.hasResource(factory)).isFalse();
assertThat(TransactionSynchronizationManager.hasResource(factory2)).isFalse();
assertThat(request.getAttribute("invoked")).isNotNull();
verify(manager).close();
verify(manager2).close();
wac.close();
}
Also used :
ArgumentMatchers.any(org.mockito.ArgumentMatchers.any)
BeforeEach(org.junit.jupiter.api.BeforeEach)
Assertions.assertThat(org.assertj.core.api.Assertions.assertThat)
AsyncWebRequest(org.springframework.web.context.request.async.AsyncWebRequest)
StaticWebApplicationContext(org.springframework.web.context.support.StaticWebApplicationContext)
Callable(java.util.concurrent.Callable)
WebAsyncManager(org.springframework.web.context.request.async.WebAsyncManager)
TransactionSynchronizationManager(org.springframework.transaction.support.TransactionSynchronizationManager)
ServletWebRequest(org.springframework.web.context.request.ServletWebRequest)
MockHttpServletResponse(org.springframework.web.testfixture.servlet.MockHttpServletResponse)
AtomicInteger(java.util.concurrent.atomic.AtomicInteger)
BDDMockito.given(org.mockito.BDDMockito.given)
SimpleAsyncTaskExecutor(org.springframework.core.task.SimpleAsyncTaskExecutor)
AsyncEvent(jakarta.servlet.AsyncEvent)
StandardServletAsyncWebRequest(org.springframework.web.context.request.async.StandardServletAsyncWebRequest)
WebAsyncUtils(org.springframework.web.context.request.async.WebAsyncUtils)
MockHttpServletRequest(org.springframework.web.testfixture.servlet.MockHttpServletRequest)
FilterChain(jakarta.servlet.FilterChain)
WebApplicationContext(org.springframework.web.context.WebApplicationContext)
MockFilterConfig(org.springframework.web.testfixture.servlet.MockFilterConfig)
Mockito.times(org.mockito.Mockito.times)
EntityManagerFactory(jakarta.persistence.EntityManagerFactory)
Mockito.verify(org.mockito.Mockito.verify)
TimeUnit(java.util.concurrent.TimeUnit)
Test(org.junit.jupiter.api.Test)
CountDownLatch(java.util.concurrent.CountDownLatch)
AfterEach(org.junit.jupiter.api.AfterEach)
AsyncListener(jakarta.servlet.AsyncListener)
EntityManager(jakarta.persistence.EntityManager)
MockAsyncContext(org.springframework.web.testfixture.servlet.MockAsyncContext)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
Mockito.reset(org.mockito.Mockito.reset)
MockServletContext(org.springframework.web.testfixture.servlet.MockServletContext)
Mockito.mock(org.mockito.Mockito.mock)
EntityManager(jakarta.persistence.EntityManager)
MockHttpServletRequest(org.springframework.web.testfixture.servlet.MockHttpServletRequest)
EntityManagerFactory(jakarta.persistence.EntityManagerFactory)
FilterChain(jakarta.servlet.FilterChain)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
PassThroughFilterChain(org.springframework.web.testfixture.servlet.PassThroughFilterChain)
StaticWebApplicationContext(org.springframework.web.context.support.StaticWebApplicationContext)
MockServletContext(org.springframework.web.testfixture.servlet.MockServletContext)
MockHttpServletResponse(org.springframework.web.testfixture.servlet.MockHttpServletResponse)
MockFilterConfig(org.springframework.web.testfixture.servlet.MockFilterConfig)
Test(org.junit.jupiter.api.Test)
Aggregations
EntityManager (jakarta.persistence.EntityManager)2
EntityManagerFactory (jakarta.persistence.EntityManagerFactory)2
AsyncEvent (jakarta.servlet.AsyncEvent)2
AsyncListener (jakarta.servlet.AsyncListener)2
FilterChain (jakarta.servlet.FilterChain)2
Callable (java.util.concurrent.Callable)2
CountDownLatch (java.util.concurrent.CountDownLatch)2
TimeUnit (java.util.concurrent.TimeUnit)2
AtomicInteger (java.util.concurrent.atomic.AtomicInteger)2
Assertions.assertThat (org.assertj.core.api.Assertions.assertThat)2
AfterEach (org.junit.jupiter.api.AfterEach)2
BeforeEach (org.junit.jupiter.api.BeforeEach)2
Test (org.junit.jupiter.api.Test)2
ArgumentMatchers.any (org.mockito.ArgumentMatchers.any)2
BDDMockito.given (org.mockito.BDDMockito.given)2
Mockito.mock (org.mockito.Mockito.mock)2
Mockito.reset (org.mockito.Mockito.reset)2
Mockito.times (org.mockito.Mockito.times)2
Mockito.verify (org.mockito.Mockito.verify)2
SimpleAsyncTaskExecutor (org.springframework.core.task.SimpleAsyncTaskExecutor)2
