D-Net

package eu.dnetlib.actionmanager.actions;

import java.io.IOException;

import java.io.InputStream;

import java.io.StringWriter;

import java.util.Arrays;

import java.util.List;

import javax.annotation.Resource;

import eu.dnetlib.actionmanager.common.Agent;

import eu.dnetlib.actionmanager.common.Agent.AGENT_TYPE;

import eu.dnetlib.actionmanager.common.Operation;

import eu.dnetlib.actionmanager.common.Provenance;

import eu.dnetlib.actionmanager.rmi.ActionManagerException;

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

import org.apache.commons.io.IOUtils;

import org.apache.commons.logging.Log;

import org.apache.commons.logging.LogFactory;

import org.apache.hadoop.hbase.KeyValue;

import org.apache.hadoop.hbase.client.Put;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.springframework.test.context.ContextConfiguration;

import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import static org.junit.Assert.*;

@RunWith(SpringJUnit4ClassRunner.class)

@ContextConfiguration(locations = { "/eu/dnetlib/actionmanager/actions/applicationContext-actionmanager-test.xml" })

public class ActionFactoryTest {

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

	private final static Agent agent = new Agent("agentId", "agentName", AGENT_TYPE.algo);

	private final static String trust = "0.9";

	private final InputStream recordDataciteInputStream = getClass().getResourceAsStream("recordDatacite.xml");

	private final InputStream recordOafInputStream = getClass().getResourceAsStream("record.xml");

	@Resource

	private ActionFactory actionFactory;

	// @Before

	// public void setUp() throws Exception {

	//

	// }

	@Test

	public void testBeans() {

		assertNotNull(actionFactory);

		assertNotNull(actionFactory.getXslts());

		assertFalse(actionFactory.getXslts().isEmpty());

	}

	@Test

	public void testDataciteInsert() throws ActionManagerException, IOException {

		XsltInfoPackageAction action =

				actionFactory.generateInfoPackageAction("datacite2insertActions", "rawSet", agent, Operation.INSERT, asString(recordDataciteInputStream),

						Provenance.sysimport_crosswalk_datasetarchive, "datacite____", trust);

		final List<Put> puts = action.asPutOperations();

		for (AtomicAction a : action.calculateAtomicActions()) {

			assertNotNull(a.getTargetValue());

			Oaf oaf = Oaf.parseFrom(a.getTargetValue());

			assertNotNull(oaf);

			//System.out.println(oaf);

		}

		for (Put put : puts) {

			assertNotNull(put);

		}

		log.info(puts.size());

	}

	@Test

	public void testDataciteUpdate() throws ActionManagerException, IOException {

		XsltInfoPackageAction action =

				actionFactory.generateInfoPackageAction("datacite2updateActions", "rawSet", agent, Operation.UPDATE, asString(recordDataciteInputStream),

						Provenance.sysimport_crosswalk_datasetarchive, "datacite____", trust);

		final List<Put> puts = action.asPutOperations();

		assertTrue(puts.size() == 1);

		for (Put put : puts) {

			assertNotNull(put);

			final List<KeyValue> target = put.getFamilyMap().get("target".getBytes());

			if (target != null) {

				for (KeyValue kv : target) {

					final String qualifier = new String(kv.getQualifier());

					if (qualifier.equals("content")) {

						Oaf oaf = Oaf.parseFrom(kv.getValue());

						assertNotNull(oaf);

						//System.out.println(oaf);

					}

				}

			}

		}

	}

	@Test

	public void testRecordOafInsert() throws ActionManagerException, IOException {

		XsltInfoPackageAction action =

				actionFactory.generateInfoPackageAction("dmf2insertActions", "rawSet", agent, Operation.INSERT, asString(recordOafInputStream),

						Provenance.sysimport_crosswalk_datasetarchive, "od_______296", trust);

		final List<Put> puts = action.asPutOperations();

		for (AtomicAction a : action.calculateAtomicActions()) {

			assertNotNull(a.getTargetValue());

			String actionJSON = a.toString();

			log.info(actionJSON);

			final AtomicAction aa = AtomicAction.fromJSON(actionJSON);

			assertEquals(a.getRawSet(), aa.getRawSet());

			assertEquals(a.getTargetColumn(), aa.getTargetColumn());

			assertEquals(a.getTargetColumnFamily(), aa.getTargetColumnFamily());

			assertEquals(a.getTargetRowKey(), aa.getTargetRowKey());

			assertEquals(a.getActionType(), aa.getActionType());

			assertEquals(a.getRowKey(), aa.getRowKey());

			assertEquals(a.getAgent().getId(), aa.getAgent().getId());

			assertEquals(a.getAgent().getName(), aa.getAgent().getName());

			assertEquals(a.getAgent().getType(), aa.getAgent().getType());

			assertTrue(Arrays.equals(a.getTargetValue(), aa.getTargetValue()));

			//System.out.println(oaf);

		}

		for (Put put : puts) {

			assertNotNull(put);

		}

		log.info("N. puts: " + puts.size());

	}

	private String asString(final InputStream i) throws IOException {

		StringWriter writer = new StringWriter();

		IOUtils.copy(i, writer);

		return writer.toString();

	}

}