/***
* Copyright (c) 2009 Caelum - www.caelum.com.br/opensource
* All rights reserved.
*
* 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 br.com.caelum.vraptor.ioc.pico;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.notNullValue;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.util.Arrays;
import java.util.HashMap;
import javax.annotation.PreDestroy;
import org.junit.Test;
import br.com.caelum.vraptor.ComponentRegistry;
import br.com.caelum.vraptor.core.RequestInfo;
import br.com.caelum.vraptor.http.MutableRequest;
import br.com.caelum.vraptor.http.MutableResponse;
import br.com.caelum.vraptor.ioc.ApplicationScoped;
import br.com.caelum.vraptor.ioc.ComponentFactory;
import br.com.caelum.vraptor.ioc.ContainerProvider;
import br.com.caelum.vraptor.ioc.GenericContainerTest;
import br.com.caelum.vraptor.ioc.RequestScoped;
import br.com.caelum.vraptor.ioc.WhatToDo;
import br.com.caelum.vraptor.test.HttpServletRequestMock;
import br.com.caelum.vraptor.test.HttpSessionMock;
public class CustomPicoProviderTest extends GenericContainerTest {
private int counter;
static class CustomPicoProvider extends PicoProvider {
@Override
protected void registerBundledComponents(ComponentRegistry registry) {
super.registerBundledComponents(registry);
registry.register(MyAppComponent.class, MyAppComponent.class);
registry.register(MyAppComponentWithLifecycle.class, MyAppComponentWithLifecycle.class);
registry.register(MyRequestComponent.class, MyRequestComponent.class);
registry.register(MyFactory.class, MyFactory.class);
}
}
@ApplicationScoped
public static class MyAppComponent {
}
@RequestScoped
public static class MyRequestComponent {
}
@ApplicationScoped
public static class MyFactory implements ComponentFactory<Void> {
private int calls = 0;
public int getCalls() {
return calls;
}
@PreDestroy
public void preDestroy() {
calls++;
}
public Void getInstance() {
return null;
}
}
@ApplicationScoped
public static class MyAppComponentWithLifecycle {
private int calls = 0;
@PreDestroy
public void z() {
calls++;
}
}
@Test
public void callsPredestroyExactlyOneTimeForAppScopedComponents() throws Exception {
MyAppComponentWithLifecycle component = getFromContainer(MyAppComponentWithLifecycle.class);
assertThat(0, is(equalTo(component.calls)));
provider.stop();
assertThat(1, is(equalTo(component.calls)));
resetProvider();
}
@Test
public void callsPredestroyExactlyOneTimeForAppScopedComponentFactories() throws Exception {
MyFactory component = getFromContainer(MyFactory.class);
assertThat(0, is(equalTo(component.calls)));
provider.stop();
assertThat(1, is(equalTo(component.calls)));
resetProvider();
}
@Test
public void canProvideCustomApplicationScopedComponents() {
MyAppComponent component = getFromContainer(MyAppComponent.class);
assertThat(component, is(notNullValue()));
}
@Test
public void canProvideCustomRequestScopedComponents() {
checkAvailabilityFor(false, Arrays.<Class<?>>asList(MyRequestComponent.class));
}
@Override
public ContainerProvider getProvider() {
return new CustomPicoProvider();
}
@Override
protected <T> T executeInsideRequest(WhatToDo<T> execution) {
final HttpSessionMock session = new HttpSessionMock(context, "session" + ++counter);
final MutableRequest request = mock(MutableRequest.class, "request" + ++counter);
when(request.getRequestURI()).thenReturn("what.ever.request.uri");
when(request.getSession()).thenReturn(session);
when(request.getParameterMap()).thenReturn(new HashMap<String, String[]>());
when(request.getParameter("view")).thenReturn(null);
MutableResponse response = mock(MutableResponse.class, "response" + counter);
RequestInfo webRequest = new RequestInfo(context, null, new HttpServletRequestMock(session, request), response);
return execution.execute(webRequest, counter);
}
@Override
protected void configureExpectations() {
try {
when(context.getRealPath("/WEB-INF/classes/vraptor.xml")).thenReturn("non-existing-vraptor.xml");
when(context.getRealPath("/WEB-INF/classes/views.properties")).thenReturn("views.properties");
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}