RequestHandlerObserverTest.java

/***
 * 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.observer;

import static br.com.caelum.vraptor.controller.HttpMethod.POST;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.util.EnumSet;

import javax.enterprise.event.Event;
import javax.servlet.FilterChain;

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;

import br.com.caelum.vraptor.controller.ControllerMethod;
import br.com.caelum.vraptor.controller.ControllerNotFoundHandler;
import br.com.caelum.vraptor.controller.HttpMethod;
import br.com.caelum.vraptor.controller.InvalidInputException;
import br.com.caelum.vraptor.controller.InvalidInputHandler;
import br.com.caelum.vraptor.controller.MethodNotAllowedHandler;
import br.com.caelum.vraptor.core.InterceptorStack;
import br.com.caelum.vraptor.events.ControllerFound;
import br.com.caelum.vraptor.events.RequestSucceded;
import br.com.caelum.vraptor.events.VRaptorRequestStarted;
import br.com.caelum.vraptor.http.MutableRequest;
import br.com.caelum.vraptor.http.MutableResponse;
import br.com.caelum.vraptor.http.UrlToControllerTranslator;
import br.com.caelum.vraptor.http.route.ControllerNotFoundException;
import br.com.caelum.vraptor.http.route.MethodNotAllowedException;

public class RequestHandlerObserverTest {

	private @Mock UrlToControllerTranslator translator;
	private @Mock MutableRequest webRequest;
	private @Mock MutableResponse webResponse;
	private @Mock ControllerNotFoundHandler notFoundHandler;
	private @Mock MethodNotAllowedHandler methodNotAllowedHandler;
	private @Mock Event<ControllerFound> controllerFoundEvent;
	private @Mock Event<RequestSucceded> requestSucceededEvent;
	private @Mock InterceptorStack interceptorStack;
	private @Mock FilterChain chain;
	private @Mock InvalidInputHandler invalidInputHandler;
	
	private VRaptorRequestStarted requestStarted;
	private RequestHandlerObserver observer;

	@Before
	public void config() {
		MockitoAnnotations.initMocks(this);
		requestStarted = new VRaptorRequestStarted(chain, webRequest, webResponse);
		observer = new RequestHandlerObserver(translator, notFoundHandler, methodNotAllowedHandler, controllerFoundEvent, requestSucceededEvent, interceptorStack, invalidInputHandler);
	}

	@Test
	public void shouldHandle404() throws Exception {
		when(translator.translate(webRequest)).thenThrow(new ControllerNotFoundException());
		observer.handle(requestStarted);
		verify(notFoundHandler).couldntFind(chain, webRequest, webResponse);
		verify(interceptorStack, never()).start();
	}

	@Test
	public void shouldHandle405() throws Exception {
		EnumSet<HttpMethod> allowedMethods = EnumSet.of(HttpMethod.GET);
		when(translator.translate(webRequest)).thenThrow(new MethodNotAllowedException(allowedMethods, POST.toString()));
		observer.handle(requestStarted);
		verify(methodNotAllowedHandler).deny(webRequest, webResponse, allowedMethods);
		verify(interceptorStack, never()).start();
	}
	
	@Test
	public void shouldHandle400() throws Exception {
		InvalidInputException invalidInputException = new InvalidInputException("");
		when(translator.translate(webRequest)).thenThrow(invalidInputException);
		observer.handle(requestStarted);
		verify(interceptorStack, never()).start();
		verify(invalidInputHandler).deny(invalidInputException);
	}

	@Test
	public void shouldUseControllerMethodFoundWithNextInterceptor() throws Exception {
		final ControllerMethod method = mock(ControllerMethod.class);
		when(translator.translate(webRequest)).thenReturn(method);
		observer.handle(requestStarted);
		verify(interceptorStack).start();
	}
	
	@Test
	public void shouldFireTheControllerWasFound() throws Exception {
		final ControllerMethod method = mock(ControllerMethod.class);
		when(translator.translate(webRequest)).thenReturn(method);
		observer.handle(requestStarted);
		verify(controllerFoundEvent).fire(any(ControllerFound.class));
	}
	
	@Test
	public void shouldFireTheRequestSuceeded() throws Exception {
		final ControllerMethod method = mock(ControllerMethod.class);
		when(translator.translate(webRequest)).thenReturn(method);
		observer.handle(requestStarted);
		verify(requestSucceededEvent).fire(any(RequestSucceded.class));
	}
}