TestSmushReducer.java

package com.ontology2.bakemono.smushObject;

import com.ontology2.bakemono.RecyclingIterable;
import com.ontology2.bakemono.joins.TaggedTextItem;
import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Reducer;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import java.io.IOException;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;

public class TestSmushReducer {
    SmushObjectReducer reducer;
    Reducer<TaggedTextItem,TaggedTextItem,Text,Text>.Context context;

    @Before
    public void setup() {
        reducer=new SmushObjectReducer();
        context=mock(Reducer.Context.class);
    }

    @Test
    public void rewriteThis() throws IOException, InterruptedException {
        RecyclingIterable<TaggedTextItem> inputz=new RecyclingIterable<>(TaggedTextItem.class,
                new TaggedTextItem("<http://dbpedia.org/resources/Usagi_Tsukino>\t<http://www.w3.org/2002/07/owl#sameAs>\t<http://dbpedia.org/resources/Sailor_Moon>\t.",1),
                new TaggedTextItem("<http://dbpedia.org/resources/Sailor_Mars>\t<http://example.org/friend>\t<http://dbpedia.org/resources/Usagi_Tsukino>\t.",16)
        );

        reducer.reduce(
                new TaggedTextItem("<http://dbpedia.org/resources/Usagi_Tsukino>",1),
                inputz,
                context
        );

        verify(context).write(
                new Text("<http://dbpedia.org/resources/Sailor_Mars>"),
                new Text("<http://example.org/friend>\t<http://dbpedia.org/resources/Sailor_Moon>\t.")
        );
    }

    @Test
    public void dontRewriteThat() throws IOException, InterruptedException {
        RecyclingIterable<TaggedTextItem> inputz=new RecyclingIterable<>(TaggedTextItem.class,
                new TaggedTextItem("<http://dbpedia.org/resources/Sailor_Mars>\t<http://example.org/friend>\t<http://dbpedia.org/resources/Usagi_Tsukino>\t.",16)
        );

        reducer.reduce(
                new TaggedTextItem("<http://dbpedia.org/resources/Usagi_Tsukino>",1),
                inputz,
                context
        );

        verifyNoMoreInteractions(context);
    }
}