1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 package org.apache.any23.openie;
18
19 import java.io.File;
20 import java.io.FileOutputStream;
21 import java.io.IOException;
22
23 import org.apache.any23.extractor.ExtractionContext;
24 import org.apache.any23.extractor.ExtractionException;
25 import org.apache.any23.extractor.ExtractionParameters;
26 import org.apache.any23.extractor.ExtractionResult;
27 import org.apache.any23.extractor.ExtractionResultImpl;
28 import org.apache.any23.plugin.extractor.openie.OpenIEExtractor;
29 import org.apache.any23.rdf.RDFUtils;
30 import org.apache.any23.util.StreamUtils;
31 import org.apache.any23.writer.RDFXMLWriter;
32 import org.apache.any23.writer.TripleHandler;
33 import org.apache.any23.writer.TripleHandlerException;
34 import org.eclipse.rdf4j.model.IRI;
35 import org.junit.After;
36 import org.junit.Before;
37 import org.junit.Test;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41
42
43
44
45 public class OpenIEExtractorTest {
46
47 private static final Logger logger = LoggerFactory.getLogger(OpenIEExtractorTest.class);
48
49 private OpenIEExtractor extractor;
50
51 @Before
52 public void setUp() throws Exception {
53 extractor = new OpenIEExtractor();
54 }
55
56 @After
57 public void tearDown() throws Exception {
58 extractor = null;
59 }
60
61 @Test
62 public void testExtractFromHTMLDocument()
63 throws IOException, ExtractionException, TripleHandlerException {
64 final IRI uri = RDFUtils.iri("http://podaac.jpl.nasa.gov/aquarius");
65 extract(uri, "/org/apache/any23/extractor/openie/example-openie.html");
66 }
67
68 public void extract(IRI uri, String filePath)
69 throws IOException, ExtractionException, TripleHandlerException {
70 FileOutputStream fos = new FileOutputStream(File.createTempFile("OpenIEExtractorTest", "tmp"));
71 final TripleHandler tHandler = new RDFXMLWriter(fos);
72 final ExtractionContext extractionContext = new ExtractionContext("rdf-openie", uri);
73 final ExtractionResult result = new ExtractionResultImpl(extractionContext, extractor, tHandler);
74 try {
75 extractor.run(
76 ExtractionParameters.newDefault(),
77 extractionContext,
78 StreamUtils.inputStreamToDocument(this.getClass().getResourceAsStream(filePath)),
79 result
80 );
81 } finally {
82 logger.debug(fos.toString());
83 tHandler.close();
84 result.close();
85 }
86 }
87
88 }