1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18 package org.apache.any23.extractor.calendar;
19
20 import org.apache.any23.extractor.html.AbstractExtractorTestCase;
21 import org.eclipse.rdf4j.model.BNode;
22 import org.eclipse.rdf4j.model.Statement;
23 import org.eclipse.rdf4j.repository.RepositoryException;
24 import org.eclipse.rdf4j.rio.RDFFormat;
25 import org.eclipse.rdf4j.rio.RDFHandler;
26 import org.eclipse.rdf4j.rio.RDFHandlerException;
27 import org.eclipse.rdf4j.rio.RDFParseException;
28 import org.eclipse.rdf4j.rio.RDFParser;
29 import org.eclipse.rdf4j.rio.Rio;
30 import org.eclipse.rdf4j.rio.helpers.ParseErrorCollector;
31 import org.junit.Assert;
32
33 import java.io.File;
34 import java.io.FileReader;
35 import java.io.IOException;
36 import java.nio.charset.StandardCharsets;
37 import java.util.ArrayList;
38 import java.util.List;
39
40 abstract class BaseCalendarExtractorTest extends AbstractExtractorTestCase {
41
42 abstract String filePrefix();
43
44 protected void extractAndVerifyAgainstNQuads(String actual, String expected)
45 throws RepositoryException, RDFHandlerException, IOException, RDFParseException {
46 String filePrefix = filePrefix();
47 assertExtract(filePrefix + actual);
48 assertModelNotEmpty();
49 List<Statement> expectedStatements = loadResultStatement(filePrefix + expected);
50 int actualStmtSize = getStatementsSize(null, null, null);
51 Assert.assertEquals(expectedStatements.size(), actualStmtSize);
52 for (Statement statement : expectedStatements) {
53 assertContains(statement.getSubject() instanceof BNode ? null : statement.getSubject(),
54 statement.getPredicate(), statement.getObject() instanceof BNode ? null : statement.getObject());
55 }
56 }
57
58 private List<Statement> loadResultStatement(String resultFilePath)
59 throws RDFHandlerException, IOException, RDFParseException {
60 RDFParser nQuadsParser = Rio.createParser(RDFFormat.NQUADS);
61 TestRDFHandler rdfHandler = new TestRDFHandler();
62 nQuadsParser.setRDFHandler(rdfHandler);
63 nQuadsParser.setParseErrorListener(new ParseErrorCollector());
64 File file = copyResourceToTempFile(resultFilePath);
65 nQuadsParser.parse(new FileReader(file, StandardCharsets.UTF_8), baseIRI.stringValue());
66 return rdfHandler.getStatements();
67 }
68
69 private static class TestRDFHandler implements RDFHandler {
70
71 private final List<Statement> statements = new ArrayList<Statement>();
72
73 protected List<Statement> getStatements() {
74 return statements;
75 }
76
77 public void startRDF() throws RDFHandlerException {
78 }
79
80 public void endRDF() throws RDFHandlerException {
81 }
82
83 public void handleNamespace(String s, String s1) throws RDFHandlerException {
84 throw new UnsupportedOperationException();
85 }
86
87 public void handleStatement(Statement statement) throws RDFHandlerException {
88 statements.add(statement);
89 }
90
91 public void handleComment(String s) throws RDFHandlerException {
92 }
93 }
94 }