D-Net

package eu.dnetlib.data.mapreduce.actions;

import com.google.gson.JsonObject;

import com.google.gson.JsonParser;

import eu.dnetlib.actionmanager.actions.ActionFactory;

import eu.dnetlib.actionmanager.actions.AtomicAction;

import eu.dnetlib.actionmanager.common.Agent;

import eu.dnetlib.data.mapreduce.hbase.dataimport.CrossRefToActions;

import org.apache.commons.lang3.StringUtils;

import org.junit.Before;

import org.junit.Test;

import java.io.BufferedReader;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

public class CrossRefToActionTest {

    private String setName;

    private Agent agent;

    @Before

    public void setup() {

        setName = "DLI";

        agent= new Agent("agentId","agentName", Agent.AGENT_TYPE.service);

    }

    @Test

    public void testSingleScholixAction2() throws IOException {

        doTestSingleCrossRefAction("/eu/dnetlib/data/mapreduce/actions/broken");

    }

    @Test

    public void testAllScholixAction() throws IOException {

        doTestAllCrossRefAction("/eu/dnetlib/data/mapreduce/actions/part-06036");

    }

    private void doTestSingleCrossRefAction(final String filePath) throws IOException {

        final InputStream is = this.getClass().getResourceAsStream(filePath);

        final BufferedReader in = new BufferedReader(new InputStreamReader(is));

        String line = in.readLine();

        System.out.println(line);

        final JsonParser parser = new JsonParser();

        JsonObject root = parser.parse(line).getAsJsonObject();

        AtomicAction actions = CrossRefToActions.generateActionsFromDump(root, new ActionFactory(), setName, agent, false);

        System.out.println(actions.toJSON());

    }

    private void doTestAllCrossRefAction(final String filePath) throws IOException {

        final InputStream is = this.getClass().getResourceAsStream(filePath);

        final BufferedReader in = new BufferedReader(new InputStreamReader(is));

        String line = in.readLine();

        while(StringUtils.isNotBlank(line)) {

            final JsonParser parser = new JsonParser();

            JsonObject root = parser.parse(line).getAsJsonObject();

            try {

                CrossRefToActions.generateActionsFromDump(root, new ActionFactory(), setName, agent, false);

            } catch (Throwable e) {

                System.out.println(line);

                throw new RuntimeException(e);

            }

            line= in.readLine();

        }

    }

}