/*
* This is eMonocot, a global online biodiversity information resource.
*
* Copyright © 2011–2015 The Board of Trustees of the Royal Botanic Gardens, Kew and The University of Oxford
*
* eMonocot is free software: you can redistribute it and/or modify it under the terms of the
* GNU Affero General Public License as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* eMonocot 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 Affero General Public License for more details.
*
* The complete text of the GNU Affero General Public License is in the source repository as the file
* ‘COPYING’. It is also available from <http://www.gnu.org/licenses/>.
*/
package org.emonocot.job.taxonmatch;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.emonocot.model.Taxon;
import org.emonocot.persistence.hibernate.SolrIndexingListener;
import org.hibernate.Session;
import org.hibernate.SessionFactory;
import org.hibernate.Transaction;
import org.joda.time.DateTime;
import org.joda.time.base.BaseDateTime;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.batch.core.Job;
import org.springframework.batch.core.JobExecution;
import org.springframework.batch.core.JobParameter;
import org.springframework.batch.core.JobParameters;
import org.springframework.batch.core.JobParametersInvalidException;
import org.springframework.batch.core.configuration.JobLocator;
import org.springframework.batch.core.launch.JobLauncher;
import org.springframework.batch.core.launch.NoSuchJobException;
import org.springframework.batch.core.repository.JobExecutionAlreadyRunningException;
import org.springframework.batch.core.repository.JobInstanceAlreadyCompleteException;
import org.springframework.batch.core.repository.JobRestartException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.core.io.ClassPathResource;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.annotation.DirtiesContext.ClassMode;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
/**
*
* @author ben
*
*/
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration({
"/META-INF/spring/batch/jobs/taxonMatch.xml",
"/META-INF/spring/applicationContext-integration.xml",
"/META-INF/spring/applicationContext-test.xml" })
@DirtiesContext(classMode = ClassMode.AFTER_CLASS)
public class TaxonMatchIntegrationTest {
private Logger logger = LoggerFactory.getLogger(TaxonMatchIntegrationTest.class);
@Autowired
private JobLocator jobLocator;
@Autowired
@Qualifier("jobLauncher")
private JobLauncher jobLauncher;
@Autowired
private SessionFactory sessionFactory;
@Autowired
private SolrIndexingListener solrIndexingListener;
/**
* 1288569600 in unix time.
*/
private static final BaseDateTime PAST_DATETIME = new DateTime(2010, 11, 1, 9, 0, 0, 0);
/**
*
* @throws IOException
* if a temporary file cannot be created.
* @throws NoSuchJobException
* if SpeciesPageHarvestingJob cannot be located
* @throws JobParametersInvalidException
* if the job parameters are invalid
* @throws JobInstanceAlreadyCompleteException
* if the job has already completed
* @throws JobRestartException
* if the job cannot be restarted
* @throws JobExecutionAlreadyRunningException
* if the job is already running
*/
@Test
public final void testMatchTaxa() throws IOException,
NoSuchJobException, JobExecutionAlreadyRunningException,
JobRestartException, JobInstanceAlreadyCompleteException,
JobParametersInvalidException {
Session session = sessionFactory.openSession();
Transaction tx = session.beginTransaction();
List<Taxon> taxa = session.createQuery("from Taxon as taxon").list();
solrIndexingListener.indexObjects(taxa);
tx.commit();
ClassPathResource input = new ClassPathResource("/org/emonocot/job/taxonmatch/input.csv");
Map<String, JobParameter> parameters = new HashMap<String, JobParameter>();
parameters.put("assume.accepted.matches", new JobParameter(Boolean.TRUE.toString()));
parameters.put("input.file", new JobParameter(input.getFile().getAbsolutePath()));
parameters.put("output.file", new JobParameter(File.createTempFile("output", "csv").getAbsolutePath()));
JobParameters jobParameters = new JobParameters(parameters);
Job taxonMatchingJob = jobLocator.getJob("TaxonMatching");
assertNotNull("TaxonMatching must not be null", taxonMatchingJob);
JobExecution jobExecution = jobLauncher.run(taxonMatchingJob, jobParameters);
assertEquals("The job should complete successfully",jobExecution.getExitStatus().getExitCode(),"COMPLETED");
FileReader file = new FileReader(jobParameters.getParameters().get("output.file").getValue().toString());
BufferedReader reader = new BufferedReader(file);
assertNotNull("There should be an output file", reader);
String ln;
while ((ln = reader.readLine()) != null) {
logger.debug(ln);
}
}
}