testuserdata.java

一套JAVA的CA证书签发系统.

/*************************************************************************
 *                                                                       *
 *  EJBCA: The OpenSource Certificate Authority                          *
 *                                                                       *
 *  This software 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 any later version.                    *
 *                                                                       *
 *  See terms of license at gnu.org.                                     *
 *                                                                       *
 *************************************************************************/
 
package se.anatom.ejbca.ra.junit;

import java.util.Collection;
import java.util.Date;
import java.util.Iterator;
import java.util.Random;

import javax.naming.Context;
import javax.naming.NamingException;

import junit.framework.TestCase;

import org.apache.log4j.Logger;

import se.anatom.ejbca.SecConst;
import se.anatom.ejbca.log.Admin;
import se.anatom.ejbca.ra.IUserAdminSessionHome;
import se.anatom.ejbca.ra.IUserAdminSessionRemote;
import se.anatom.ejbca.ra.UserAdminData;
import se.anatom.ejbca.ra.UserDataHome;
import se.anatom.ejbca.ra.UserDataPK;
import se.anatom.ejbca.ra.UserDataRemote;

/** Tests the UserData entity bean and some parts of UserAdminSession.
 *
 * @version $Id: TestUserData.java,v 1.20 2004/05/13 09:11:07 anatom Exp $
 */
public class TestUserData extends TestCase {

    private static Logger log = Logger.getLogger(TestUserData.class);
    private static Context ctx;
    private static UserDataHome home;
    private static IUserAdminSessionRemote usersession;
    private static String username;
    private static String username1;
    private static String pwd;
    private static String pwd1;
    private static int caid;
    private static Admin admin = null;

    /**
     * Creates a new TestUserData object.
     *
     * @param name DOCUMENT ME!
     */
    public TestUserData(String name) {
        super(name);
    }

    protected void setUp() throws Exception {

        log.debug(">setUp()");
        ctx = getInitialContext();

        Object obj = ctx.lookup("UserData");
        home = (UserDataHome) javax.rmi.PortableRemoteObject.narrow(obj, UserDataHome.class);
        caid = "CN=TEST".hashCode();

        obj = ctx.lookup("UserAdminSession");        
        IUserAdminSessionHome userhome = (IUserAdminSessionHome) javax.rmi.PortableRemoteObject.narrow(obj, IUserAdminSessionHome.class);        
        usersession = userhome.create();
        admin = new Admin(Admin.TYPE_INTERNALUSER);

        log.debug("<setUp()");
    }

    protected void tearDown() throws Exception {
    }

    private Context getInitialContext() throws NamingException {
        log.debug(">getInitialContext");

        Context ctx = new javax.naming.InitialContext();
        log.debug("<getInitialContext");

        return ctx;
    }

    private String genRandomUserName() throws Exception {
        // Gen random user
        Random rand = new Random(new Date().getTime()+4711);
        String username = "";
        for (int i = 0; i < 6; i++) {
            int randint = rand.nextInt(9);
            username += (new Integer(randint)).toString();
        }
        log.debug("Generated random username: username =" + username);

        return username;
    } // genRandomUserName

