D-Net

package eu.dnetlib.data.mapreduce.hbase.dataexport;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.googlecode.protobuf.format.JsonFormat;

import eu.dnetlib.data.proto.OafProtos;

import eu.dnetlib.dhp.schema.oaf.*;

import org.apache.commons.io.IOUtils;

import org.junit.Test;

import java.io.IOException;

import static org.junit.Assert.assertNotNull;

import static org.junit.Assert.assertTrue;

public class ProtoConverterTest {

    @Test

    public void convertDatasourceTest() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/datasource.json");

        assertNotNull(result);

        assertTrue(result instanceof Datasource);

        Datasource ds = (Datasource) result;

        assertNotNull(ds.getId());

        System.out.println(ds.getId());

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(result));

    }

    @Test

    public void convertOrganizationTest() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/organization.json");

        assertNotNull(result);

        assertTrue(result instanceof Organization);

        Organization ds = (Organization) result;

        assertNotNull(ds.getId());

        System.out.println(ds.getId());

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(result));

    }

    @Test

    public void convertPublicationTest() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/publication.json");

        assertNotNull(result);

        assertTrue(result instanceof Publication);

        Publication p = (Publication) result;

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(p));

    }

    @Test

    public void convertDatasetTest() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/dataset.json");

        assertNotNull(result);

        assertTrue(result instanceof Dataset);

        Dataset d = (Dataset) result;

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(d));

    }

    @Test

    public void convertORPTest() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/orp.json");

        assertNotNull(result);

        assertTrue(result instanceof OtherResearchProduct);

        OtherResearchProduct orp = (OtherResearchProduct) result;

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(orp));

    }

    @Test

    public void convertSoftware() throws Exception {

        Oaf result = getOaf("/eu/dnetlib/data/mapreduce/hbase/dataexport/software.json");

        assertNotNull(result);

        assertTrue(result instanceof Software);

        Software s = (Software) result;

        ObjectMapper mapper = new ObjectMapper();

        System.out.println(mapper.writeValueAsString(s));

    }

    private Oaf getOaf(String s2) throws IOException {

        final String json = IOUtils.toString(this.getClass().getResourceAsStream(s2));

        final OafProtos.Oaf.Builder b = OafProtos.Oaf.newBuilder();

        JsonFormat.merge(json, b);

        return ProtoConverter.convert(b.build());

    }

}