com.trenako.web.controllers.AuthControllerTests.java Source code

Introduction

Here is the source code for com.trenako.web.controllers.AuthControllerTests.java

Source

/*
 * Copyright 2012 the original author or authors.
 *
 * 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 com.trenako.web.controllers;

import static org.mockito.Mockito.*;
import static org.junit.Assert.*;
import static org.springframework.test.web.ModelAndViewAssert.*;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.runners.MockitoJUnitRunner;

import org.springframework.dao.DuplicateKeyException;
import org.springframework.ui.ModelMap;
import org.springframework.ui.ExtendedModelMap;
import org.springframework.validation.BindingResult;
import org.springframework.web.servlet.ModelAndView;

import com.trenako.entities.Account;
import com.trenako.web.security.SignupService;

/**
 * 
 * @author Carlo Micieli
 *
 */
@RunWith(MockitoJUnitRunner.class)
public class AuthControllerTests {

    @Mock
    ModelMap mockModel;
    @Mock
    BindingResult mockResult;

    @Mock
    SignupService mockService;
    AuthController controller;

    @Before
    public void setup() {
        MockitoAnnotations.initMocks(this);
        controller = new AuthController(mockService);
    }

    @Test
    public void shouldRenderLoginForm() {
        String viewName = controller.login();
        assertEquals("auth/login", viewName);
    }

    @Test
    public void shouldRenderSignupForm() {
        ModelAndView mav = controller.signUp();
        assertViewName(mav, "auth/signup");
        assertModelAttributeAvailable(mav, "account");
    }

    @Test
    public void shouldRedirectAfterAccountValidationErrors() {
        when(mockResult.hasErrors()).thenReturn(true);
        ModelMap model = new ExtendedModelMap();

        String viewName = controller.createUser(newAccount(), mockResult, model);

        assertEquals("auth/signup", viewName);
        assertTrue(model.containsAttribute("account"));
    }

    @Test
    public void shouldRedirectAfterDatabaseErrors() {
        doThrow(new DuplicateKeyException("E11000 duplicate key error index")).when(mockService)
                .createAccount(eq(newAccount()));
        when(mockResult.hasErrors()).thenReturn(false);
        ModelMap model = new ExtendedModelMap();

        String viewName = controller.createUser(newAccount(), mockResult, model);

        assertEquals("auth/signup", viewName);
        assertTrue(model.containsAttribute("account"));
        assertTrue(model.containsAttribute("message"));
        assertEquals(AuthController.EMAIL_ADDRESS_OR_DISPLAY_NAME_ALREADY_TAKEN, model.get("message"));
        verify(mockService, times(0)).authenticate(eq(newAccount()));
    }

    @Test
    public void shouldCreateNewAccounts() {
        Account account = new Account.Builder("mail@mail.com").displayName("Bob").build();
        Account newAccount = new Account.Builder("mail@mail.com").displayName("Bob").build();

        when(mockResult.hasErrors()).thenReturn(false);
        when(mockService.createAccount(eq(account))).thenReturn(newAccount);

        String viewName = controller.createUser(account, mockResult, mockModel);

        assertEquals("redirect:/default", viewName);
        verify(mockService, times(1)).createAccount(eq(newAccount));
        verify(mockService, times(1)).authenticate(eq(newAccount));
    }

    Account newAccount() {
        return new Account.Builder("mail@mail.com").displayName("Bob").build();
    }
}