/**
 * 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 org.candlepin.test;

import org.candlepin.controller.PoolManager;
import org.candlepin.model.Consumer;
import org.candlepin.model.Entitlement;
import org.candlepin.model.Pool;
import org.candlepin.model.PoolQuantity;
import org.candlepin.policy.ValidationResult;
import org.candlepin.policy.js.entitlement.Enforcer;
import org.candlepin.policy.js.entitlement.PreUnbindHelper;

import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;


/**
 * EnforcerForTesting
 */
public class EnforcerForTesting implements Enforcer {

    @Override
    public void postEntitlement(PoolManager manager, Consumer consumer, Map<String, Entitlement> ent,
        List<Pool> subPoolsForStackIds, boolean isUpdate, Map<String, PoolQuantity> poolQuantityMap) {
    }

    @Override
    public ValidationResult preEntitlement(Consumer consumer, Pool enitlementPool,
        Integer quantity) {
        return new ValidationResult();
    }

    @Override
    public ValidationResult preEntitlement(Consumer consumer, Pool enitlementPool,
        Integer quantity, CallerType caller) {
        return new ValidationResult();
    }

    public PreUnbindHelper preUnbind(Consumer consumer, Pool entitlementPool) {
        return new PreUnbindHelper(null);
    }

    @Override
    public void postUnbind(Consumer consumer, PoolManager poolManager,
        Entitlement ent) {
    }

    @Override
    public ValidationResult update(Consumer consumer, Entitlement entitlement, Integer change) {
        return new ValidationResult();
    }

    @Override
    public List<Pool> filterPools(Consumer consumer, List<Pool> pools,
        boolean showAll) {
        return pools;
    }

    @Override
    public Map<String, ValidationResult> preEntitlement(Consumer consumer,
        Collection<PoolQuantity> entitlementPoolQuantities, CallerType caller) {
        Map<String, ValidationResult> result = new HashMap<String, ValidationResult>();
        for (PoolQuantity pool : entitlementPoolQuantities) {
            result.put(pool.getPool().getId(), preEntitlement(consumer, pool.getPool(), pool.getQuantity()));
        }
        return result;
    }

    @Override
    public Map<String, ValidationResult> preEntitlement(Consumer consumer, Consumer host,
        Collection<PoolQuantity> entitlementPoolQuantities, CallerType caller) {
        return preEntitlement(consumer, entitlementPoolQuantities, caller);
    }
}