001/*
002 * Copyright 2015 DuraSpace, Inc.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 *     http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016package org.fcrepo.migration.pidlist;
017
018import org.junit.After;
019import org.junit.Assert;
020import org.junit.Before;
021import org.junit.Test;
022
023import java.io.File;
024import java.util.Arrays;
025import java.util.List;
026
027/**
028 * Unit test class for ResumePidListManager
029 *
030 * @author awoods
031 * @since 2019-11-08
032 */
033public class ResumePidListManagerTest {
034
035    private ResumePidListManager manager;
036    private final String testDir = System.getProperty("test.output.dir");
037
038    private List<String> pidList;
039
040    @Before
041    public void setUp() {
042        // Test PIDs
043        pidList = Arrays.asList("pid:1", "pid:2", "pid:3", "pid:4");
044
045        System.out.println("test dir: " + testDir);
046
047        // Define directory in which to find resume-file.
048        manager = new ResumePidListManager(new File(testDir), false);
049    }
050
051    @After
052    public void tearDown() {
053        manager.reset();
054    }
055
056    @Test
057    public void accept() {
058        pidList.forEach(pid -> Assert.assertTrue(pid + " should be accepted", manager.accept(pid)));
059    }
060
061    @Test
062    public void acceptIncrementalRuns() {
063        Assert.assertTrue("pid:1 should be accepted", manager.accept("pid:1"));
064        Assert.assertTrue("pid:2 should be accepted", manager.accept("pid:2"));
065
066        // Simulate stopping the migration process... and start over
067        manager = new ResumePidListManager(new File(testDir), false);
068        Assert.assertFalse("pid:1 should NOT be accepted", manager.accept("pid:1"));
069        Assert.assertFalse("pid:2 should NOT be accepted", manager.accept("pid:2"));
070
071        // ..however, unprocessed PIDs should be "accepted"
072        Assert.assertTrue("pid:3 should be accepted", manager.accept("pid:3"));
073        Assert.assertTrue("pid:4 should be accepted", manager.accept("pid:4"));
074
075        // Starting over again... no PIDs should be accepted
076        manager = new ResumePidListManager(new File(testDir), false);
077        pidList.forEach(pid -> Assert.assertFalse(pid + " should NOT be accepted", manager.accept(pid)));
078    }
079
080    @Test
081    public void acceptAll() {
082        Assert.assertTrue("pid:1 should be accepted", manager.accept("pid:1"));
083        Assert.assertTrue("pid:2 should be accepted", manager.accept("pid:2"));
084
085        // Simulate stopping the migration process... and start over - but accept all
086        manager = new ResumePidListManager(new File(testDir), true);
087        Assert.assertTrue("pid:1 should be accepted", manager.accept("pid:1"));
088        Assert.assertTrue("pid:2 should be accepted", manager.accept("pid:2"));
089
090        // ..however, unprocessed PIDs should be "accepted" - accept all
091        Assert.assertTrue("pid:3 should be accepted", manager.accept("pid:3"));
092        Assert.assertTrue("pid:4 should be accepted", manager.accept("pid:4"));
093
094        // Starting over again... no PIDs should be accepted - but, accept all
095        manager = new ResumePidListManager(new File(testDir), true);
096        pidList.forEach(pid -> Assert.assertTrue(pid + " should be accepted", manager.accept(pid)));
097    }
098}