D-Net

import java.io.InputStream;

import java.util.HashMap;

import java.util.Map;

import com.google.common.base.Function;

import com.google.common.base.Predicate;

import com.google.common.collect.Iterables;

import com.google.common.collect.Maps;

import com.google.protobuf.InvalidProtocolBufferException;

import eu.dnetlib.data.proto.TypeProtos;

import eu.dnetlib.data.transform.Column;

import eu.dnetlib.data.transform.Row;

import eu.dnetlib.data.transform.XsltRowTransformer;

import eu.dnetlib.data.transform.XsltRowTransformerFactory;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.dom4j.Document;

import org.dom4j.DocumentException;

import org.dom4j.io.SAXReader;

import static org.junit.Assert.assertFalse;

import static org.junit.Assert.assertNotNull;

import static org.junit.Assert.assertTrue;

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

	private static String basePathProfiles = "/eu/dnetlib/test/profiles/TransformationRuleDSResources/TransformationRuleDSResourceType/";

	protected XsltRowTransformerFactory factory;

	protected DedupConfig getResultResourceTypeConf() {

		return DedupConfig.load(readFromClasspath("/eu/dnetlib/pace/result.test1.pace.conf"));

	}

	protected MapDocument getMapDocument(final Config config, final String filename, final String xsltName, final TypeProtos.Type type) throws Exception {

		final Iterable<Row> rows = Iterables.filter(asRows(loadFromTransformationProfile(xsltName), load(filename)), new Predicate<Row>() {

			@Override

			public boolean apply(final Row row) {

				final String cf = row.getColumnFamily();

				return cf.equals(type.toString());

			}

		});

		final Iterable<Oaf> oaf = Iterables.transform(rows, new Function<Row, Oaf>() {

			@Override

			public Oaf apply(final Row row) {

				final Column<String, byte[]> column = row.getColumn("body");

				final byte[] value = column.getValue();

				try {

					return Oaf.parseFrom(value);

				} catch (InvalidProtocolBufferException e) {

					throw new IllegalArgumentException(row.toString(), e);

				}

			}

		});

		final MapDocument doc = Iterables.getOnlyElement(Iterables.transform(oaf, new Function<Oaf, MapDocument>() {

			@Override

			public MapDocument apply(final Oaf oaf) {

				return ProtoDocumentBuilder.newInstance(oaf.getEntity().getId(), oaf.getEntity(), config.model());

			}

		}));

		return doc;

	}

	protected List<Row> asRows(final InputStream xsltStream, final InputStream recordStream) throws Exception {

		return asRows(xsltStream, new HashMap<String, Object>(), recordStream);

	}

	protected List<Row> asRows(final InputStream xsltStream, final Map<String, Object> params, final InputStream recordStream) throws Exception {

		final String xslt = IOUtils.toString(xsltStream);

		final XsltRowTransformer transformer = factory.getTransformer(xslt, params);

		assertNotNull(transformer);

		final String record = IOUtils.toString(recordStream);

		final List<Row> rows = transformer.apply(record);

		assertNotNull(rows);

		assertFalse(rows.isEmpty());

		return rows;

	}

	protected InputStream loadFromTransformationProfile(final String profilePath) {

		log.info("Loading xslt from: " + basePathProfiles + profilePath);

		InputStream profile = getClass().getResourceAsStream(basePathProfiles + profilePath);

		SAXReader saxReader = new SAXReader();

		Document doc = null;

		try {

			doc = saxReader.read(profile);

		} catch (DocumentException e) {

			e.printStackTrace();

			throw new RuntimeException(e);

		}

		String xslt = doc.selectSingleNode("//SCRIPT/CODE/*[local-name()='stylesheet']").asXML();

		return IOUtils.toInputStream(xslt);

	}

	protected InputStream load(final String fileName) {

		return getClass().getResourceAsStream(fileName);

	}