com.abiquo.server.core.enterprise.UserGenerator.java Source code

Introduction

Here is the source code for com.abiquo.server.core.enterprise.UserGenerator.java

Source

/**
 * Abiquo community edition
 * cloud management application for hybrid clouds
 * Copyright (C) 2008-2010 - Abiquo Holdings S.L.
 *
 * This application 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 under
 * version 3 of the License
 *
 * 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 v.3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

package com.abiquo.server.core.enterprise;

import java.util.Calendar;
import java.util.Date;
import java.util.List;

import org.apache.commons.codec.digest.DigestUtils;

import com.abiquo.server.core.cloud.VirtualDatacenterGenerator;
import com.abiquo.server.core.common.DefaultEntityGenerator;
import com.abiquo.server.core.enterprise.User.AuthType;
import com.softwarementors.commons.test.SeedGenerator;
import com.softwarementors.commons.testng.AssertEx;

public class UserGenerator extends DefaultEntityGenerator<User> {

    EnterpriseGenerator enterpriseGenerator;

    RoleGenerator roleGenerator;

    VirtualDatacenterGenerator virtualDatacenterGenerator;

    public UserGenerator(final SeedGenerator seed) {
        super(seed);

        enterpriseGenerator = new EnterpriseGenerator(seed);

        roleGenerator = new RoleGenerator(seed);

        virtualDatacenterGenerator = new VirtualDatacenterGenerator(seed);

    }

    @Override
    public void assertAllPropertiesEqual(final User obj1, final User obj2) {
        AssertEx.assertPropertiesEqualSilent(obj1, obj2, User.NAME_PROPERTY, User.NICK_PROPERTY,
                User.LOCALE_PROPERTY, User.PASSWORD_PROPERTY, User.SURNAME_PROPERTY, User.ACTIVE_PROPERTY,
                User.EMAIL_PROPERTY);
    }

    @Override
    public User createUniqueInstance() {
        Enterprise enterprise = enterpriseGenerator.createUniqueInstance();
        return createInstance(enterprise);
    }

    public User createInstance(final Enterprise enterprise) {
        Role role = roleGenerator.createUniqueInstance();

        return createInstance(enterprise, role);
    }

    public User createInstance(final Role role) {
        Enterprise enterprise = enterpriseGenerator.createUniqueInstance();

        return createInstance(enterprise, role);
    }

    public User createInstance(final Enterprise enterprise, final Role role) {
        String password = newString(nextSeed(), 0, 255);

        return createInstance(enterprise, role, password);
    }

    public User createInstance(final AuthType authType) {
        String password = newString(nextSeed(), 0, 255);
        Enterprise enterprise = enterpriseGenerator.createUniqueInstance();
        Role role = roleGenerator.createUniqueInstance();
        return createInstance(enterprise, role, password);
    }

    public User createInstance(final AuthType authType, final Role role) {
        String password = newString(nextSeed(), 0, 255);
        Enterprise enterprise = enterpriseGenerator.createUniqueInstance();
        return createInstance(enterprise, role, password);
    }

    public User createInstance(final Enterprise enterprise, final Role role, final String password) {
        String name = newString(nextSeed(), 0, 255);
        String surname = newString(nextSeed(), 0, 255);
        String email = "abc@example.com";
        String nick = newString(nextSeed(), 0, 255);
        return createInstance(enterprise, role, password, name, surname, email, nick, User.AuthType.ABIQUO);
    }

    public User createInstance(final Enterprise enterprise, final Role role, final String password,
            final AuthType authType) {
        String name = newString(nextSeed(), 0, 255);
        String surname = newString(nextSeed(), 0, 255);
        String email = "abc@example.com";
        String nick = newString(nextSeed(), 0, 255);
        return createInstance(enterprise, role, password, name, surname, email, nick, authType);
    }

    public User createInstance(final Enterprise enterprise, final Role role, final String nick,
            final String password) {
        String name = newString(nextSeed(), 0, 255);
        String surname = newString(nextSeed(), 0, 255);
        String email = "abc@example.com";
        return createInstance(enterprise, role, password, name, surname, email, nick, User.AuthType.ABIQUO);
    }

    @Deprecated
    public User createInstance(final Enterprise enterprise, final Role role, final String password,
            final String name, final String surname, final String email, final String nick) {
        String locale = newString(nextSeed(), 0, 255);

        User user = new User(enterprise, role, name, surname, email, nick, DigestUtils.md5Hex(password), locale,
                User.AuthType.ABIQUO);

        user.setActive(1);
        return user;
    }

    public User createInstance(final Enterprise enterprise, final Role role, final String password,
            final String name, final String surname, final String email, final String nick,
            final AuthType authType) {
        String locale = newString(nextSeed(), 0, 255);

        User user = new User(enterprise, role, name, surname, email, nick, DigestUtils.md5Hex(password), locale,
                authType);

        user.setActive(1);
        return user;
    }

    public User createUserWithSession() {
        User user = createUniqueInstance();
        String key = newString(nextSeed(), 0, 255);

        Calendar cal = Calendar.getInstance();
        cal.add(Calendar.DAY_OF_YEAR, 5);

        Date expireDate = cal.getTime();

        Session session = new Session(user, key, expireDate, AuthType.ABIQUO.name());
        user.addSession(session);

        return user;
    }

    @Override
    public void addAuxiliaryEntitiesToPersist(final User entity, final List<Object> entitiesToPersist) {
        super.addAuxiliaryEntitiesToPersist(entity, entitiesToPersist);

        Enterprise enterprise = entity.getEnterprise();
        enterpriseGenerator.addAuxiliaryEntitiesToPersist(enterprise, entitiesToPersist);
        entitiesToPersist.add(enterprise);

        Role role = entity.getRole();
        roleGenerator.addAuxiliaryEntitiesToPersist(role, entitiesToPersist);
        entitiesToPersist.add(role);

    }

}