utils.AbstractTest.java Source code

Introduction

Here is the source code for utils.AbstractTest.java

Source

/* AbstractTest.java
 *
 * Copyright (C) 2015 Universidad de Sevilla
 * 
 * The use of this project is hereby constrained to the conditions of the 
 * TDG Licence, a copy of which you may download from 
 * http://www.tdg-seville.info/License.html
 * 
 */

package utils;

import org.junit.Before;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.authentication.TestingAuthenticationToken;
import org.springframework.security.core.context.SecurityContext;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;

import security.LoginService;
import utilities.PopulateDatabase;

public class AbstractTest {

    // Supporting services --------------------------------

    @Autowired
    private LoginService loginService;

    // Set up and tear down -------------------------------

    @Before
    public void setUp() {
        PopulateDatabase.main(null);
    }

    // Supporting methods ---------------------------------

    public void authenticate(String username) {
        UserDetails userDetails;
        TestingAuthenticationToken authenticationToken;
        SecurityContext context;

        if (username == null)
            authenticationToken = null;
        else {
            userDetails = loginService.loadUserByUsername(username);
            authenticationToken = new TestingAuthenticationToken(userDetails, null);
        }

        context = SecurityContextHolder.getContext();
        context.setAuthentication(authenticationToken);
    }

    public void unauthenticate() {
        authenticate(null);
    }

}