InfolisBaseTest.java

package io.github.infolis;

import io.github.infolis.datastore.DataStoreClient;
import io.github.infolis.datastore.DataStoreClientFactory;
import io.github.infolis.datastore.DataStoreStrategy;
import io.github.infolis.datastore.FileResolver;
import io.github.infolis.datastore.FileResolverFactory;
import io.github.infolis.model.entity.Entity;
import io.github.infolis.model.entity.InfolisFile;
import io.github.infolis.util.SerializationUtils;

import java.io.OutputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.ArrayList;
import java.util.List;

import javax.ws.rs.client.Client;
import javax.ws.rs.client.ClientBuilder;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.IOUtils;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.pdmodel.PDPage;
import org.apache.pdfbox.pdmodel.edit.PDPageContentStream;
import org.apache.pdfbox.pdmodel.font.PDType1Font;
import org.glassfish.jersey.media.multipart.MultiPartFeature;

public class InfolisBaseTest {
	protected DataStoreClient dataStoreClient;
	protected FileResolver fileResolver;
	{
		DataStoreStrategy strategy;
		if (Boolean.valueOf(System.getProperty("infolisRemoteTests", "false"))) {
			strategy = DataStoreStrategy.CENTRAL;
		} else {
			strategy = DataStoreStrategy.TEMPORARY;
		}
		setupClients(strategy);
	}

	protected void setupClients(DataStoreStrategy strategy) {
		dataStoreClient = DataStoreClientFactory.create(strategy);
		fileResolver =  FileResolverFactory.create(strategy);
	}

	protected static Client jerseyClient = ClientBuilder.newBuilder()
			// .property(HttpUrlConnectorProvider.SET_METHOD_WORKAROUND, true)
			.register(MultiPartFeature.class)
			// .register(JacksonFeature.class)
			// .register(JacksonJsonProvider.class)
			.build();

	protected List<InfolisFile> createTestTextFiles(int nrFiles, String[] testStrings) throws Exception {
		List<InfolisFile> ret = new ArrayList<>();
		int j = 0;
		for (int i = 0; i < nrFiles; i++) {
			j = i % testStrings.length;
			String data = testStrings[j];
			Path tempFile = Files.createTempFile("infolis-", ".txt");
			FileUtils.write(tempFile.toFile(), data);

			InfolisFile file = new InfolisFile();
			file.setMd5(SerializationUtils.getHexMd5(data));
			file.setFileName(tempFile.toString());
			file.setFileStatus("AVAILABLE");
			file.setMediaType("text/plain");
			
			OutputStream os = fileResolver.openOutputStream(file);
			IOUtils.write(data, os);
			os.close();
			Entity entity = new Entity();
			dataStoreClient.post(Entity.class, entity);
			file.setManifestsEntity(entity.getUri());
			
			dataStoreClient.post(InfolisFile.class, file);
			ret.add(file);
		}
		return ret;
	}

	protected List<InfolisFile> createTestPdfFiles(int nrFiles, String[] testStrings) throws Exception {
		List<InfolisFile> ret = new ArrayList<>();

		int j = 0;
		for (int i = 0; i < nrFiles; i++) {
			j = i % testStrings.length;
			String data = testStrings[j];
			Path tempFile = Files.createTempFile("infolis-", ".pdf");

			// create the pdf
			PDDocument document = new PDDocument();
			PDPage page = new PDPage();
			document.addPage(page);
			PDPageContentStream contentStream = new PDPageContentStream(document, page);
			contentStream.beginText();
			contentStream.setFont(PDType1Font.COURIER, 12);
			contentStream.moveTextPositionByAmount(100, 100);
			contentStream.drawString(data);
			contentStream.endText();
			contentStream.close();
			// Important, otherwise hashing won't work because different random ID
			document.setDocumentId(0L);

			// Save to temp file
			OutputStream tempOut = Files.newOutputStream(tempFile);
			document.save(tempOut);
			document.close();
			tempOut.close();

			// create infolis file
			InfolisFile file = new InfolisFile();
			file.setMd5(SerializationUtils.getHexMd5(IOUtils.toByteArray(Files.newInputStream(tempFile))));
			file.setFileName(tempFile.toString());
			file.setFileStatus("AVAILABLE");
			file.setMediaType("application/pdf");

			// Write the data to the file
			OutputStream os = fileResolver.openOutputStream(file);
			IOUtils.copy(Files.newInputStream(tempFile), os);
			os.close();

			Entity entity = new Entity();
			dataStoreClient.post(Entity.class, entity);
			file.setManifestsEntity(entity.getUri());
			dataStoreClient.post(InfolisFile.class, file);
			ret.add(file);
		}
		return ret;
	}
}