/*
* JBoss, Home of Professional Open Source.
* Copyright 2013, Red Hat, Inc., and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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 software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.jboss.as.test.integration.batch.chunk;
import java.net.URL;
import java.util.Properties;
import java.util.PropertyPermission;
import java.util.concurrent.TimeUnit;
import javax.batch.operations.JobOperator;
import javax.batch.operations.NoSuchJobExecutionException;
import javax.batch.runtime.BatchRuntime;
import javax.batch.runtime.BatchStatus;
import javax.batch.runtime.JobExecution;
import javax.inject.Inject;
import org.jboss.arquillian.container.test.api.Deployment;
import org.jboss.arquillian.container.test.api.RunAsClient;
import org.jboss.arquillian.junit.Arquillian;
import org.jboss.arquillian.test.api.ArquillianResource;
import org.jboss.as.arquillian.container.ManagementClient;
import org.jboss.as.controller.client.helpers.Operations;
import org.jboss.as.test.integration.batch.common.AbstractBatchTestCase;
import org.jboss.as.test.integration.batch.common.CountingItemWriter;
import org.jboss.as.test.integration.batch.common.JobExecutionMarshaller;
import org.jboss.as.test.integration.batch.common.StartBatchServlet;
import org.jboss.as.test.shared.TimeoutUtil;
import org.jboss.remoting3.security.RemotingPermission;
import org.jboss.shrinkwrap.api.asset.StringAsset;
import org.jboss.shrinkwrap.api.spec.WebArchive;
import org.junit.Assert;
import org.junit.Test;
import org.junit.runner.RunWith;
import static org.jboss.as.test.shared.integration.ejb.security.PermissionUtils.createPermissionsXmlAsset;
/**
* @author <a href="mailto:jperkins@redhat.com">James R. Perkins</a>
*/
@RunWith(Arquillian.class)
public class ChunkPartitionTestCase extends AbstractBatchTestCase {
@ArquillianResource
private ManagementClient managementClient;
@Inject
private CountingItemWriter countingItemWriter;
@Deployment
public static WebArchive createDeployment() {
return createDefaultWar("batch-chunk-partition.war", ChunkPartitionTestCase.class.getPackage(), "chunkPartition.xml", "chunk-suspend.xml")
.addPackage(ChunkPartitionTestCase.class.getPackage())
.addClass(Operations.class)
.addAsResource(new StringAsset("Dependencies: org.jboss.dmr, org.jboss.as.controller, org.jboss.remoting3\n"), "META-INF/MANIFEST.MF")
.addAsManifestResource(createPermissionsXmlAsset(
new RemotingPermission("createEndpoint"),
new RemotingPermission("connect"),
new PropertyPermission("ts.timeout.factor", "read")
), "permissions.xml");
}
@RunAsClient
@Test
public void chunks(@ArquillianResource final URL url) throws Exception {
for (int i = 10; i >= 8; i--) {
final UrlBuilder builder = UrlBuilder.of(url, "start");
builder.addParameter(StartBatchServlet.JOB_XML_PARAMETER, "chunkPartition");
builder.addParameter("thread.count", i);
builder.addParameter("writer.sleep.time", 100);
final String result = performCall(builder.build());
final JobExecution jobExecution = JobExecutionMarshaller.unmarshall(result);
Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());
// final String exitStatus = stepExecution0.getExitStatus();
// Assert.assertTrue(exitStatus.startsWith("PASS"));
}
final UrlBuilder builder = UrlBuilder.of(url, "start");
builder.addParameter(StartBatchServlet.JOB_XML_PARAMETER, "chunkPartition");
builder.addParameter("thread.count", 1);
builder.addParameter("skip.thread.check", "true");
builder.addParameter("writer.sleep.time", 0);
final String result = performCall(builder.build());
final JobExecution jobExecution = JobExecutionMarshaller.unmarshall(result);
Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());
}
@Test
public void testSuspend() throws Exception {
try {
final Properties jobProperties = new Properties();
jobProperties.setProperty("reader.end", "10");
final JobOperator jobOperator = BatchRuntime.getJobOperator();
// Start the first job
long executionId = jobOperator.start("chunk-suspend", jobProperties);
JobExecution jobExecution = jobOperator.getJobExecution(executionId);
// Wait until the job is complete for a maximum of 5 seconds
waitForTermination(jobExecution, 5);
Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());
// Check that count
Assert.assertEquals(10, countingItemWriter.getWrittenItemSize());
// Suspend the server
managementClient.getControllerClient().execute(Operations.createOperation("suspend"));
// Submit another job which should be queued, should be safe with an InMemoryJobRepository (the default)
executionId = jobOperator.start("chunk-suspend", jobProperties);
// The job should not exist yet as the server is suspended
try {
jobOperator.getJobExecution(executionId);
} catch (NoSuchJobExecutionException expected) {
Assert.fail("Job should not exist as the server is suspended: " + executionId);
}
// Resume the server which should kick of queued jobs
managementClient.getControllerClient().execute(Operations.createOperation("resume"));
// Get the execution
jobExecution = jobOperator.getJobExecution(executionId);
// Wait until the job is complete for a maximum of 5 seconds
waitForTermination(jobExecution, 5);
Assert.assertEquals(BatchStatus.COMPLETED, jobExecution.getBatchStatus());
// Check that count
Assert.assertEquals(20, countingItemWriter.getWrittenItemSize());
} finally {
managementClient.getControllerClient().execute(Operations.createOperation("resume"));
}
}
private static void waitForTermination(final JobExecution jobExecution, final int timeout) {
long waitTimeout = TimeoutUtil.adjust(timeout * 1000);
long sleep = 100L;
while (true) {
switch (jobExecution.getBatchStatus()) {
case STARTED:
case STARTING:
case STOPPING:
try {
TimeUnit.MILLISECONDS.sleep(sleep);
} catch (InterruptedException e) {
throw new RuntimeException(e);
}
waitTimeout -= sleep;
sleep = Math.max(sleep / 2, 100L);
break;
default:
return;
}
if (waitTimeout <= 0) {
throw new IllegalStateException("Batch job did not complete within allotted time.");
}
}
}
}