AbstractAuthenticatingTests.java

package uk.ac.ox.zoo.seeg.abraid.mp.publicsite;

import org.junit.Before;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.context.SecurityContextImpl;
import uk.ac.ox.zoo.seeg.abraid.mp.publicsite.domain.PublicSiteUser;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

/**
 * Base class to ease the setup of users in tests.
 * Copyright (c) 2014 University of Oxford
 */
public abstract class AbstractAuthenticatingTests {
    @Before
    public void setupSecurityContext() {
        SecurityContextHolder.setContext(new SecurityContextImpl());
        setupAnonymousUser();
    }

    public static void setupCurrentUser(PublicSiteUser user) {
        Authentication authentication = mock(Authentication.class);
        when(authentication.getPrincipal()).thenReturn(user);
        SecurityContextHolder.getContext().setAuthentication(authentication);
    }

    public static void setupAnonymousUser() {
        Authentication authentication = mock(Authentication.class);
        when(authentication.getPrincipal()).thenReturn("anonymousUser"); // This is the documented behaviour of spring
        SecurityContextHolder.getContext().setAuthentication(authentication);
    }
}