/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.ode.scheduler.simple;
import java.util.ArrayList;
import java.util.Date;
import java.util.Properties;
import java.util.concurrent.Callable;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.apache.ode.bpel.iapi.Scheduler;
import org.apache.ode.dao.scheduler.SchedulerDAOConnection;
import org.junit.Ignore;
import org.junit.Test;
/**
* @author Matthieu Riou <mriou@apache.org>
*/
public class RetriesTest extends SchedulerTestBase implements Scheduler.JobProcessor {
private static final Log __log = LogFactory.getLog(RetriesTest.class);
SimpleScheduler _scheduler;
ArrayList<Scheduler.JobInfo> _jobs;
ArrayList<Scheduler.JobInfo> _commit;
int _tried = 0;
@Override
protected Properties getProperties() {
Properties p = new Properties();
p.put("needed.Rollback", "true");
return p;
}
public void setUp() throws Exception {
super.setUp();
_scheduler = newScheduler("n1");
_jobs = new ArrayList<Scheduler.JobInfo>(100);
_commit = new ArrayList<Scheduler.JobInfo>(100);
}
public void tearDown() throws Exception {
_scheduler.shutdown();
super.tearDown();
}
public void testRetries() throws Exception {
// speed things up a bit to hit the right code paths
_scheduler.setNearFutureInterval(5000);
_scheduler.setImmediateInterval(1000);
_scheduler.start();
SchedulerDAOConnection conn = _factory.getConnection();
_txm.begin();
try {
_scheduler.schedulePersistedJob(newDetail("123"), new Date());
} finally {
_txm.commit();
conn.close();
}
Thread.sleep(20000);
assertEquals(8, _tried);
}
public void testExecTransaction() throws Exception {
final int[] tryCount = new int[1];
tryCount[0] = 0;
Callable<Void> transaction = new Callable<Void>() {
public Void call() throws Exception {
tryCount[0]++;
if( tryCount[0] < 3 ) {
throw new Exception("any");
} else {
return null;
}
}
};
_scheduler.execTransaction(transaction);
assertEquals(3, tryCount[0]);
}
public void onScheduledJob(Scheduler.JobInfo jobInfo) throws Scheduler.JobProcessorException {
__log.info("onScheduledJob " + jobInfo.jobName + ": " + jobInfo.retryCount);
_tried ++;
throw new Scheduler.JobProcessorException(jobInfo.retryCount < 1);
}
Scheduler.JobDetails newDetail(String x) {
Scheduler.JobDetails jd = new Scheduler.JobDetails();
jd.getDetailsExt().put("foo", x);
return jd;
}
private SimpleScheduler newScheduler(String nodeId) {
SimpleScheduler scheduler = new SimpleScheduler(nodeId, _factory, _txm, new Properties());
scheduler.setTransactionManager(_txm);
scheduler.setJobProcessor(this);
return scheduler;
}
}