AnalyzerJobPartitioningTest.java example

Explorer
AnalyzerBeans-master
/**
 * AnalyzerBeans
 * Copyright (C) 2014 Neopost - Customer Information Management
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 */
package org.eobjects.analyzer.test.full.scenarios;

import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;

import junit.framework.TestCase;

import org.eobjects.analyzer.beans.valuedist.ValueDistributionAnalyzerResult;
import org.eobjects.analyzer.configuration.AnalyzerBeansConfiguration;
import org.eobjects.analyzer.configuration.AnalyzerBeansConfigurationImpl;
import org.eobjects.analyzer.connection.DatastoreCatalog;
import org.eobjects.analyzer.connection.DatastoreCatalogImpl;
import org.eobjects.analyzer.descriptors.ClasspathScanDescriptorProvider;
import org.eobjects.analyzer.descriptors.DescriptorProvider;
import org.eobjects.analyzer.job.AnalysisJob;
import org.eobjects.analyzer.job.JaxbJobReader;
import org.eobjects.analyzer.job.builder.AnalysisJobBuilder;
import org.eobjects.analyzer.job.runner.AnalysisResultFuture;
import org.eobjects.analyzer.job.runner.AnalysisRunner;
import org.eobjects.analyzer.job.runner.AnalysisRunnerImpl;
import org.eobjects.analyzer.result.AnalyzerResult;
import org.eobjects.analyzer.result.CrosstabResult;
import org.eobjects.analyzer.result.renderer.CrosstabTextRenderer;
import org.eobjects.analyzer.test.TestHelper;

public class AnalyzerJobPartitioningTest extends TestCase {

	public void testScenario() throws Exception {
		DatastoreCatalog datastoreCatalog = new DatastoreCatalogImpl(
				TestHelper.createSampleDatabaseDatastore("my database"));
		DescriptorProvider descriptorProvider = new ClasspathScanDescriptorProvider()
				.scanPackage("org.eobjects.analyzer.beans", true);
		AnalyzerBeansConfiguration conf = new AnalyzerBeansConfigurationImpl()
				.replace(datastoreCatalog).replace(descriptorProvider);

		AnalysisRunner runner = new AnalysisRunnerImpl(conf);

		AnalysisJobBuilder jobBuilder = new JaxbJobReader(conf)
				.create(new File(
						"src/test/resources/example-job-partitioning.xml"));

		AnalysisJob analysisJob = jobBuilder.toAnalysisJob();
		assertEquals(6, analysisJob.getAnalyzerJobs().size());

		AnalysisResultFuture resultFuture = runner.run(analysisJob);
		assertTrue(resultFuture.isSuccessful());

		List<AnalyzerResult> results = resultFuture.getResults();

		int vdResults = 0;
		List<CrosstabResult> saResults = new ArrayList<CrosstabResult>();

		for (AnalyzerResult analyzerResult : results) {
			if (analyzerResult instanceof ValueDistributionAnalyzerResult) {
				vdResults++;
			} else if (analyzerResult instanceof CrosstabResult) {
				saResults.add((CrosstabResult) analyzerResult);
			} else {
				fail("Unexpected result: " + analyzerResult);
			}
		}

		assertEquals(4, vdResults);
		assertEquals(2, saResults.size());

		final int dimensionIndex = saResults.get(0).getCrosstab()
				.getDimensionIndex("Column");

		Collections.sort(saResults, new Comparator<CrosstabResult>() {
			@Override
			public int compare(CrosstabResult o1, CrosstabResult o2) {
				int count1 = o1.getCrosstab().getDimension(dimensionIndex)
						.getCategoryCount();
				int count2 = o2.getCrosstab().getDimension(dimensionIndex)
						.getCategoryCount();
				return count1 - count2;
			}
		});

		String[] resultLines1 = new CrosstabTextRenderer().render(
				saResults.get(0)).split("\n");
		assertEquals("                                      CUSTOMERNAME ",
				resultLines1[0]);
		assertEquals("Row count                                      122 ",
				resultLines1[1]);

		String[] resultLines2 = new CrosstabTextRenderer().render(
				saResults.get(1)).split("\n");
		assertEquals(
				"                                      FIRSTNAME  LASTNAME     EMAIL ",
				resultLines2[0]);
		assertEquals(
				"Row count                                    23        23        23 ",
				resultLines2[1]);
	}
}