eu.freme.eservices.epublishing.webservice.EPubTest.java Source code

Java tutorial

Introduction

Here is the source code for eu.freme.eservices.epublishing.webservice.EPubTest.java

Source

/**
 * Copyright (C) 2015 Agro-Know, Deutsches Forschungszentrum fr Knstliche Intelligenz, iMinds,
 * Institut fr Angewandte Informatik e. V. an der Universitt Leipzig,
 * Istituto Superiore Mario Boella, Tilde, Vistatec, WRIPL (http://freme-project.eu)
 *
 * 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 eu.freme.eservices.epublishing.webservice;

import com.google.gson.Gson;
import nl.siegmann.epublib.domain.CreatorContributor;
import nl.siegmann.epublib.domain.Book;
import nl.siegmann.epublib.domain.Resource;
import nl.siegmann.epublib.epub.EpubReader;
import org.apache.commons.io.IOUtils;
import org.glassfish.jersey.media.multipart.FormDataMultiPart;
import org.glassfish.jersey.media.multipart.file.FileDataBodyPart;
import org.junit.Assert;
import org.junit.Test;

import javax.ws.rs.client.Entity;
import javax.ws.rs.core.Response;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import org.junit.Ignore;

/**
 * <p>
 * Basic e-publishing test</p>
 *
 * <p>
 * Copyright 2015 MMLab, UGent</p>
 *
 * @author Gerald Haesendonck
 */
public class EPubTest extends ClientAndLocalServerBase {

    @Ignore
    @Test
    public void TestAlice() throws IOException {
        String title = "Alice in Utopia";
        String author = "Joske Vermeulen";
        String author2 = "Marieke Vermalen";
        String description = "Dit is een heel mooi boekske.";

        File zippedHTMLFile = new File("src/test/resources/alice.zip");
        System.out.println("Converting " + zippedHTMLFile + " to EPUB format.");

        Metadata metadata = new Metadata();
        ArrayList<String> titles = new ArrayList<>();
        ArrayList<String> authors = new ArrayList<>();
        ArrayList<String> descriptions = new ArrayList<>();
        titles.add(title);
        authors.add(author);
        authors.add(author2);
        metadata.setTitles(titles);
        //metadata.setAuthors(authors);
        descriptions.add(description);
        metadata.setDescriptions(descriptions);
        Identifier id = new Identifier(null, "urn:ean:1234-7956-1356-1123");
        metadata.setIdentifier(id);

        Gson gson = new Gson();

        // create body part containing the ePub file
        FileDataBodyPart ePubFilePart = new FileDataBodyPart("htmlZip", zippedHTMLFile);

        // the form, to which parameters and binary content can be added
        FormDataMultiPart multiPart = new FormDataMultiPart();
        multiPart.bodyPart(ePubFilePart); // add ePub
        multiPart.field("metadata", gson.toJson(metadata));
        System.out.println(gson.toJson(metadata));

        // now send it to the service
        Response response = target.path("e-publishing/html").request("application/epub+zip")
                .post(Entity.entity(multiPart, multiPart.getMediaType()));
        InputStream in = response.readEntity(InputStream.class);

        // write it to a (temporary) file
        File ePubFile = File.createTempFile("alice", ".epub");
        System.out.println("Writing result to " + ePubFile);
        IOUtils.copy(in, new FileOutputStream(ePubFile));

        //read file and perform checks
        EpubReader r = new EpubReader();
        Book b = r.readEpub(new FileInputStream(ePubFile));
        List<String> bookTitles = b.getMetadata().getTitles();
        List<CreatorContributor> bookAuthors = b.getMetadata().getAuthors();
        List<String> bookAuthorsNames = new ArrayList<>();

        for (CreatorContributor a : bookAuthors) {
            bookAuthorsNames.add(a.getFirstname() + " " + a.getLastname());
        }

        Assert.assertTrue(bookTitles.containsAll(titles));
        Assert.assertTrue(titles.containsAll(bookTitles));
        System.out.println(bookAuthorsNames);
        Assert.assertTrue(bookAuthorsNames.containsAll(authors));
        Assert.assertTrue(authors.containsAll(bookAuthorsNames));
    }

