D-Net

package eu.dnetlib.data.transform.xml.vtd;

import java.io.IOException;

import java.util.function.Function;

import eu.dnetlib.data.proto.OafProtos.Oaf;

import eu.dnetlib.data.transform.xml2.DatasetToProto;

import eu.dnetlib.data.transform.xml2.PublicationToProto;

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.junit.Test;

import static org.junit.Assert.assertNotNull;

public class VtdParserToProtoTest {

	private static final Log log = LogFactory.getLog(VtdParserToProtoTest.class);

	@Test

	public void testParsePublication() throws IOException {

		doTest("/eu/dnetlib/data/transform/publication.xml", new PublicationToProto());

	}

	@Test

	public void testParseDataset() throws IOException {

		doTest("/eu/dnetlib/data/transform/dataset.xml", new DatasetToProto());

	}

	@Test

	public void testParseDataset2() throws IOException {

		doTest("/eu/dnetlib/data/transform/dataset2.xml", new DatasetToProto());

	}

	private void doTest(final String filePath, Function<String, Oaf> f) throws IOException {

		final String xml = IOUtils.toString(getClass().getResourceAsStream(filePath));

		assertNotNull(xml);

		final Oaf oaf = f.apply(xml);

		assertNotNull(oaf);

		log.info(oaf.toString());

	}

}