/*
 * Copyright 2005 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.usermgt.hibernate;

import org.wamblee.system.adapters.DefaultContainer;
import org.wamblee.system.adapters.ObjectConfiguration;
import org.wamblee.system.core.Scope;
import org.wamblee.system.spring.component.DatabaseTesterComponent;

import org.wamblee.test.spring.TestTransactionCallback;

import org.wamblee.usermgt.GroupSet;
import org.wamblee.usermgt.InMemoryGroupSetTest;

import java.sql.ResultSet;
import java.sql.SQLException;

import java.util.HashMap;
import java.util.Map;

/**
 * Tests for {@link org.wamblee.usermgt.hibernate.HibernateGroupSet}
 * 
 * @author Erik Brakkee
 */
public class HibernateGroupSetTest extends InMemoryGroupSetTest {
    private static final String GROUP_TABLE = "GROUPS";

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

    private DefaultContainer container;

    private Scope scope;

    private DatabaseTesterComponent databaseTester;

    private GroupSet groupSet;

    @Override
    protected void setUp() throws Exception {
        container = new UserMgtRepositoryTestContainer("top");

        ObjectConfiguration config = new ObjectConfiguration(
            HibernateGroupSetTest.class);
        config.getSetterConfig().clear().add("setGroupSet").add(
            "setDatabaseTester");
        container.addComponent("testcase", this, config);

        scope = container.start();

        databaseTester.cleanDatabase();
        super.setUp();
    }

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

    public void setDatabaseTester(DatabaseTesterComponent aDatabaseTester) {
        databaseTester = aDatabaseTester;
    }

    public void setGroupSet(GroupSet aGroupSet) {
        groupSet = aGroupSet;
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupCount(int)
     */
    @Override
    protected void checkGroupCount(int aSize) throws SQLException {
        databaseTester.flush();
        super.checkGroupCount(aSize);
        assertEquals(aSize, databaseTester.getTableSize(GROUP_TABLE));
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupExists(java.lang.String
     * )
     */
    @Override
    protected void checkGroupExists(final String aGroup) throws SQLException {
        databaseTester.flush();

        Map<String, Integer> result = databaseTester
            .executeTransaction(new TestTransactionCallback() {
                /*
                 * (non-Javadoc)
                 * 
                 * @see org.wamblee.test.TestTransactionCallback#execute()
                 */
                @Override
                public Map execute() throws Exception {
                    ResultSet result = databaseTester.executeQuery(GROUP_QUERY,
                        aGroup);
                    Map<String, Integer> res = new HashMap<String, Integer>();
                    res.put("result", databaseTester.countResultSet(result));

                    return res;
                }
            });

        int count = result.get("result");
        assertEquals(1, count);
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.wamblee.usermgt.InMemoryGroupSetTest#checkGroupNotExists(java.lang
     * .String)
     */
    @Override
    protected void checkGroupNotExists(String aGroup) throws SQLException {
        databaseTester.flush();

        ResultSet result = databaseTester.executeQuery(GROUP_QUERY, aGroup);
        assertEquals(0, databaseTester.countResultSet(result));
    }

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