Fork of https://github.com/oxigraph/oxigraph.git for the purpose of NextGraph project
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
oxigraph/lib/tests/service_test_cases.rs

263 lines
7.0 KiB

use oxigraph::model::*;
use oxigraph::sparql::*;
use oxigraph::*;
5 years ago
use std::io::BufRead;
5 years ago
#[test]
fn simple_service_test() {
5 years ago
struct TestServiceHandler;
impl ServiceHandler for TestServiceHandler {
fn handle<'a>(
&'a self,
_: &NamedNode,
graph_pattern: &'a GraphPattern,
) -> Result<QuerySolutionsIterator<'a>> {
let triples =
b"<http://example.com/s> <http://example.com/p> <http://example.com/o> .".as_ref();
do_pattern(triples, graph_pattern, QueryOptions::default())
5 years ago
}
5 years ago
}
5 years ago
let query = r#"
5 years ago
SELECT ?s ?p ?o
WHERE
{
SERVICE <http://service1.org>
{ ?s ?p ?o
}
}
5 years ago
"#
.to_string();
let options = QueryOptions::default().with_service_handler(TestServiceHandler);
let collected = do_query(b"".as_ref(), query, options)
.unwrap()
.map(|b| {
b.unwrap()
.iter()
.map(|(_, v)| v.clone())
.collect::<Vec<_>>()
})
5 years ago
.collect::<Vec<_>>();
let solution = vec![vec![ex("s"), ex("p"), ex("o")]];
5 years ago
assert_eq!(collected, solution);
5 years ago
}
5 years ago
#[test]
fn two_service_test() {
5 years ago
#[derive(Clone, Copy)]
struct TwoServiceTest;
impl ServiceHandler for TwoServiceTest {
fn handle<'a>(
&'a self,
named_node: &NamedNode,
graph_pattern: &'a GraphPattern,
) -> Result<QuerySolutionsIterator<'a>> {
let service1 = NamedNode::new("http://service1.org").unwrap();
let service2 = NamedNode::new("http://service2.org").unwrap();
if named_node == &service1 {
let triples = br#"
5 years ago
<http://example.com/bob> <http://xmlns.com/foaf/0.1/name> "Bob" .
<http://example.com/alice> <http://xmlns.com/foaf/0.1/name> "Alice" .
5 years ago
"#
.as_ref();
do_pattern(triples, graph_pattern, QueryOptions::default())
} else if named_node == &service2 {
let triples = br#"
5 years ago
<http://example.com/bob> <http://xmlns.com/foaf/0.1/mbox> <mailto:bob@example.com> .
<http://example.com/alice> <http://xmlns.com/foaf/0.1/mbox> <mailto:alice@example.com> .
5 years ago
"#
.as_ref();
do_pattern(triples, graph_pattern, QueryOptions::default())
} else {
Err(Error::msg("not found"))
}
5 years ago
}
5 years ago
}
5 years ago
let query = r#"
5 years ago
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
SELECT ?name ?mbox
WHERE
{
SERVICE <http://service1.org>
{ ?s foaf:name ?name
}
SERVICE <http://service2.org>
{ ?s foaf:mbox ?mbox
}
}
ORDER BY ?name
5 years ago
"#
.to_string();
let options = QueryOptions::default().with_service_handler(TwoServiceTest);
let collected = do_query(b"".as_ref(), query, options)
.unwrap()
.map(|b| {
b.unwrap()
.iter()
.map(|(_, v)| v.clone())
.collect::<Vec<_>>()
})
5 years ago
.collect::<Vec<_>>();
let solution = vec![
vec![literal("Alice"), mailto("alice@example.com")],
vec![literal("Bob"), mailto("bob@example.com")],
5 years ago
];
5 years ago
assert_eq!(collected, solution);
5 years ago
}
5 years ago
#[test]
5 years ago
fn silent_service_empty_set_test() {
5 years ago
#[derive(Clone, Copy)]
struct ServiceTest;
impl ServiceHandler for ServiceTest {
fn handle<'a>(
&'a self,
_: &NamedNode,
_: &'a GraphPattern,
) -> Result<QuerySolutionsIterator<'a>> {
Err(Error::msg("This is supposed to fail"))
5 years ago
}
5 years ago
}
5 years ago
let query = r#"
5 years ago
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
SELECT ?name ?mbox
WHERE
{
5 years ago
SERVICE SILENT <http://service1.org>
5 years ago
{ ?s foaf:name ?name
}
5 years ago
}
ORDER BY ?name
5 years ago
"#
.to_string();
let triples = b"".as_ref();
let options = QueryOptions::default().with_service_handler(ServiceTest);
assert_eq!(do_query(triples, query, options).unwrap().count(), 1);
5 years ago
}
5 years ago
5 years ago
#[test]
fn non_silent_service_test() {
5 years ago
#[derive(Clone, Copy)]
struct ServiceTest;
impl ServiceHandler for ServiceTest {
fn handle<'a>(
&'a self,
_: &NamedNode,
_: &'a GraphPattern,
) -> Result<QuerySolutionsIterator<'a>> {
Err(Error::msg("This is supposed to fail"))
5 years ago
}
5 years ago
}
5 years ago
let query = r#"
5 years ago
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
SELECT ?name
WHERE
{
SERVICE <http://service1.org>
{ ?s foaf:name ?name
}
5 years ago
}
ORDER BY ?name
5 years ago
"#
.to_string();
let triples = b"".as_ref();
let options = QueryOptions::default().with_service_handler(ServiceTest);
let mut solutions = do_query(triples, query, options).unwrap();
if let Some(Err(_)) = solutions.next() {
} else {
panic!("This should have been an error since the service fails")
5 years ago
}
5 years ago
}
fn ex(id: &str) -> Term {
Term::NamedNode(NamedNode::new(format!("http://example.com/{}", id)).unwrap())
5 years ago
}
fn mailto(id: &str) -> Term {
Term::NamedNode(NamedNode::new(format!("mailto:{}", id)).unwrap())
5 years ago
}
fn literal(str: &str) -> Term {
5 years ago
Term::Literal(Literal::new_simple_literal(str))
5 years ago
}
fn make_store(reader: impl BufRead) -> Result<MemoryStore> {
let store = MemoryStore::new();
store
.load_graph(
reader,
GraphSyntax::NTriples,
&GraphName::DefaultGraph,
None,
)
5 years ago
.unwrap();
Ok(store)
5 years ago
}
fn query_store<'a>(
store: MemoryStore,
5 years ago
query: String,
options: QueryOptions<'a>,
) -> Result<QuerySolutionsIterator<'a>> {
match store.prepare_query(&query, options)?.exec()? {
QueryResult::Solutions(iterator) => {
let (variables, iter) = iterator.destruct();
5 years ago
let collected = iter.collect::<Vec<_>>();
Ok(QuerySolutionsIterator::new(
variables,
5 years ago
Box::new(collected.into_iter()),
))
}
_ => Err(Error::msg("Excpected bindings but got another QueryResult")),
5 years ago
}
5 years ago
}
fn pattern_store<'a>(
store: MemoryStore,
pattern: &'a GraphPattern,
5 years ago
options: QueryOptions<'a>,
) -> Result<QuerySolutionsIterator<'a>> {
match store
.prepare_query_from_pattern(&pattern, options)?
.exec()?
{
QueryResult::Solutions(iterator) => {
5 years ago
let (varaibles, iter) = iterator.destruct();
let collected = iter.collect::<Vec<_>>();
Ok(QuerySolutionsIterator::new(
5 years ago
varaibles,
Box::new(collected.into_iter()),
))
}
_ => Err(Error::msg("Expected bindings but got another QueryResult")),
5 years ago
}
5 years ago
}
5 years ago
fn do_query<'a>(
reader: impl BufRead,
query: String,
options: QueryOptions<'a>,
) -> Result<QuerySolutionsIterator<'a>> {
let store = make_store(reader)?;
query_store(store, query, options)
5 years ago
}
5 years ago
5 years ago
fn do_pattern<'a>(
reader: impl BufRead,
pattern: &'a GraphPattern,
5 years ago
options: QueryOptions<'a>,
) -> Result<QuerySolutionsIterator<'a>> {
let store = make_store(reader)?;
pattern_store(store, pattern, options)
5 years ago
}