MockCatalog.java

/*
 * Copyright 2010-2011 Ning, Inc.
 *
 * Ning licenses this file to you under the Apache License, version 2.0
 * (the "License"); you may not use this file except in compliance with the
 * License.  You may obtain a copy of the License at:
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

package com.ning.billing.catalog;

import com.ning.billing.catalog.api.BillingPeriod;
import com.ning.billing.catalog.api.PhaseType;
import com.ning.billing.catalog.api.ProductCategory;
import com.ning.billing.catalog.rules.CaseCancelPolicy;
import com.ning.billing.catalog.rules.CaseChangePlanAlignment;
import com.ning.billing.catalog.rules.CaseChangePlanPolicy;
import com.ning.billing.catalog.rules.CaseCreateAlignment;
import com.ning.billing.catalog.rules.PlanRules;

import java.util.Date;

public class MockCatalog extends StandaloneCatalog {
    private static final String[] PRODUCT_NAMES = new String[]{ "TestProduct1", "TestProduct2", "TestProduct3"};
    
    public MockCatalog() {
        setEffectiveDate(new Date());
        setProducts(MockProduct.createAll());
        setPlans(MockPlan.createAll());
        populateRules();
        populatePriceLists();
    }
    
    public void populateRules(){
        setPlanRules(new PlanRules());
    }

    public void setRules( 
            CaseChangePlanPolicy[] caseChangePlanPolicy,
            CaseChangePlanAlignment[] caseChangePlanAlignment,
            CaseCancelPolicy[] caseCancelPolicy,
            CaseCreateAlignment[] caseCreateAlignment
            ){
        
    }

    public void populatePriceLists() {
        DefaultPlan[] plans = getCurrentPlans();
        
        DefaultPriceList[] priceList = new DefaultPriceList[plans.length - 1];
        for(int i = 1; i < plans.length; i++) {
            priceList[i-1] = new DefaultPriceList(new DefaultPlan[]{plans[i]},plans[i].getName()+ "-pl");
        }
        
        DefaultPriceListSet set = new DefaultPriceListSet(new PriceListDefault(new DefaultPlan[]{plans[0]}),priceList);
        setPriceLists(set);
    }
    
    public String[] getProductNames() {
        return PRODUCT_NAMES;
    }


    
}