@Before
public void setup() {
pse3mapper = new PSE3Mapper();
pse3mapper.mos = mock(RealMultipleOutputs.class);
pse3mapper.accepted = mock(KeyValueAcceptor.class);
pse3mapper.rejected = mock(KeyValueAcceptor.class);
}
@Test
public void rejectsABadTriple() throws IOException, InterruptedException {
Context mockContext = mock(Context.class);
pse3mapper.map(
new LongWritable(24562L),
new Text("<http://example.com/A>\t<http://example.com/B>\t\"2001-06\"^^xsd:datetime."),
mockContext);
verify(pse3mapper.rejected)
.write(
new Text("<http://example.com/A>"),
new Text("<http://example.com/B>\t\"2001-06\"^^xsd:datetime\t."),
mockContext);
verifyNoMoreInteractions(pse3mapper.accepted);
verifyNoMoreInteractions(pse3mapper.rejected);
}
@Test
public void arnoldSchwarzeneggerIsAFilmActor() throws IOException, InterruptedException {
Context mockContext = mock(Context.class);
pse3mapper.map(
new LongWritable(944L),
new Text(
"<http://rdf.basekb.com/ns/m.0tc7> <http://www.w3.org/1999/02/22-rdf-syntax-ns#type>\t<http://rdf.basekb.com/ns/film.actor>."),
mockContext);
verify(pse3mapper.accepted)
.write(
new WritableTriple(
Node_URI.createURI("http://rdf.basekb.com/ns/m.0tc7"),
Node_URI.createURI("http://www.w3.org/1999/02/22-rdf-syntax-ns#type"),
Node_URI.createURI("http://rdf.basekb.com/ns/film.actor")),
new LongWritable(1),
mockContext);
verifyNoMoreInteractions(pse3mapper.accepted);
}
@Test
public void acceptsARealFreebaseTriple() throws IOException, InterruptedException {
Context mockContext = mock(Context.class);
pse3mapper.map(
new LongWritable(944L),
new Text(
"<http://rdf.basekb.com/ns/m.06fm3lj>\t<http://rdf.basekb.com/ns/book.written_work.author>\t<http://rdf.basekb.com/ns/m.03qp7yf>."),
mockContext);
verify(pse3mapper.accepted)
.write(
new WritableTriple(
Node_URI.createURI("http://rdf.basekb.com/ns/m.06fm3lj"),
Node_URI.createURI("http://rdf.basekb.com/ns/book.written_work.author"),
Node_URI.createURI("http://rdf.basekb.com/ns/m.03qp7yf")),
new LongWritable(1),
mockContext);
verifyNoMoreInteractions(pse3mapper.accepted);
}
@Test
public void acceptsAGoodTriple() throws IOException, InterruptedException {
Context mockContext = mock(Context.class);
pse3mapper.map(
new LongWritable(944L),
new Text("<http://example.com/A>\t<http://example.com/B>\t<http://example.com/C>."),
mockContext);
verify(pse3mapper.accepted)
.write(
new WritableTriple(
Node_URI.createURI("http://example.com/A"),
Node_URI.createURI("http://example.com/B"),
Node_URI.createURI("http://example.com/C")),
new LongWritable(1),
mockContext);
verifyNoMoreInteractions(pse3mapper.accepted);
verifyNoMoreInteractions(pse3mapper.rejected);
}
@Test
public void rejects$Escapes() throws IOException, InterruptedException {
Context mockContext = mock(Context.class);
pse3mapper.map(
new LongWritable(944L),
new Text(
"<http://rdf.basekb.com/ns/m.0tc7> <http://rdf.basekb.com/ns/common.topic.topic_equivalent_webpage> <http://www.ranker.com/review/arnold-schwarzenegger$002F493404> ."),
mockContext);
verify(pse3mapper.accepted)
.write(
new WritableTriple(
Node_URI.createURI("http://rdf.basekb.com/ns/m.0tc7"),
Node_URI.createURI(
"http://rdf.basekb.com/ns/common.topic.topic_equivalent_webpage"),
Node_URI.createURI("http://www.ranker.com/review/arnold-schwarzenegger/493404")),
new LongWritable(1),
mockContext);
verifyNoMoreInteractions(pse3mapper.accepted);
}
@Test
public void closesMosOnShutdown() throws IOException, InterruptedException {
pse3mapper.cleanup(mock(Context.class));
verify(pse3mapper.mos).close();
verifyNoMoreInteractions(pse3mapper.mos);
}