/*
* JBoss, Home of Professional Open Source
* Copyright 2010, Red Hat Middleware LLC, and individual contributors
* by the @authors tag. See the copyright.txt in the distribution for a
* full listing of individual contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* http://www.apache.org/licenses/LICENSE-2.0
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jboss.solder.servlet.test.weld.beanManager;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertNull;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import javax.enterprise.inject.Instance;
import javax.enterprise.inject.spi.BeanManager;
import javax.inject.Inject;
import javax.servlet.ServletContext;
import javax.servlet.ServletContextEvent;
import javax.servlet.ServletRequestEvent;
import javax.servlet.http.HttpServletRequest;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.shrinkwrap.api.Archive;
import org.jboss.solder.beanManager.BeanManagerLocator;
import org.jboss.solder.servlet.ServerInfo;
import org.jboss.solder.servlet.beanManager.ServletContextAttributeProvider;
import org.jboss.solder.servlet.event.ServletEventBridgeListener;
import org.jboss.solder.servlet.test.weld.util.Deployments;
import org.junit.Test;
import org.junit.runner.RunWith;
/**
* @author <a href="http://community.jboss.org/people/dan.j.allen">Dan Allen</a>
*/
@RunWith(Arquillian.class)
public class ServletContextAttributeProviderTest {
@Deployment
public static Archive<?> createDeployment() {
return Deployments.createMockableBeanWebArchive()
.addClasses(ServletContextAttributeProviderTest.class);
}
@Inject
BeanManager manager;
@Inject
ServletEventBridgeListener listener;
// TODO this should be in a separate test
@Inject
@ServerInfo
Instance<String> serverInfoProvider;
@Test
public void should_register_and_locate_bean_manager() {
String MOCK_SERVLET_CONTEXT = "Mock Servlet Context";
ServletContext ctx = mock(ServletContext.class);
when(ctx.getServerInfo()).thenReturn(MOCK_SERVLET_CONTEXT);
listener.contextInitialized(new ServletContextEvent(ctx));
verify(ctx).setAttribute(BeanManager.class.getName(), manager);
assertEquals(MOCK_SERVLET_CONTEXT, serverInfoProvider.get());
when(ctx.getAttribute(BeanManager.class.getName())).thenReturn(manager);
BeanManagerLocator locator = new BeanManagerLocator();
assertTrue(locator.isBeanManagerAvailable());
assertEquals(manager, locator.getBeanManager());
}
@Test
public void testBeanManagerLookupFromWorkerThreads() {
// mock the servlet context and a request
ServletContext context = mock(ServletContext.class);
HttpServletRequest request = mock(HttpServletRequest.class);
// let the ServletEventBridgeListener fire a @Initialized InternalServletRequestEvent
listener.requestInitialized(new ServletRequestEvent(context, request));
// the provider knows about the ServletContext -> the lookup must work
ServletContextAttributeProvider provider = new ServletContextAttributeProvider();
assertEquals(manager, provider.getBeanManager());
// let the ServletEventBridgeListener fire a @Destroyed InternalServletRequestEvent
listener.requestDestroyed(new ServletRequestEvent(context, request));
// The ServletContext is not available any more: lookup not possible
assertNull(provider.getBeanManager());
}
}