com.redhat.rhn.domain.server.test.HostBuilder.java Source code

Introduction

Here is the source code for com.redhat.rhn.domain.server.test.HostBuilder.java

Source

/**
 * Copyright (c) 2009--2012 Red Hat, Inc.
 *
 * This software is licensed to you under the GNU General Public License,
 * version 2 (GPLv2). There is NO WARRANTY for this software, express or
 * implied, including the implied warranties of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. You should have received a copy of GPLv2
 * along with this software; if not, see
 * http://www.gnu.org/licenses/old-licenses/gpl-2.0.txt.
 *
 * Red Hat trademarks are not licensed under GPLv2. No permission is
 * granted to use or replicate Red Hat trademarks that are incorporated
 * in this software or its documentation.
 */
package com.redhat.rhn.domain.server.test;

import com.redhat.rhn.common.hibernate.HibernateFactory;
import com.redhat.rhn.domain.server.Server;
import com.redhat.rhn.domain.server.VirtualInstance;
import com.redhat.rhn.domain.user.User;
import com.redhat.rhn.testing.ServerTestUtils;

import org.hibernate.Session;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

/**
 * HostBuilder is a class based on the GoF Builder pattern that constructs systems that are
 * virtual hosts. That is, systems with one of the virtualization entitlements.
 * @version $Rev$
 */
public class HostBuilder {

    private User owner;
    private Server host;

    public HostBuilder(User theOwner) {
        owner = theOwner;
    }

    /**
     * This is the final step in building or compiling a host. The host and its guests (if
     * there are any) will be persisted, flushed, and evicted from the current hibernate
     * session.
     *
     * <br/><br/>
     *
     * The builder does not maintain a reference to a host once it is built; so, calling
     * <code>build</code> successive times will simply return <code>null</code>. One of the
     * <i>create</i> methods must be called before every invocation of this method.
     *
     * @return The built host whose state will have persisted and synhronized with the
     * database. The host and its guests will have been removed from the hibernate session
     * cache.
     */
    public Server build() {
        if (host == null) {
            return null;
        }

        Server compiledHost;
        Session session = HibernateFactory.getSession();

        session.flush();
        session.evict(host);

        for (Iterator iterator = host.getGuests().iterator(); iterator.hasNext();) {
            session.evict(iterator.next());
        }

        compiledHost = host;
        host = null;

        return compiledHost;
    }

    /**
     * Creates a Server with the Virtualization Host entitlement.
     *
     * @return This builder
     *
     * @throws Exception if an error occurs
     */
    public HostBuilder createVirtHost() throws Exception {
        host = ServerTestUtils.createVirtHostWithGuests(owner, 0);
        return this;
    }

    /**
     * Creates a Server with the Virtualization Platform entitlement.
     *
     * @return This builder
     *
     * @throws Exception if an error occurs
     */
    public HostBuilder createVirtPlatformHost() throws Exception {
        host = ServerTestUtils.createVirtPlatformHost(owner);
        return this;
    }

    /**
     * Creates a server without any of the virtualization entitlements. Note that a non-virt
     * host is useful in those situations in which a guest consumes physical entitlements.
     *
     * @return This builder
     *
     * @throws Exception if an error occurs
     */
    public HostBuilder createNonVirtHost() throws Exception {
        host = ServerTestUtils.createTestSystem(owner);
        return this;
    }

    /**
     * Creates the specified number of guests for the host under construction. Each guest
     * will be registered and therefore have an associated system.
     *
     * @param numberOfGuests The number of guests to create
     *
     * @return This builder
     *
     * @throws Exception if an error occurs
     */
    public HostBuilder withGuests(int numberOfGuests) throws Exception {
        createGuests(owner, numberOfGuests, true);
        return this;
    }

    /**
     * Creates the specified number of guests for the host under construction. None of the
     * guests will be registered and therefore will not have an associated system.
     *
     * @param numberOfGuests The number of guests to create
     *
     * @return This builder
     *
     * @throws Exception if an error occurs
     */
    public HostBuilder withUnregisteredGuests(int numberOfGuests) throws Exception {
        createGuests(owner, numberOfGuests, false);
        return this;
    }

    /**
     * Creates the specified number of guests for the host under construction. Each guest
     * will be registered and therefore have an associated system.
     *
     * @param guests a map of users to the number of guests to create owned by that user.
     *              This is useful when you want the guests to be in orgs different to
     *               what the host .
     *
     * @return This builder
     * @throws Exception if an error occurs
     */

    public HostBuilder withGuests(Map<User, Integer> guests, boolean register) throws Exception {
        for (User u : guests.keySet()) {
            createGuests(u, guests.get(u), register);
        }
        return this;
    }

    private List<VirtualInstance> createGuests(User user, int numberOfGuests, boolean register) throws Exception {
        List<VirtualInstance> guests = new LinkedList<VirtualInstance>();
        for (int i = 0; i < numberOfGuests; i++) {
            VirtualInstanceManufacturer vim = new VirtualInstanceManufacturer(user);

            VirtualInstance vi = null;

            if (register) {
                vi = vim.newRegisteredGuestWithoutHost();
            } else {
                vi = vim.newUnregisteredGuest();
            }

            if (host != null) {
                vi.setConfirmed(1L);
                host.addGuest(vi);
            } else {
                vi.setConfirmed(0L);
            }
            guests.add(vi);
        }
        return guests;
    }

    public List<VirtualInstance> withOrphanedGuests(int numberOfGuests) throws Exception {
        return createGuests(owner, numberOfGuests, true);
    }

}