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

import static br.com.caelum.vraptor.interceptor.FlashInterceptor.FLASH_INCLUDED_PARAMETERS;
import static java.util.Collections.singletonMap;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.anyObject;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyZeroInteractions;
import static org.mockito.Mockito.when;

import java.util.Collections;
import java.util.Map;

import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;

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

import br.com.caelum.vraptor.Result;
import br.com.caelum.vraptor.core.InterceptorStack;
import br.com.caelum.vraptor.http.MutableResponse;
import br.com.caelum.vraptor.http.VRaptorResponse;

/**
 * Tests FlashInterceptor
 *
 * @author Lucas Cavalcanti
 * @author Adriano Almeida
 * @author Caires Vinicius
 */
public class FlashInterceptorTest {


	private @Mock HttpSession session;
	private @Mock Result result;
	private @Mock InterceptorStack stack;
	private @Mock HttpServletResponse mockResponse;
	private MutableResponse response;
	private FlashInterceptor interceptor;

	@Before
	public void setUp() throws Exception {
		MockitoAnnotations.initMocks(this);
		
		response = new VRaptorResponse(mockResponse);
		interceptor = new FlashInterceptor(session, result, response);
	}

	@Test
	public void shouldAcceptAlways() {
		assertTrue(interceptor.accepts(null));
	}
	
	@Test
	public void shouldDoNothingWhenThereIsNoFlashParameters() throws Exception {
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(null);

		interceptor.intercept(stack, null, null);
		verifyZeroInteractions(result);
	}

	@Test
	public void shouldAddAllFlashParametersToResult() throws Exception {
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(singletonMap("Abc", 1002));
		
		interceptor.intercept(stack, null, null);
		verify(result).include("Abc", 1002);
	}

	@Test
	public void shouldRemoveFlashIncludedParameters() throws Exception {
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(singletonMap("Abc", 1002));
		
		interceptor.intercept(stack, null, null);
		verify(session).removeAttribute(FLASH_INCLUDED_PARAMETERS);
	}
	
	@Test
	public void shouldIncludeFlashParametersWhenARedirectHappens() throws Exception {
		Map<String, Object> parameters = Collections.<String, Object>singletonMap("Abc", 1002);

		when(result.included()).thenReturn(parameters);
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(null);

		interceptor.intercept(stack, null, null);
		response.sendRedirect("Anything");
		
		verify(session).setAttribute(FLASH_INCLUDED_PARAMETERS, parameters);
	}
	@Test
	public void shouldNotIncludeFlashParametersWhenThereIsNoIncludedParameter() throws Exception {
		Map<String, Object> parameters = Collections.emptyMap();

		when(result.included()).thenReturn(parameters);
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(null);

		interceptor.intercept(stack, null, null);
		response.sendRedirect("Anything");

		verify(session, never()).setAttribute(anyString(), anyObject());
	}
	
	@Test
	public void shouldNotCrashWhenSessionIsInvalid() throws Exception {
		Map<String, Object> parameters = Collections.<String, Object>singletonMap("Abc", 1002);

		when(result.included()).thenReturn(parameters);
		doThrow(new IllegalStateException()).when(session).setAttribute(FLASH_INCLUDED_PARAMETERS, parameters);
		when(session.getAttribute(FLASH_INCLUDED_PARAMETERS)).thenReturn(null);

		interceptor.intercept(stack, null, null);
		response.sendRedirect("Anything");
	}
}