de.uzk.hki.da.convert.PublishPDFConversionStrategyTests.java Source code

Introduction

Here is the source code for de.uzk.hki.da.convert.PublishPDFConversionStrategyTests.java

Source

/*
  DA-NRW Software Suite | ContentBroker
  Copyright (C) 2013 Historisch-Kulturwissenschaftliche Informationsverarbeitung
  Universitt zu Kln
    
  This program is free software: you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation, either version 3 of the License, or
  (at your option) any later version.
    
  This program 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 General Public License for more details.
    
  You should have received a copy of the GNU General Public License
  along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package de.uzk.hki.da.convert;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.File;
import java.io.IOException;
import java.util.List;

import org.apache.commons.io.FileUtils;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import de.uzk.hki.da.convert.PublishPDFConversionStrategy;
import de.uzk.hki.da.core.C;
import de.uzk.hki.da.model.ConversionInstruction;
import de.uzk.hki.da.model.ConversionRoutine;
import de.uzk.hki.da.model.DAFile;
import de.uzk.hki.da.model.Event;
import de.uzk.hki.da.model.Node;
import de.uzk.hki.da.model.Object;
import de.uzk.hki.da.model.PublicationRight;
import de.uzk.hki.da.model.TextRestriction;
import de.uzk.hki.da.model.PublicationRight.Audience;
import de.uzk.hki.da.model.WorkArea;
import de.uzk.hki.da.test.TC;
import de.uzk.hki.da.test.TESTHelper;
import de.uzk.hki.da.util.Path;
import de.uzk.hki.da.util.RelativePath;

/**
 * The Class PublishPDFConversionStrategyTests.
 */
public class PublishPDFConversionStrategyTests {

    Path workAreaRootPath = Path.make(TC.TEST_ROOT_CONVERT, "PublishPDFConversionStrategyTests");
    Path dataPath = Path.make(workAreaRootPath, "work/TEST/1", C.WA_DATA);
    Path dipPath = Path.make(dataPath, C.WA_DIP);

    /** The cs. */
    PublishPDFConversionStrategy cs = new PublishPDFConversionStrategy();

    /** The cr. */
    ConversionRoutine cr;

    /** The o. */
    Object o;
    private Node n;

    @Before
    public void setUp() {
        dipPath.toFile().mkdirs();

        n = new Node();
        n.setWorkAreaRootPath(workAreaRootPath);
    }

    @After
    public void tearDown() throws IOException {
        FileUtils.deleteQuietly(dipPath.toFile());
    }

    /**
     * Test.
     *
     * @throws IOException Signals that an I/O exception has occurred.
     */
    @Test
    public void test() throws IOException {

        o = TESTHelper.setUpObject("1", new RelativePath(workAreaRootPath));
        WorkArea wa = new WorkArea(n, o);
        PublicationRight right = new PublicationRight();
        right.setAudience(Audience.PUBLIC);
        right.setTextRestriction(new TextRestriction());
        right.getTextRestriction().setCertainPages(new int[] { 1, 2, 7, 10, 12, 14, 15 });
        o.getRights().getPublicationRights().add(right);

        DAFile sourceFile = new DAFile(o.getLatestPackage(), "a", "filename.pdf");

        ConversionInstruction ci = new ConversionInstruction();
        ci.setSource_file(sourceFile);
        ci.setTarget_folder("target/");

        ConversionRoutine cr = new ConversionRoutine();
        cr.setTarget_suffix("pdf");
        ci.setConversion_routine(cr);

        cs.setObject(o);
        List<Event> events = cs.convertFile(new WorkArea(n, o), ci);

        File targetFile = wa.toFile(events.get(0).getTarget_file());
        assertTrue(targetFile.exists());
        assertEquals("filename.pdf", targetFile.getName());

        // contract states that the PDF should have 7 pages
        PDDocument targetDoc = PDDocument.load(targetFile);
        assertEquals(7, targetDoc.getDocumentCatalog().getAllPages().size());

        targetFile = wa.toFile(events.get(1).getTarget_file());
        assertTrue(targetFile.exists());
        assertEquals("filename.pdf", targetFile.getName());

        // contract has no restrictions for institution, PDF should have
        targetDoc = PDDocument.load(targetFile);
        assertEquals(227, targetDoc.getDocumentCatalog().getAllPages().size());

    }

}