GoldLinkerTest.java

package io.github.infolis.algorithm;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import org.apache.commons.io.IOUtils;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.github.infolis.InfolisBaseTest;
import io.github.infolis.datastore.FileResolverFactory;
import io.github.infolis.model.Execution;
import io.github.infolis.model.entity.Entity;
import io.github.infolis.model.entity.EntityLink;
import io.github.infolis.model.entity.InfolisFile;
import io.github.infolis.model.entity.SearchResult;
import io.github.infolis.util.SerializationUtils;

/**
 * 
 * @author kata
 *
 */
public class GoldLinkerTest extends InfolisBaseTest {
	
	private static Logger log = LoggerFactory.getLogger(GoldLinkerTest.class);
	
	@Test
	public void test() throws IOException {
		File inputDir = new File(getClass().getResource("/linkImporter/").getFile());
		List<String> uris = postFiles(inputDir, "text/json");
		Entity entity = new Entity();
		entity.setName("name1");
		dataStoreClient.post(Entity.class, entity);
		SearchResult searchResult = new SearchResult();
		dataStoreClient.post(SearchResult.class, searchResult);
		
		Execution exec = new Execution();
		exec.setAlgorithm(GoldLinker.class);
		exec.setInputFiles(uris);
		exec.setSearchResults(Arrays.asList(searchResult.getUri()));
		exec.setLinkedEntities(Arrays.asList(entity.getUri()));
		exec.instantiateAlgorithm(dataStoreClient, dataStoreClient, FileResolverFactory.local(), fileResolver).run();
		log.debug("links: " + exec.getLinks());
		for (EntityLink link : dataStoreClient.get(EntityLink.class, exec.getLinks())) {
			log.debug("fromEntity: " + link.getFromEntity());
			log.debug("toEntity: " + link.getToEntity());
			log.debug("confidence: " + link.getConfidence());
			log.debug("linkReason: " + link.getLinkReason());
			log.debug("entityRelations: " + link.getEntityRelations());
			log.debug("tags: " + link.getTags());
		}
		
	}
	
	
	private List<String> postFiles(File dir, String mimetype) throws IOException {
        List<InfolisFile> infolisFiles = new ArrayList<>();
        for (File file : dir.listFiles()) {
            InfolisFile infolisFile = new InfolisFile();
            InputStream inputStream = Files.newInputStream(Paths.get(file.getAbsolutePath()));
            byte[] bytes = IOUtils.toByteArray(inputStream);
            infolisFile.setMd5(SerializationUtils.getHexMd5(bytes));
            infolisFile.setFileName(file.toString());
            infolisFile.setMediaType(mimetype);
            infolisFile.setFileStatus("AVAILABLE");
            infolisFiles.add(infolisFile);
        }
        return dataStoreClient.post(InfolisFile.class, infolisFiles);
    }
}