source: ComponentRegistry/trunk/ComponentRegistry/src/test/java/clarin/cmdi/componentregistry/impl/database/ComponentRegistryFactoryDbImplTest.java @ 3449

package clarin.cmdi.componentregistry.impl.database;

import org.springframework.jdbc.core.JdbcTemplate;
import org.junit.Before;

import clarin.cmdi.componentregistry.model.RegistryUser;
import clarin.cmdi.componentregistry.BaseUnitTest;
import clarin.cmdi.componentregistry.ComponentRegistry;
import clarin.cmdi.componentregistry.ComponentStatus;
import clarin.cmdi.componentregistry.OwnerUser;
import clarin.cmdi.componentregistry.UserCredentials;
import clarin.cmdi.componentregistry.UserUnauthorizedException;
import clarin.cmdi.componentregistry.rest.DummyPrincipal;

import org.springframework.beans.factory.annotation.Autowired;
import org.junit.Test;

import static org.junit.Assert.fail;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotSame;

/**
 * 
 * @author Twan Goosen <twan.goosen@mpi.nl>
 * @author George.Georgovassilis@mpi.nl
 */
public class ComponentRegistryFactoryDbImplTest extends BaseUnitTest {

    @Autowired
    private ComponentRegistryFactoryDbImpl componentRegistryFactory;
    @Autowired
    private JdbcTemplate jdbcTemplate;

    @Before
    public void init() {
        ComponentRegistryTestDatabase.resetDatabase(jdbcTemplate);
        ComponentRegistryTestDatabase.createTableRegistryUser(jdbcTemplate);
    }

    @Test
    public void testInjection() {
        assertNotNull(componentRegistryFactory);
        assertNotNull(jdbcTemplate);
    }

    @Test
    public void testGetPublicRegistry() {
        ComponentRegistry registry = componentRegistryFactory
                .getPublicRegistry();
        assertNotNull(registry);
    }

    @Test
    public void getComponentRegistry() throws UserUnauthorizedException {
        // Get public
        assertNotNull(componentRegistryFactory.getComponentRegistry(
                ComponentStatus.PUBLISHED, null, null));

        // Get for non-existing user
        final RegistryUser testUser = UserDaoTest.createTestUser();
        UserCredentials credentials = new DummyPrincipal(
                testUser.getPrincipalName()).getCredentials();

        ComponentRegistryDbImpl cr1 = (ComponentRegistryDbImpl) componentRegistryFactory
                .getComponentRegistry(ComponentStatus.PRIVATE, null,
                        credentials);
        assertNotNull(cr1);
        // Get for existing user
        ComponentRegistryDbImpl cr2 = (ComponentRegistryDbImpl) componentRegistryFactory
                .getComponentRegistry(ComponentStatus.PRIVATE, null,
                        credentials);
        ;
        assertNotNull(cr2);
        assertEquals(cr1.getOwner(), cr2.getOwner());

        // Get for another new user
        UserCredentials credentials2 = new DummyPrincipal(
                testUser.getPrincipalName() + "2").getCredentials();
        ComponentRegistryDbImpl cr3 = (ComponentRegistryDbImpl) componentRegistryFactory
                .getComponentRegistry(ComponentStatus.PRIVATE, null,
                        credentials2);
        assertNotNull(cr3);
        assertNotSame(cr1.getOwner(), cr3.getOwner());
    }

    @Test
    public void testGetOtherUserComponentRegistry()
            throws UserUnauthorizedException {
        UserCredentials userCredentials = DummyPrincipal.DUMMY_PRINCIPAL
                .getCredentials();

        // Create registry for new user
        ComponentRegistryDbImpl cr1 = (ComponentRegistryDbImpl) componentRegistryFactory
                .getComponentRegistry(ComponentStatus.PRIVATE, null,
                        userCredentials);

        Number id = cr1.getOwner().getId();

        // Get it as admin
        ComponentRegistryDbImpl cr2 = (ComponentRegistryDbImpl) componentRegistryFactory
                .getOtherUserComponentRegistry(
                        DummyPrincipal.DUMMY_ADMIN_PRINCIPAL,
                        ComponentStatus.PRIVATE, new OwnerUser(id));
        assertNotNull(cr2);
        // Should be this user's registry
        assertEquals(cr1.getOwner(), cr2.getOwner());

        // Try get it as non-admin
        try {
            componentRegistryFactory.getOtherUserComponentRegistry(
                    DummyPrincipal.DUMMY_PRINCIPAL, ComponentStatus.PRIVATE,
                    new OwnerUser(id));
            fail("Non-admin can get other user's component registry");
        } catch (Exception ex) {
            // Exception should occur
        }
    }
}