org.activiti.crystalball.simulator.SimulateBottleneckTest.java Source code

Introduction

Here is the source code for org.activiti.crystalball.simulator.SimulateBottleneckTest.java

Source

package org.activiti.crystalball.simulator;

/*
 * #%L
 * simulator
 * %%
 * Copyright (C) 2012 - 2013 crystalball
 * %%
 * 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.
 * #L%
 */

import org.activiti.crystalball.generator.AbstractProcessEngineGraphGenerator;
import org.activiti.crystalball.simulator.impl.StartProcessEventHandler;
import org.activiti.crystalball.simulator.impl.persistence.entity.ResultEntity;
import org.activiti.engine.ProcessEngine;
import org.activiti.engine.RepositoryService;
import org.apache.commons.io.FileUtils;
import org.junit.Test;
import org.springframework.context.support.AbstractApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;

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

import static org.junit.Assert.assertTrue;

public class SimulateBottleneckTest {
    protected static final String tempDir = System.getProperty("tempDir", "target");
    protected static final String LIVE_DB = tempDir + "/live-SimulateBottleneckTest";
    protected static final String PROCESS_KEY = "parallelusertaskprocess";

    @Test
    public void testOK() throws Exception {
        System.setProperty("liveDB", LIVE_DB);
        System.setProperty("_SIM_DB_PATH",
                tempDir + "/simulationRunDB-SimulateBottleNeck-OK-" + Thread.currentThread().getId());

        FileUtils.copyFile(new File(LIVE_DB + ".h2.db"), new File(
                tempDir + "/simulationRunDB-SimulateBottleNeck-OK-" + Thread.currentThread().getId() + ".h2.db"));

        ClassPathXmlApplicationContext appContext = new ClassPathXmlApplicationContext(
                "/org/activiti/crystalball/simulator/SimEngine-simulateBottleneck-h2-context.xml");
        ProcessEngine processEngine = (ProcessEngine) appContext.getBean("simProcessEngine");

        runSimulation(appContext, tempDir + "/SimulateBottleneckTest-1.png");

        processEngine.close();
        appContext.close();

        File expected = new File(System.getProperty("baseDir", ".")
                + "/src/test/resources/org/activiti/crystalball/simulator/simulateBottleneckTest-1.png");
        File generated = new File(tempDir + "/SimulateBottleneckTest-1.png");
        assertTrue(FileUtils.contentEquals(expected, generated));
        // delete database file
        File f = new File(System.getProperty("_SIM_DB_PATH") + ".h2.db");
        if (!f.delete())
            System.err.println("unable to delete file");
    }

    /**
     * results differs - that's why this test is ignored - possible bug. 
     * @throws Exception 
     */
    @Test
    public void testUser2OverLoaded() throws Exception {
        System.setProperty("liveDB", LIVE_DB);
        System.setProperty("_SIM_DB_PATH",
                tempDir + "/simulationRunDB-SimulateBottleNeck-Overload-" + Thread.currentThread().getId());

        FileUtils.copyFile(new File(LIVE_DB + ".h2.db"), new File(tempDir
                + "/simulationRunDB-SimulateBottleNeck-Overload-" + Thread.currentThread().getId() + ".h2.db"));

        ClassPathXmlApplicationContext appContext = new ClassPathXmlApplicationContext(
                "/org/activiti/crystalball/simulator/SimEngine-simulateBottleneck-h2-context.xml");
        ProcessEngine processEngine = (ProcessEngine) appContext.getBean("simProcessEngine");

        // increase frequency of process starts to 45 mins (default is 1 hour)
        StartProcessEventHandler startProcessEventHandler = appContext.getBean(StartProcessEventHandler.class);
        startProcessEventHandler.setPeriod(2700000);

        runSimulation(appContext, tempDir + "/SimulateBottleneckTest-2.png");

        processEngine.close();
        appContext.close();

        File expected = new File(System.getProperty("baseDir", ".")
                + "/src/test/resources/org/activiti/crystalball/simulator/simulateBottleneckTest-2.png");
        File generated = new File(tempDir + "/SimulateBottleneckTest-2.png");
        assertTrue(FileUtils.contentEquals(expected, generated));
        // delete database file

        File f = new File(System.getProperty("_SIM_DB_PATH") + ".h2.db");
        if (!f.delete())
            System.err.println("unable to delete file");

    }

    /**
     * run simulation for 30 days and generate report
     * 
     * @param appContext
     * @throws Exception 
     */
    protected void runSimulation(AbstractApplicationContext appContext, String generatedImage) throws Exception {

        SimulationRun simRun = (SimulationRun) appContext.getBean(SimulationRun.class);

        Calendar c = Calendar.getInstance();
        c.clear();
        c.set(2013, 1, 1);
        Date startDate = c.getTime();
        c.add(Calendar.DAY_OF_YEAR, 30);
        Date finishDate = c.getTime();
        // run simulation for 30 days
        @SuppressWarnings("unused")
        List<ResultEntity> resultEventList = simRun.execute(startDate, finishDate);

        AbstractProcessEngineGraphGenerator generator = (AbstractProcessEngineGraphGenerator) appContext
                .getBean("reportGenerator");

        RepositoryService simRepositoryService = (RepositoryService) appContext.getBean("simRepositoryService");

        String processDefinitionId = simRepositoryService.createProcessDefinitionQuery()
                .processDefinitionKey(PROCESS_KEY).singleResult().getId();

        generator.generateReport(processDefinitionId, startDate, finishDate, generatedImage);

    }

}