org.lobid.lodmill.hadoop.UnitTestGndNTriplesToJsonLd.java Source code

Introduction

Here is the source code for org.lobid.lodmill.hadoop.UnitTestGndNTriplesToJsonLd.java

Source

/* Copyright 2012-2013 Fabian Steeg. Licensed under the Eclipse Public License 1.0 */

package org.lobid.lodmill.hadoop; // NOPMD

import static org.lobid.lodmill.hadoop.UnitTestCollectSubjects.GND_CREATOR_ID;
import static org.lobid.lodmill.hadoop.UnitTestCollectSubjects.gnd;

import java.io.IOException;
import java.util.Arrays;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.io.LongWritable;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mrunit.TestDriver;
import org.apache.hadoop.mrunit.mapreduce.MapDriver;
import org.apache.hadoop.mrunit.mapreduce.ReduceDriver;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.JSONValue;
import org.junit.Before;
import org.junit.Test;
import org.lobid.lodmill.hadoop.NTriplesToJsonLd.NTriplesToJsonLdMapper;
import org.lobid.lodmill.hadoop.NTriplesToJsonLd.NTriplesToJsonLdReducer;

import com.google.common.collect.ImmutableMap;

/**
 * Test the {@link NTriplesToJsonLd} class.
 * 
 * @author Fabian Steeg (fsteeg)
 */
@SuppressWarnings("javadoc")
public final class UnitTestGndNTriplesToJsonLd {

    private static final String TRIPLE_ID = "http://d-nb.info/gnd/" + GND_CREATOR_ID;
    private static final String TRIPLE_URI = "<" + TRIPLE_ID + ">";
    static final String GND_TRIPLE_1 = gnd("preferredNameForThePerson", "Adamucci, Antonio");
    static final String GND_TRIPLE_2 = gnd("dateOfBirth", "1828");
    static final String GND_TRIPLE_3 = gnd("dateOfDeath", "1885");
    private static final String INDEX = "gnd-index";
    private static final String TYPE = "json-ld-gnd";
    private MapDriver<LongWritable, Text, Text, Text> mapDriver;
    private ReduceDriver<Text, Text, Text, Text> reduceDriver;

    @Before
    public void setUp() {
        final Configuration configuration = new Configuration();
        configuration.set(NTriplesToJsonLd.INDEX_NAME, INDEX);
        configuration.set(NTriplesToJsonLd.INDEX_TYPE, TYPE);
        final NTriplesToJsonLdMapper mapper = new NTriplesToJsonLdMapper();
        final NTriplesToJsonLdReducer reducer = new NTriplesToJsonLdReducer();
        mapDriver = MapDriver.newMapDriver(mapper);
        reduceDriver = ReduceDriver.newReduceDriver(reducer);
        setConfiguration(mapDriver);
        setConfiguration(reduceDriver);
    }

    private static void setConfiguration(final TestDriver<?, ?, ?, ?, ?> driver) {
        driver.getConfiguration().set(NTriplesToJsonLd.INDEX_NAME, INDEX);
        driver.getConfiguration().set(NTriplesToJsonLd.INDEX_TYPE, TYPE);
    }

    @Test
    public void testMapper() throws IOException { // NOPMD (MRUnit)
        mapDriver.addInput(new LongWritable(), new Text(GND_TRIPLE_1));
        mapDriver.addInput(new LongWritable(), new Text(GND_TRIPLE_2));
        mapDriver.addInput(new LongWritable(), new Text(GND_TRIPLE_3));
        mapDriver.withOutput(new Text(TRIPLE_URI), new Text(GND_TRIPLE_1))
                .withOutput(new Text(TRIPLE_URI), new Text(GND_TRIPLE_2))
                .withOutput(new Text(TRIPLE_URI), new Text(GND_TRIPLE_3));
        mapDriver.runTest();
    }

    @Test
    public void testReducer() throws IOException { // NOPMD (MRUnit)
        reduceDriver.withInput(new Text(TRIPLE_URI),
                Arrays.asList(new Text(GND_TRIPLE_1), new Text(GND_TRIPLE_2), new Text(GND_TRIPLE_3)));
        reduceDriver.withOutput(new Text(""), new Text(JSONValue.toJSONString(correctJson())));
        reduceDriver.runTest();
    }

    @SuppressWarnings({ "unchecked" })
    static JSONObject correctJson() {
        JSONArray array = new JSONArray();
        JSONObject obj = new JSONObject();
        obj.put("@id", "http://d-nb.info/gnd/118643606");
        obj.put("http://d-nb.info/standards/elementset/gnd#preferredNameForThePerson",
                Arrays.asList(ImmutableMap.of("@value", "Adamucci, Antonio")));
        obj.put("http://d-nb.info/standards/elementset/gnd#dateOfDeath",
                Arrays.asList(ImmutableMap.of("@value", "1885")));
        obj.put("http://d-nb.info/standards/elementset/gnd#dateOfBirth",
                Arrays.asList(ImmutableMap.of("@value", "1828")));
        array.add(obj);
        return new JSONObject(
                ImmutableMap.of(NTriplesToJsonLd.INTERNAL_ID, "http://d-nb.info/gnd/118643606", "@graph", array));
    }
}