Java Code Examples for org.eclipse.rdf4j.query.QueryResults#report()

protected void doTupleLinksAndStylesheet(TupleQueryResultFormat format, TupleQueryResult input,
		TupleQueryResult expected, List<String> links, String stylesheetUrl) throws QueryResultHandlerException,
		QueryEvaluationException, QueryResultParseException, UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.handleStylesheet(stylesheetUrl);
	writer.startHeader();
	writer.handleLinks(links);
	QueryResults.report(input, writer);

	// System.out.println("output: " + out.toString("UTF-8"));

	ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
	TupleQueryResult output = parseTupleInternal(format, in);

	assertQueryResultsEqual(expected, output);
}
 

@Test
public void testmultipleVarResults() throws Exception {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(createTupleMultipleBindingSets(), writer);

	assertRegex("a,b,c\r\n" + "foo:bar,_:bnode,baz\r\n" + "1,,Hello World!\r\n"
			+ "http://example.org/test/ns/bindingA,http://example.com/other/ns/bindingB,\"http://example.com/other/ns/binding,C\"\r\n"
			+ "\"string with newline at the end       \n\",string with space at the end         ,    \r\n"
			+ "''single-quoted string,\"\"\"\"\"double-quoted string\",\t\tunencoded tab characters followed by encoded \t\t(\r\n)?",
			out.toString("UTF-8"));
}
 

@Test
public void testSingleVarResults() throws Exception {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(createTupleSingleVarMultipleBindingSets(), writer);

	System.out.println(out.toString("UTF-8"));
	assertRegex("a\r\n" + "foo:bar\r\n" + "2.0(E0)?\r\n" + "_:bnode3\r\n" + "''single-quoted string\r\n"
			+ "\"\"\"\"\"double-quoted string\"\r\n" + "space at the end         \r\n"
			+ "space at the end         \r\n" + "\"\"\"\"\"double-quoted string with no datatype\"\r\n"
			+ "\"newline at the end \n\"(\r\n)?"
			+ "urn:rdf4j:triple:PDw8dXJuOmE-IDxodHRwOi8vd3d3LnczLm9yZy8xOTk5LzAyLzIyLXJkZi1zeW50YXgtbnMjdHlwZT4gPHVybjpiPj4-(\r\n)?",
			out.toString("UTF-8"));
}
 

protected void doTupleLinksAndStylesheetNoStarts(TupleQueryResultFormat format, TupleQueryResult input,
		TupleQueryResult expected, List<String> links, String stylesheetUrl) throws QueryResultHandlerException,
		QueryEvaluationException, QueryResultParseException, UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	// Test for handling when startDocument and startHeader are not called
	writer.handleStylesheet(stylesheetUrl);
	writer.handleLinks(links);
	QueryResults.report(input, writer);

	// System.out.println("output: " + out.toString("UTF-8"));

	ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
	TupleQueryResult output = parseTupleInternal(format, in);

	assertQueryResultsEqual(expected, output);
}
 

protected void doTupleLinksAndStylesheetAndNamespaces(TupleQueryResultFormat format, TupleQueryResult input,
		TupleQueryResult expected, List<String> links, String stylesheetUrl, Map<String, String> namespaces)
		throws QueryResultHandlerException, QueryEvaluationException, QueryResultParseException,
		UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	for (String nextPrefix : namespaces.keySet()) {
		writer.handleNamespace(nextPrefix, namespaces.get(nextPrefix));
	}
	writer.startDocument();
	writer.handleStylesheet(stylesheetUrl);
	writer.startHeader();
	writer.handleLinks(links);
	QueryResults.report(input, writer);

	// System.out.println("output: " + out.toString("UTF-8"));

	ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
	TupleQueryResult output = parseTupleInternal(format, in);

	assertQueryResultsEqual(expected, output);
}
 

protected void doTupleLinks(TupleQueryResultFormat format, TupleQueryResult input, TupleQueryResult expected,
		List<String> links) throws QueryResultHandlerException, QueryEvaluationException, QueryResultParseException,
		UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(links);
	QueryResults.report(input, writer);

	// System.out.println("output: " + out.toString("UTF-8"));

	ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
	TupleQueryResult output = parseTupleInternal(format, in);

	assertQueryResultsEqual(expected, output);
}
 

/**
 * Test specifically for JSONP callback support.
 */
protected void doTupleJSONPCallback(TupleQueryResultFormat format, TupleQueryResult input,
		TupleQueryResult expected) throws QueryResultHandlerException, QueryEvaluationException,
		QueryResultParseException, UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);

	// only do this test if the callback is enabled
	if (writer.getSupportedSettings().contains(BasicQueryWriterSettings.JSONP_CALLBACK)) {

		String callback = "nextfunctionname" + Math.abs(new Random().nextInt());

		writer.getWriterConfig().set(BasicQueryWriterSettings.JSONP_CALLBACK, callback);

		QueryResults.report(input, writer);

		String result = out.toString("UTF-8");

		// System.out.println("output: " + result);

		assertTrue(result.startsWith(callback + "("));
		assertTrue(result.endsWith(");"));

		// Strip off the callback function and verify that it contains a
		// valid
		// JSON object containing the correct results
		result = result.substring(callback.length() + 1, result.length() - 2);

		ByteArrayInputStream in = new ByteArrayInputStream(result.getBytes("UTF-8"));
		TupleQueryResult output = parseTupleInternal(format, in);

		assertQueryResultsEqual(expected, output);
	}
}
 