    private String genRandomPwd() throws Exception {
        // Gen random pwd
        Random rand = new Random(new Date().getTime()+4812);
        String password = "";

        for (int i = 0; i < 8; i++) {
            int randint = rand.nextInt(9);
            password += (new Integer(randint)).toString();
        }

        log.debug("Generated random pwd: password=" + password);

        return password;
    } // genRandomPwd


    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test01CreateNewUser() throws Exception {
        log.debug(">test01CreateNewUser()");
        UserDataRemote data1=null;
        username = genRandomUserName();
        pwd = genRandomPwd();
        data1 = home.create(username, pwd, "C=SE, O=AnaTom, CN="+username, caid);
        assertNotNull("Error creating", data1);
        data1.setEndEntityProfileId(SecConst.EMPTY_ENDENTITYPROFILE);
        data1.setCertificateProfileId(SecConst.CERTPROFILE_FIXED_ENDUSER);
        log.debug("created it!");
        log.debug("<test01CreateNewUser()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test02LookupAndChangeUser() throws Exception {
        log.debug(">test02LookupAndChangeUser()");

        UserDataPK pk = new UserDataPK(username);
        log.debug("pk="+ pk);
        UserDataRemote data2 = home.findByPrimaryKey(pk);
        log.debug("found by key! ="+ data2);
        log.debug("username="+data2.getUsername());
        assertTrue( "wrong username", data2.getUsername().equals(username) );
        log.debug("subject="+data2.getSubjectDN());
        assertTrue( "wrong DN", data2.getSubjectDN().indexOf(username) != -1);
        log.debug("email="+data2.getSubjectEmail());
        assertNull( "wrong email", data2.getSubjectEmail());
        log.debug("status="+data2.getStatus());
        assertTrue( "wrong status", data2.getStatus() == UserDataRemote.STATUS_NEW );
        log.debug("type="+data2.getType());
        assertTrue( "wrong type", data2.getType() == SecConst.USER_INVALID);
        log.debug("password foo123 returned " + data2.comparePassword("foo123"));
        assertTrue( "wrong pwd (foo123 works)", data2.comparePassword("foo123")==false);
        log.debug("password "+pwd+" returned " + data2.comparePassword(pwd));
        assertTrue( "wrong pwd "+pwd, data2.comparePassword(pwd));
        data2.setStatus(UserDataRemote.STATUS_GENERATED);
        data2.setType(SecConst.USER_ENDUSER);
        data2.setPassword("foo123");
        data2.setSubjectEmail(username+"@anatom.se");
        log.debug("Changed it");
        log.debug("<test02LookupAndChangeUser()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test03LookupChangedUser() throws Exception {
        log.debug(">test03LookupChangedUser()");

        UserDataPK pk = new UserDataPK(username);
        UserDataRemote data = home.findByPrimaryKey(pk);
        log.debug("found by key! ="+ data);
        log.debug("username="+data.getUsername());
        assertTrue( "wrong username", data.getUsername().equals(username) );
        log.debug("subject="+data.getSubjectDN());
        assertTrue( "wrong DN", data.getSubjectDN().indexOf(username)!=-1 );
        log.debug("email="+data.getSubjectEmail());
        assertNotNull("Email should not be null now.", data.getSubjectEmail());
        assertTrue( "wrong email", data.getSubjectEmail().equals(username+"@anatom.se"));
        log.debug("status="+data.getStatus());
        assertTrue( "wrong status", data.getStatus() == UserDataRemote.STATUS_GENERATED );
        log.debug("type="+data.getType());
        assertTrue( "wrong type", data.getType() == SecConst.USER_ENDUSER);
        log.debug("password foo123 returned " + data.comparePassword("foo123"));
        assertTrue( "wrong pwd foo123", data.comparePassword("foo123"));
        log.debug("password "+pwd+" returned " + data.comparePassword(pwd));
        assertTrue( "wrong pwd ("+pwd+" works)", data.comparePassword(pwd)==false);
        data.setOpenPassword("foo234");
        log.debug("<test03LookupChangedUser()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test03LookupChangedUser2() throws Exception {
        log.debug(">test03LookupChangedUser2()");

        UserDataPK pk = new UserDataPK(username);
        UserDataRemote data = home.findByPrimaryKey(pk);
        log.debug("found by key! ="+ data);
        log.debug("username="+data.getUsername());
        assertTrue( "wrong username", data.getUsername().equals(username) );
        log.debug("subject="+data.getSubjectDN());
        assertTrue( "wrong DN", data.getSubjectDN().indexOf(username)!=-1 );
        log.debug("email="+data.getSubjectEmail());
        assertNotNull("Email should not be null now.", data.getSubjectEmail());
        assertTrue( "wrong email", data.getSubjectEmail().equals(username+"@anatom.se"));
        log.debug("status="+data.getStatus());
        assertTrue( "wrong status", data.getStatus() == UserDataRemote.STATUS_GENERATED );
        log.debug("type="+data.getType());
        assertTrue( "wrong type", data.getType() == SecConst.USER_ENDUSER);
        log.debug("password foo234 returned " + data.comparePassword("foo234"));
        assertTrue( "wrong pwd foo234", data.comparePassword("foo234"));
        assertEquals( "wrong clear pwd foo234", data.getClearPassword(), "foo234");
        log.debug("password "+pwd+" returned " + data.comparePassword(pwd));
        assertTrue( "wrong pwd ("+pwd+" works)", data.comparePassword(pwd)==false);
        data.setOpenPassword("foo234");
        log.debug("<test03LookupChangedUser2()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test04CreateNewUser() throws Exception {
        log.debug(">test04CreateNewUser()");
        UserDataRemote data4=null;
        username1 = genRandomUserName();
        pwd1 = genRandomPwd();
        data4 = home.create(username1, pwd1, "C=SE, O=AnaTom, CN="+username, caid);
        assertNotNull("Error creating", data4);
        data4.setEndEntityProfileId(SecConst.EMPTY_ENDENTITYPROFILE);
        data4.setCertificateProfileId(SecConst.CERTPROFILE_FIXED_ENDUSER);
        log.debug("created it again!");
        log.debug("<test04CreateNewUser()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test05ListNewUser() throws Exception {
        log.debug(">test05ListNewUser()");

        Object obj1 = ctx.lookup("UserAdminSession");
        IUserAdminSessionHome adminhome = (IUserAdminSessionHome) javax.rmi.PortableRemoteObject.narrow(obj1, IUserAdminSessionHome.class);
        IUserAdminSessionRemote admin = adminhome.create();
        Collection coll = admin.findAllUsersByStatus(new Admin(Admin.TYPE_INTERNALUSER), UserDataRemote.STATUS_NEW);
        Iterator iter = coll.iterator();
        while (iter.hasNext())
        {

            UserAdminData data = (UserAdminData)iter.next();
            log.debug("New user: "+data.getUsername()+", "+data.getDN()+", "+data.getEmail()+", "+data.getStatus()+", "+data.getType());
            admin.setUserStatus(new Admin(Admin.TYPE_INTERNALUSER), data.getUsername(), UserDataRemote.STATUS_GENERATED);
        }

        Collection coll1 = admin.findAllUsersByStatus(new Admin(Admin.TYPE_INTERNALUSER), UserDataRemote.STATUS_NEW);
        assertTrue("found NEW users though there should be none!", coll1.isEmpty());
        log.debug("<test05ListNewUser()");
    }

    /**
     * DOCUMENT ME!
     *
     * @throws Exception DOCUMENT ME!
     */
    public void test06RemoveUser() throws Exception {
        log.debug(">test06RemoveUser()");

        UserDataPK pk = new UserDataPK(username);
        home.remove(pk);
        pk.username = username1;
        home.remove(pk);
        log.debug("Removed it!");
        log.debug("<test06RemoveUser()");
    }
}