JDBCIdentityStoreTest.java

/*
 * Atricore IDBus
 *
 * Copyright (c) 2009, Atricore Inc.
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */

package org.atricore.idbus.idojos.dbidentitystore.test;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.xbean.spring.context.ClassPathXmlApplicationContext;
import org.springframework.context.ApplicationContext;
import org.springframework.jdbc.core.JdbcTemplate;
import org.junit.BeforeClass;
import org.junit.Test;
import org.atricore.idbus.idojos.dbidentitystore.JDBCIdentityStore;
import org.atricore.idbus.kernel.main.store.SimpleUserKey;
import org.atricore.idbus.kernel.main.authn.*;
import org.atricore.idbus.kernel.main.authn.scheme.UsernamePasswordCredentialProvider;
import org.atricore.idbus.kernel.main.authn.scheme.UsernameCredential;
import org.atricore.idbus.kernel.main.authn.scheme.PasswordCredential;

import javax.sql.DataSource;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.BufferedReader;

/**
 * User: <a href=mailto:ajadzinsky@atricore.org>ajadzinsky</a>
 * Date: Dec 3, 2008
 * Time: 6:33:23 PM
 */
public class JDBCIdentityStoreTest {
    private static final Log logger = LogFactory.getLog( JDBCIdentityStoreTest.class );

    protected static JDBCIdentityStore db;

    @BeforeClass
    public static void beforeTest () throws Exception {
        ApplicationContext ctxt = new ClassPathXmlApplicationContext( "org/atricore/idbus/idojos/dbidentitystore/test/hsdb-identity-store.xml" );
        db = (JDBCIdentityStore) ctxt.getBean( "dbStore" );

        JdbcTemplate template = new JdbcTemplate( (DataSource) ctxt.getBean( "dataSource" ) );
        createTables( template );
        insertData( template );
    }

    @Test
    public void testSelectUser() throws Exception {
        final SimpleUserKey uk = new SimpleUserKey( "user1" );
        BaseUser bu = db.loadUser( uk );
        assert bu != null : "can not load user " + uk.getId();
        assert bu.getName().equals( "User 1 full name" ) : "expected user name \"User 1 full name\" got \"" + bu.getName() + "\"";
        assert bu.getProperties().length == 3 : "expected 3 properties got " + bu.getProperties().length;
    }

    @Test
    public void testRolesByUser() throws Exception {
        final SimpleUserKey uk = new SimpleUserKey( "user1" );
        BaseRole[] brs = db.findRolesByUserKey( uk );
        assert brs.length == 2 : "expected 2 roles got " + brs.length;
        assert brs[0].getName().equals( "role1" ) : "expected role \"role1\" got " + brs[0].getName();
        assert brs[1].getName().equals( "role2" ) : "expected role \"role2\" got " + brs[1].getName();
    }

    @Test
    public void testLoadCredentials() throws Exception {
        final CredentialKey uk = new SimpleUserKey( "user1" );

        Credential[] cs = db.loadCredentials( uk, new UsernamePasswordCredentialProvider() );
        assert cs.length == 2 : "expected 2 credentials got " + cs.length;
        assert UsernameCredential.class.isInstance( cs[0] ) : "expected UsernameCredential class got " + cs[0].getClass().getName();
        assert PasswordCredential.class.isInstance( cs[1] ) : "expected PasswordCredential class got " + cs[1].getClass().getName();
    }


    private static void createTables ( JdbcTemplate template ) throws Exception {
        template.execute( getQueryFromFile( "sso.sql" ) );
    }

    private static void insertData ( JdbcTemplate template ) throws Exception {
        template.execute( getQueryFromFile( "sso-data.sql" ) );
    }

    private static String getQueryFromFile ( String resource ) throws Exception {
        InputStream is = JDBCIdentityStoreTest.class.getResourceAsStream( resource );
        InputStreamReader isr = new InputStreamReader( is );
        BufferedReader br = new BufferedReader( isr );

        String s = br.readLine();
        StringBuilder sb = new StringBuilder();
        while ( s != null ) {
            sb.append( s );
            s = br.readLine();
        }

        is.close();
        return sb.toString();
    }
}