/* * Copyright 2006-2013 the original author or authors. * * 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. */ package org.springframework.batch.core.resource; import java.io.IOException; import junit.framework.TestCase; import org.springframework.batch.core.JobExecution; import org.springframework.batch.core.JobInstance; import org.springframework.batch.core.JobParameters; import org.springframework.batch.core.JobParametersBuilder; import org.springframework.batch.core.Step; import org.springframework.batch.core.StepExecution; import org.springframework.batch.core.step.StepSupport; import org.springframework.batch.repeat.RepeatContext; /** * Unit tests for {@link StepExecutionSimpleCompletionPolicy} * * @author Dave Syer */ public class StepExecutionSimpleCompletionPolicyTests extends TestCase { /** * Object under test */ private StepExecutionSimpleCompletionPolicy policy = new StepExecutionSimpleCompletionPolicy(); private JobInstance jobInstance; private StepExecution stepExecution; /** * mock step context */ @Override protected void setUp() throws Exception { JobParameters jobParameters = new JobParametersBuilder().addLong("commit.interval", 2L).toJobParameters(); jobInstance = new JobInstance(new Long(0), "testJob"); JobExecution jobExecution = new JobExecution(jobInstance, jobParameters); Step step = new StepSupport("bar"); stepExecution = jobExecution.createStepExecution(step.getName()); policy.beforeStep(stepExecution); } public void testToString() throws Exception { String msg = policy.toString(); assertTrue("String does not contain chunk size", msg.indexOf("chunkSize=2")>=0); } public void testKeyName() throws Exception, IOException { RepeatContext context = policy.start(null); assertFalse(policy.isComplete(context)); } }