/**
 * Test specifically for QName support.
 */
protected void doTupleLinksAndStylesheetAndNamespacesQName(TupleQueryResultFormat format, TupleQueryResult input,
		TupleQueryResult expected, List<String> links, String stylesheetUrl, Map<String, String> namespaces)
		throws QueryResultHandlerException, QueryEvaluationException, QueryResultParseException,
		UnsupportedQueryResultFormatException, IOException {
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	if (writer.getSupportedSettings().contains(BasicQueryWriterSettings.ADD_SESAME_QNAME)) {
		// System.out.println("Enabling Sesame qname support");
		writer.getWriterConfig().set(BasicQueryWriterSettings.ADD_SESAME_QNAME, true);
	}

	for (String nextPrefix : namespaces.keySet()) {
		writer.handleNamespace(nextPrefix, namespaces.get(nextPrefix));
	}
	writer.startDocument();
	writer.handleStylesheet(stylesheetUrl);
	writer.startHeader();
	writer.handleLinks(links);
	QueryResults.report(input, writer);

	String result = out.toString("UTF-8");

	// System.out.println("output: " + result);

	ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
	TupleQueryResult output = parseTupleInternal(format, in);

	assertQueryResultsEqual(expected, output);

	// only do this additional test if sesame q:qname is supported by this
	// writer
	if (writer.getSupportedSettings().contains(BasicQueryWriterSettings.ADD_SESAME_QNAME)) {
		assertTrue(result.contains("test:bindingA"));
		assertFalse(result.contains("other:bindingB"));
		assertTrue(result.contains("other:binding,C"));
	}

}
 

/**
 * Writes a graph query result document in a specific RDF format to an output stream.
 *
 * @param gqr    The query result to write.
 * @param format The file format of the document to write.
 * @param out    An OutputStream to write the document to.
 * @throws IOException                  If an I/O error occurred while writing the query result document to the
 *                                      stream.
 * @throws RDFHandlerException          If such an exception is thrown by the used RDF writer.
 * @throws QueryEvaluationException
 * @throws UnsupportedRDFormatException If an unsupported query result file format was specified.
 */
public static void writeGraph(GraphQueryResult gqr, RDFFormat format, OutputStream out)
		throws IOException, RDFHandlerException, UnsupportedRDFormatException, QueryEvaluationException {
	RDFWriter writer = Rio.createWriter(format, out);
	try {
		QueryResults.report(gqr, writer);
	} catch (RDFHandlerException e) {
		if (e.getCause() instanceof IOException) {
			throw (IOException) e.getCause();
		} else {
			throw e;
		}
	}
}
 

private String toString(TupleQueryResult results) throws QueryResultHandlerException,
		TupleQueryResultHandlerException, QueryEvaluationException, UnsupportedEncodingException {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(results, writer);

	return out.toString("UTF-8");
}
 

private String toString(TupleQueryResult results) throws QueryResultHandlerException,
		TupleQueryResultHandlerException, QueryEvaluationException, UnsupportedEncodingException {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(results, writer);

	return out.toString("UTF-8");
}
 

@Test
public void testEndOfLine() throws Exception {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(createTupleNoBindingSets(), writer);

	assertEquals("\r\n", out.toString("UTF-8").replaceAll("\\S+", ""));
}
 

@Test
public void testEndOfLine() throws Exception {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(createTupleNoBindingSets(), writer);

	assertEquals("\r\n", out.toString("UTF-8").replaceAll("\\S+", ""));
}
 

@Test
public void testEmptyResults() throws Exception {
	TupleQueryResultFormat format = getTupleFormat();
	ByteArrayOutputStream out = new ByteArrayOutputStream(4096);
	TupleQueryResultWriter writer = QueryResultIO.createTupleWriter(format, out);
	writer.startDocument();
	writer.startHeader();
	writer.handleLinks(Arrays.<String>asList());
	QueryResults.report(createTupleNoBindingSets(), writer);

	assertRegex("a,b,c(\r\n)?", out.toString("UTF-8"));
}
 

@Override
public void evaluate(TupleQueryResultHandler handler)
		throws QueryEvaluationException, TupleQueryResultHandlerException {
	TupleQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}
 

@Override
public void evaluate(RDFHandler handler) throws QueryEvaluationException, RDFHandlerException {
	GraphQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}
 

@Override
public void evaluate(RDFHandler handler) throws QueryEvaluationException, RDFHandlerException {
	GraphQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}
 

@Override
public void evaluate(TupleQueryResultHandler handler)
		throws QueryEvaluationException, TupleQueryResultHandlerException {
	TupleQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}
 

@Override
public void evaluate(RDFHandler handler) throws QueryEvaluationException, RDFHandlerException {
	GraphQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}
 

@Override
public void evaluate(TupleQueryResultHandler handler)
		throws QueryEvaluationException, TupleQueryResultHandlerException {
	TupleQueryResult queryResult = evaluate();
	QueryResults.report(queryResult, handler);
}