    @Ignore
    @Test
    public void TestSections() throws IOException {
        String title = "Alice in Utopia";
        String author = "Joske Vermeulen";

        File zippedHTMLFile = new File("src/test/resources/alice.zip");
        System.out.println("Converting " + zippedHTMLFile + " to EPUB format.");

        Metadata metadata = new Metadata();
        ArrayList<String> titles = new ArrayList<>();
        ArrayList<String> authors = new ArrayList<>();
        ArrayList<Section> toc = new ArrayList<>();
        titles.add(title);
        authors.add(author);
        metadata.setTitles(titles);
        //metadata.setAuthors(authors);
        toc.add(new Section("Chapter 1", "01.xhtml"));
        toc.add(new Section("Chapter 2", "02.xhtml"));
        metadata.setTableOfContents(toc);

        Gson gson = new Gson();

        // create body part containing the ePub file
        FileDataBodyPart ePubFilePart = new FileDataBodyPart("htmlZip", zippedHTMLFile);

        // the form, to which parameters and binary content can be added
        FormDataMultiPart multiPart = new FormDataMultiPart();
        multiPart.bodyPart(ePubFilePart); // add ePub
        multiPart.field("metadata", gson.toJson(metadata));

        // now send it to the service
        Response response = target.path("e-publishing/html").request("application/epub+zip")
                .post(Entity.entity(multiPart, multiPart.getMediaType()));
        InputStream in = response.readEntity(InputStream.class);

        // write it to a (temporary) file
        File ePubFile = File.createTempFile("alice", ".epub");
        System.out.println("Writing result to " + ePubFile);
        IOUtils.copy(in, new FileOutputStream(ePubFile));

        //read file and perform checks
        EpubReader r = new EpubReader();
        Book b = r.readEpub(new FileInputStream(ePubFile));
        List<String> bookTitles = b.getMetadata().getTitles();
        List<CreatorContributor> bookAuthors = b.getMetadata().getAuthors();
        List<String> bookAuthorsNames = new ArrayList<>();

        for (CreatorContributor a : bookAuthors) {
            bookAuthorsNames.add(a.getFirstname() + " " + a.getLastname());
        }

        Assert.assertTrue("All titles are in the EPUB.", bookTitles.containsAll(titles));
        Assert.assertTrue("All EPUB titles are in the given titles.", titles.containsAll(bookTitles));
        System.out.println(bookAuthorsNames);
        Assert.assertTrue("All authors are in the EPUB.", bookAuthorsNames.containsAll(authors));
        Assert.assertTrue("All EPUB authors are in the given authors.", authors.containsAll(bookAuthorsNames));
    }

    @Ignore
    @Test
    public void TestCoverImage() throws IOException {
        String title = "Alice in Utopia";
        String author = "Joske Vermeulen";
        String coverImage = "cover.jpeg";

        File zippedHTMLFile = new File("src/test/resources/alice.zip");
        File coverFile = new File("src/test/resources/" + coverImage);

        Metadata metadata = new Metadata();
        ArrayList<String> titles = new ArrayList<>();
        ArrayList<String> authors = new ArrayList<>();
        ArrayList<Section> toc = new ArrayList<>();
        titles.add(title);
        authors.add(author);
        metadata.setTitles(titles);
        //metadata.setAuthors(authors);
        toc.add(new Section("Chapter 1", "01.xhtml"));
        toc.add(new Section("Chapter 2", "02.xhtml"));
        metadata.setTableOfContents(toc);
        metadata.setCoverImage(coverImage);

        Gson gson = new Gson();

        // create body part containing the ePub file
        FileDataBodyPart ePubFilePart = new FileDataBodyPart("htmlZip", zippedHTMLFile);

        // the form, to which parameters and binary content can be added
        FormDataMultiPart multiPart = new FormDataMultiPart();
        multiPart.bodyPart(ePubFilePart); // add ePub
        multiPart.field("metadata", gson.toJson(metadata));

        // now send it to the service
        Response response = target.path("e-publishing/html").request("application/epub+zip")
                .post(Entity.entity(multiPart, multiPart.getMediaType()));
        InputStream in = response.readEntity(InputStream.class);

        // write it to a (temporary) file
        File ePubFile = File.createTempFile("alice", ".epub");
        IOUtils.copy(in, new FileOutputStream(ePubFile));

        //read file and perform checks
        EpubReader r = new EpubReader();
        Book b = r.readEpub(new FileInputStream(ePubFile));

        Resource cover = b.getCoverImage();
        Assert.assertEquals("Covers have the same size.", cover.getSize(), coverFile.length());
    }
}