/*
 * Copyright 2005-2010 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 org.wamblee.security.authentication.jpa;

import java.sql.Connection;
import java.sql.ResultSet;
import java.util.Set;

import org.wamblee.cache.EhCache;
import org.wamblee.io.ClassPathResource;
import org.wamblee.security.authentication.Group;
import org.wamblee.security.authentication.GroupSet;
import org.wamblee.security.authentication.InMemoryUserSetTest;
import org.wamblee.security.authentication.Md5HexMessageDigester;
import org.wamblee.security.authentication.MessageDigester;
import org.wamblee.security.authentication.NameValidator;
import org.wamblee.security.authentication.RegexpNameValidator;
import org.wamblee.security.authentication.User;
import org.wamblee.security.authentication.UserSet;
import org.wamblee.support.persistence.JpaTester;
import org.wamblee.support.persistence.TransactionProxyFactory;
import org.wamblee.support.persistence.DatabaseUtils.JdbcUnitOfWork;

/**
 * Tests for {@link org.wamblee.usermgt.hibernate.HibernateGroupSet}
 * 
 * @author Erik Brakkee
 */
public class JpaUserSetTest extends InMemoryUserSetTest {
    private static final String USER_TABLE = "SEC_USER";

    private static final String GROUP_TABLE = "SEC_GROUP";

    private static final String USER_QUERY = "select * from " + USER_TABLE +
        " where name = ?";

    private static final String GROUP_QUERY = "select * from " + GROUP_TABLE +
        " where name = ?";

    private UserSet userset;

    private GroupSet groupset;

    private EhCache<String, User> userCache;

    private JpaTester jpaTester; 

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryUserSetTest#setUp()
     */
    @Override
    protected void setUp() throws Exception {
        jpaTester = new JpaTester(new SecurityPersistenceUnit());
        jpaTester.start();

        userCache = new EhCache<String, User>(
            new ClassPathResource("properties/org.wamblee.security.ehcache.xml"), "users");
     
        userset = createUserSetImpl();
        groupset = createGroupSetImpl();
        
        clearUserCache();

        super.setUp();
    }
    
    private UserSet createUserSetImpl() {
        NameValidator passwordValidator = new RegexpNameValidator(
            ".{5,}", "INVALID_PASSWORD", "Password must have at least 5 characters");
        
        MessageDigester passwordDigester = new Md5HexMessageDigester();
        TransactionProxyFactory<UserSet> factory = new TransactionProxyFactory<UserSet>(
            jpaTester.getJpaBuilder(), UserSet.class);
        UserSet jpaUserset = new JpaUserSet(userCache, passwordValidator, passwordDigester,
            factory.getTransactionScopedEntityManager());
        return factory.getProxy(jpaUserset);
    }
    
    private GroupSet createGroupSetImpl() {
        TransactionProxyFactory<GroupSet> factory = new TransactionProxyFactory<GroupSet>(
            jpaTester.getJpaBuilder(), GroupSet.class);
        GroupSet groups = new JpaGroupSet(factory.getTransactionScopedEntityManager());
        GroupSet proxy = factory.getProxy(groups);
        return proxy;
    }

    @Override
    protected void tearDown() throws Exception {
        jpaTester.stop();
        super.tearDown();
    }

    /**
     * Clears the user cache.
     */
    private void clearUserCache() {
        userCache.clear();
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupCount(int)
     */
    @Override
    protected void checkUserCount(int aSize) throws Exception {
        super.checkUserCount(aSize);
        assertEquals(aSize, jpaTester.getDbUtils().getTableSize(USER_TABLE));
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupExists(java.lang.String
     * )
     */
    @Override
    protected void checkUserExists(final String aUser) throws Exception {
        assertEquals(1, countUser(aUser));
    }

    private int countUser(final String aUser) throws Exception {
        int count = jpaTester.getDbUtils().executeInTransaction(new JdbcUnitOfWork<Integer>() {
            @Override
            public Integer execute(Connection aConnection) throws Exception {
                ResultSet res = jpaTester.getDbUtils().executeQuery(aConnection, USER_QUERY, aUser);
                return jpaTester.getDbUtils().countResultSet(res);
            }
        });
        return count;
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupNotExists(java.lang
     * .String)
     */
    @Override
    protected void checkUserNotExists(String aUser) throws Exception {
        assertEquals(0, countUser(aUser));
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupCount(int)
     */
    @Override
    protected void checkGroupCount(int aSize) throws Exception {
        assertEquals(aSize, jpaTester.getDbUtils().getTableSize(GROUP_TABLE));
    }
    
    private int countGroup(final String aGroup) throws Exception {
        int count = jpaTester.getDbUtils().executeInTransaction(new JdbcUnitOfWork<Integer>() {
            @Override
            public Integer execute(Connection aConnection) throws Exception {
                ResultSet res = jpaTester.getDbUtils().executeQuery(aConnection, GROUP_QUERY, aGroup);
                return jpaTester.getDbUtils().countResultSet(res);
            }
        });
        return count;
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupExists(java.lang.String
     * )
     */
    @Override
    protected void checkGroupExists(String aGroup) throws Exception {
        assertEquals(1, countGroup(aGroup));
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupNotExists(java.lang
     * .String)
     */
    @Override
    protected void checkGroupNotExists(String aGroup) throws Exception {
        assertEquals(0, countGroup(aGroup));
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryGroupSetTest#createGroupSet()
     */
    @Override
    protected UserSet createUserSet() {
        return userset;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryUserSetTest#createGroupSet()
     */
    @Override
    protected GroupSet createGroupSet() {
        return groupset;
    }

    /**
     * Reproduction of a bug. Create a user which is in group1 Add it to a
     * second group group2. Remove the user from group1. Verify the user is in
     * group2.
     * 
     */
    public void testVerifyAddRemove() throws Exception {
        jpaTester.getDbUtils().cleanDatabase(new SecurityTables()); // super class setup always creates one group. 
        
        GroupSet groups = getGroups();
        assertEquals(0, groups.size());

        Group group1 = createGroup("group1");
        Group group2 = createGroup("group2");
        groups.add(group1);
        groups.add(group2);
        checkGroupExists("group1");
        checkGroupExists("group2");

        User user = createUser("user", PASSWORD, group1);
        getUsers().add(user);
        checkUserExists("user");

        addUserToGroup(user, group2);
        getUsers().userModified(user);
        clearUserCache();
        
        User user2 = getUsers().find("user");
        
        Set<Group> userGroups = user2.getGroups();
        assertTrue(user2.isInGroup("group1"));
        assertTrue(user2.isInGroup("group2"));
        assertEquals(2, userGroups.size());

        removeUserFromGroup(user, group1);
        getUsers().userModified(user);
        clearUserCache();
        user2 = getUsers().find("user");
        userGroups = user2.getGroups();
        assertFalse(user2.isInGroup("group1"));
        assertTrue(user2.isInGroup("group2"));
        assertEquals(1, userGroups.size());
    }
}