Tpt
4 years ago
parent
36bc870ca8
commit
a3dce12d94
File diff suppressed because it is too large
Load Diff
@ -1,262 +0,0 @@ |
|||||||
use oxigraph::model::*; |
|
||||||
use oxigraph::sparql::*; |
|
||||||
use oxigraph::*; |
|
||||||
use std::io::BufRead; |
|
||||||
|
|
||||||
#[test] |
|
||||||
fn simple_service_test() { |
|
||||||
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()) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
let query = r#" |
|
||||||
SELECT ?s ?p ?o |
|
||||||
WHERE |
|
||||||
{
|
|
||||||
SERVICE <http://service1.org>
|
|
||||||
{ ?s ?p ?o |
|
||||||
} |
|
||||||
} |
|
||||||
"# |
|
||||||
.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<_>>() |
|
||||||
}) |
|
||||||
.collect::<Vec<_>>(); |
|
||||||
let solution = vec![vec![ex("s"), ex("p"), ex("o")]]; |
|
||||||
assert_eq!(collected, solution); |
|
||||||
} |
|
||||||
|
|
||||||
#[test] |
|
||||||
fn two_service_test() { |
|
||||||
#[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#" |
|
||||||
<http://example.com/bob> <http://xmlns.com/foaf/0.1/name> "Bob" .
|
|
||||||
<http://example.com/alice> <http://xmlns.com/foaf/0.1/name> "Alice" .
|
|
||||||
"# |
|
||||||
.as_ref(); |
|
||||||
do_pattern(triples, graph_pattern, QueryOptions::default()) |
|
||||||
} else if named_node == &service2 { |
|
||||||
let triples = br#" |
|
||||||
<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> .
|
|
||||||
"# |
|
||||||
.as_ref(); |
|
||||||
do_pattern(triples, graph_pattern, QueryOptions::default()) |
|
||||||
} else { |
|
||||||
Err(Error::msg("not found")) |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
let query = r#" |
|
||||||
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 |
|
||||||
"# |
|
||||||
.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<_>>() |
|
||||||
}) |
|
||||||
.collect::<Vec<_>>(); |
|
||||||
let solution = vec![ |
|
||||||
vec![literal("Alice"), mailto("alice@example.com")], |
|
||||||
vec![literal("Bob"), mailto("bob@example.com")], |
|
||||||
]; |
|
||||||
assert_eq!(collected, solution); |
|
||||||
} |
|
||||||
|
|
||||||
#[test] |
|
||||||
fn silent_service_empty_set_test() { |
|
||||||
#[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")) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
let query = r#" |
|
||||||
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
|
|
||||||
SELECT ?name ?mbox
|
|
||||||
WHERE |
|
||||||
{
|
|
||||||
SERVICE SILENT <http://service1.org>
|
|
||||||
{ ?s foaf:name ?name |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
ORDER BY ?name |
|
||||||
"# |
|
||||||
.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); |
|
||||||
} |
|
||||||
|
|
||||||
#[test] |
|
||||||
fn non_silent_service_test() { |
|
||||||
#[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")) |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
let query = r#" |
|
||||||
PREFIX foaf: <http://xmlns.com/foaf/0.1/>
|
|
||||||
SELECT ?name |
|
||||||
WHERE |
|
||||||
{
|
|
||||||
SERVICE <http://service1.org>
|
|
||||||
{ ?s foaf:name ?name |
|
||||||
} |
|
||||||
|
|
||||||
} |
|
||||||
ORDER BY ?name |
|
||||||
"# |
|
||||||
.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") |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
fn ex(id: &str) -> Term { |
|
||||||
Term::NamedNode(NamedNode::new(format!("http://example.com/{}", id)).unwrap()) |
|
||||||
} |
|
||||||
|
|
||||||
fn mailto(id: &str) -> Term { |
|
||||||
Term::NamedNode(NamedNode::new(format!("mailto:{}", id)).unwrap()) |
|
||||||
} |
|
||||||
|
|
||||||
fn literal(str: &str) -> Term { |
|
||||||
Term::Literal(Literal::new_simple_literal(str)) |
|
||||||
} |
|
||||||
|
|
||||||
fn make_store(reader: impl BufRead) -> Result<MemoryStore> { |
|
||||||
let store = MemoryStore::new(); |
|
||||||
store |
|
||||||
.load_graph( |
|
||||||
reader, |
|
||||||
GraphSyntax::NTriples, |
|
||||||
&GraphName::DefaultGraph, |
|
||||||
None, |
|
||||||
) |
|
||||||
.unwrap(); |
|
||||||
Ok(store) |
|
||||||
} |
|
||||||
|
|
||||||
fn query_store( |
|
||||||
store: MemoryStore, |
|
||||||
query: &str, |
|
||||||
options: QueryOptions, |
|
||||||
) -> Result<QuerySolutionsIterator<'_>> { |
|
||||||
match store.prepare_query(query, options)?.exec()? { |
|
||||||
QueryResult::Solutions(iterator) => { |
|
||||||
let (variables, iter) = iterator.destruct(); |
|
||||||
let collected = iter.collect::<Vec<_>>(); |
|
||||||
Ok(QuerySolutionsIterator::new( |
|
||||||
variables, |
|
||||||
Box::new(collected.into_iter()), |
|
||||||
)) |
|
||||||
} |
|
||||||
_ => Err(Error::msg("Excpected bindings but got another QueryResult")), |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
fn pattern_store<'a>( |
|
||||||
store: MemoryStore, |
|
||||||
pattern: &'a GraphPattern, |
|
||||||
options: QueryOptions, |
|
||||||
) -> Result<QuerySolutionsIterator<'a>> { |
|
||||||
match store |
|
||||||
.prepare_query_from_pattern(&pattern, options)? |
|
||||||
.exec()? |
|
||||||
{ |
|
||||||
QueryResult::Solutions(iterator) => { |
|
||||||
let (varaibles, iter) = iterator.destruct(); |
|
||||||
let collected = iter.collect::<Vec<_>>(); |
|
||||||
Ok(QuerySolutionsIterator::new( |
|
||||||
varaibles, |
|
||||||
Box::new(collected.into_iter()), |
|
||||||
)) |
|
||||||
} |
|
||||||
_ => Err(Error::msg("Expected bindings but got another QueryResult")), |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
fn do_query( |
|
||||||
reader: impl BufRead, |
|
||||||
query: &str, |
|
||||||
options: QueryOptions, |
|
||||||
) -> Result<QuerySolutionsIterator<'_>> { |
|
||||||
let store = make_store(reader)?; |
|
||||||
query_store(store, query, options) |
|
||||||
} |
|
||||||
|
|
||||||
fn do_pattern<'a>( |
|
||||||
reader: impl BufRead, |
|
||||||
pattern: &'a GraphPattern, |
|
||||||
options: QueryOptions, |
|
||||||
) -> Result<QuerySolutionsIterator<'a>> { |
|
||||||
let store = make_store(reader)?; |
|
||||||
pattern_store(store, pattern, options) |
|
||||||
} |
|
||||||
Loading…
Reference in new issue