fr.acxio.tools.agia.tasks.FileOperationTaskletTest.java Source code

Introduction

Here is the source code for fr.acxio.tools.agia.tasks.FileOperationTaskletTest.java

Source

package fr.acxio.tools.agia.tasks;

/*
 * Copyright 2014 Acxio
 * 
 * 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.
 */

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import java.io.File;
import java.util.Collection;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.filefilter.WildcardFileFilter;
import org.junit.After;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;
import org.springframework.batch.core.StepContribution;
import org.springframework.batch.repeat.RepeatStatus;
import org.springframework.core.io.FileSystemResource;

import fr.acxio.tools.agia.io.FileOperationException;
import fr.acxio.tools.agia.io.AbstractFileOperations.Operation;

@RunWith(JUnit4.class)
public class FileOperationTaskletTest {

    @Rule
    public ExpectedException exception = ExpectedException.none();

    @After
    public void tearDown() throws Exception {
        Collection<File> aFilesToDelete = FileUtils.listFiles(new File("target"),
                new WildcardFileFilter("input-copy*.csv"), null);
        for (File aFile : aFilesToDelete) {
            FileUtils.deleteQuietly(aFile);
        }
    }

    @Test
    public void testExecuteCopy() throws Exception {
        FileOperationTasklet aTasklet = new FileOperationTasklet();
        aTasklet.setOrigin(new FileSystemResource("src/test/resources/testFiles/input.csv"));
        aTasklet.setDestination(new FileSystemResource("target/input-copy.csv"));
        aTasklet.setOperation(Operation.COPY);
        aTasklet.afterPropertiesSet();
        StepContribution aStepContribution = mock(StepContribution.class);
        assertEquals(RepeatStatus.FINISHED, aTasklet.execute(aStepContribution, null));
        verify(aStepContribution, times(1)).incrementReadCount();
        verify(aStepContribution, times(1)).incrementWriteCount(1);
    }

    @Test(expected = FileOperationException.class)
    public void testExecuteCopyFileNotFound() throws Exception {
        FileOperationTasklet aTasklet = new FileOperationTasklet();
        aTasklet.setOrigin(new FileSystemResource("src/test/resources/testFiles/notfound.csv"));
        aTasklet.setDestination(new FileSystemResource("target/input-copy.csv"));
        aTasklet.setOperation(Operation.COPY);
        aTasklet.afterPropertiesSet();
        StepContribution aStepContribution = mock(StepContribution.class);
        aTasklet.execute(aStepContribution, null);
        verify(aStepContribution, times(0)).incrementReadCount();
        verify(aStepContribution, times(0)).incrementWriteCount(1);
    }

    @Test(expected = IllegalArgumentException.class)
    public void testExecuteCopyDestinationFileNotFound() throws Exception {
        FileOperationTasklet aTasklet = new FileOperationTasklet();
        aTasklet.setOrigin(new FileSystemResource("src/test/resources/testFiles/notfound.csv"));
        aTasklet.setOperation(Operation.COPY);
        aTasklet.afterPropertiesSet();
    }
}