PlanTest.java

/*******************************************************************************
 * Copyright 2006 - 2012 Vienna University of Technology,
 * Department of Software Technology and Interactive Systems, IFS
 * 
 * Licensed 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 eu.scape_project.planning.model;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import java.util.HashMap;

import org.junit.Test;

import eu.scape_project.planning.exception.PlanningException;

public class PlanTest {
    @Test
    public void initializeExperimentInfos_createNonExistingExperimentInfos() throws PlanningException {
        // set up
        Plan plan = new Plan();

        Alternative alt1 = new Alternative("alt1", "alt1");
        Alternative alt2 = new Alternative("alt2", "alt2");

        SampleObject obj1 = new SampleObject("ob1");
        SampleObject obj2 = new SampleObject("ob2");

        Experiment exp1 = new Experiment();
        Experiment exp2 = new Experiment();

        HashMap<SampleObject, DetailedExperimentInfo> hm1 = new HashMap<SampleObject, DetailedExperimentInfo>();
        hm1.put(obj1, null);
        hm1.put(obj2, null);
        HashMap<SampleObject, DetailedExperimentInfo> hm2 = new HashMap<SampleObject, DetailedExperimentInfo>();
        hm2.put(obj1, null);
        hm2.put(obj2, null);

        exp1.setDetailedInfo(hm1);
        exp2.setDetailedInfo(hm2);

        alt1.setExperiment(exp1);
        alt2.setExperiment(exp2);

        plan.getAlternativesDefinition().addAlternative(alt1);
        plan.getAlternativesDefinition().addAlternative(alt2);

        plan.getSampleRecordsDefinition().addRecord(obj1);
        plan.getSampleRecordsDefinition().addRecord(obj2);

        // execute
        plan.initializeExperimentInfos();

        // assert
        for (Alternative a : plan.getAlternativesDefinition().getAlternatives()) {
            for (SampleObject s : plan.getSampleRecordsDefinition().getRecords()) {
                DetailedExperimentInfo expInfo = a.getExperiment().getDetailedInfo().get(s);
                assertNotNull(expInfo);
                assertNotNull(expInfo.getProgramOutput());
            }
        }
    }

    @Test
    public void initializeExperimentInfos_ExistingExperimentInfosStayUnmodified() throws PlanningException {
        // set up
        Plan plan = new Plan();
        plan.setAlternativesDefinition(new AlternativesDefinition());
        plan.setSampleRecordsDefinition(new SampleRecordsDefinition());

        Alternative alt1 = new Alternative("alt1", "alt1");
        Alternative alt2 = new Alternative("alt2", "alt2");

        SampleObject obj1 = new SampleObject("ob1");
        SampleObject obj2 = new SampleObject("ob2");

        Experiment exp1 = new Experiment();
        Experiment exp2 = new Experiment();

        DetailedExperimentInfo detExpInfo = new DetailedExperimentInfo();
        detExpInfo.setId(88);
        detExpInfo.setProgramOutput("fixed");

        HashMap<SampleObject, DetailedExperimentInfo> hm1 = new HashMap<SampleObject, DetailedExperimentInfo>();
        hm1.put(obj1, detExpInfo);
        hm1.put(obj2, detExpInfo);
        HashMap<SampleObject, DetailedExperimentInfo> hm2 = new HashMap<SampleObject, DetailedExperimentInfo>();
        hm2.put(obj1, detExpInfo);
        hm2.put(obj2, detExpInfo);

        exp1.setDetailedInfo(hm1);
        exp2.setDetailedInfo(hm2);

        alt1.setExperiment(exp1);
        alt2.setExperiment(exp2);

        plan.getAlternativesDefinition().addAlternative(alt1);
        plan.getAlternativesDefinition().addAlternative(alt2);

        plan.getSampleRecordsDefinition().addRecord(obj1);
        plan.getSampleRecordsDefinition().addRecord(obj2);

        // execute
        plan.initializeExperimentInfos();

        // assert
        for (Alternative a : plan.getAlternativesDefinition().getAlternatives()) {
            for (SampleObject s : plan.getSampleRecordsDefinition().getRecords()) {
                DetailedExperimentInfo expInfo = a.getExperiment().getDetailedInfo().get(s);
                assertEquals(88, expInfo.getId());
                assertEquals("fixed", expInfo.getProgramOutput());
            }
        }
    }

}