Thomas Tanon
5 years ago
committed by
GitHub
commit
033ee373d2
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,240 @@ |
||||
use rudf::model::*; |
||||
use rudf::{GraphSyntax, Repository, RepositoryConnection, MemoryRepository, Result}; |
||||
use rudf::sparql::{BindingsIterator, GraphPattern, PreparedQuery, QueryOptions, QueryResult, ServiceHandler}; |
||||
use failure::format_err; |
||||
use std::io::BufRead; |
||||
|
||||
|
||||
|
||||
#[test] |
||||
fn simple_service_test() { |
||||
|
||||
struct TestServiceHandler; |
||||
impl ServiceHandler for TestServiceHandler { |
||||
fn handle<'a>(&'a self, _named_node: NamedNode) -> Option<(fn(GraphPattern) -> Result<BindingsIterator<'a>>)> { |
||||
fn pattern_handler<'a>(graph_pattern: GraphPattern) -> Result<BindingsIterator<'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()) |
||||
}; |
||||
Some(pattern_handler) |
||||
} |
||||
} |
||||
|
||||
let query = r#" |
||||
SELECT ?s ?p ?o |
||||
WHERE |
||||
{
|
||||
SERVICE <http://service1.org>
|
||||
{ ?s ?p ?o |
||||
} |
||||
} |
||||
"#.to_string(); |
||||
|
||||
|
||||
let options = QueryOptions::default().with_service_handler(Box::new(TestServiceHandler)); |
||||
let results = do_query(b"".as_ref(), query, options).unwrap(); |
||||
let collected = results.into_values_iter().map(move |b| b.unwrap()).collect::<Vec<_>>(); |
||||
let solution = vec![ |
||||
vec![ Some(ex(String::from("s"))), Some(ex(String::from("p"))), Some(ex(String::from("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) -> Option<(fn(GraphPattern) -> Result<BindingsIterator<'a>>)> { |
||||
let service1 = NamedNode::parse("http://service1.org").unwrap(); |
||||
let service2 = NamedNode::parse("http://service2.org").unwrap(); |
||||
if named_node == service1 { Some(TwoServiceTest::handle_service1) } |
||||
else if named_node == service2 { Some(TwoServiceTest::handle_service2) } |
||||
else { None }
|
||||
} |
||||
} |
||||
|
||||
|
||||
impl TwoServiceTest { |
||||
|
||||
fn handle_service1<'a>(graph_pattern: GraphPattern) -> Result<BindingsIterator<'a>> { |
||||
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()) |
||||
} |
||||
|
||||
fn handle_service2<'a>(graph_pattern: GraphPattern) -> Result<BindingsIterator<'a>> { |
||||
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()) |
||||
} |
||||
|
||||
} |
||||
|
||||
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(Box::new(TwoServiceTest)); |
||||
let results = do_query(b"".as_ref(), query, options).unwrap(); |
||||
let collected = results.into_values_iter().map(move |b| b.unwrap()).collect::<Vec<_>>(); |
||||
let solution = vec![ |
||||
vec![ Some(literal("Alice".to_string())), Some(mailto("alice@example.com".to_string())) ], |
||||
vec![ Some(literal("Bob".to_string())), Some(mailto("bob@example.com".to_string())) ], |
||||
]; |
||||
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, _named_node: NamedNode) -> Option<(fn(GraphPattern) -> Result<BindingsIterator<'a>>)> { |
||||
Some(ServiceTest::handle_service)
|
||||
} |
||||
} |
||||
|
||||
impl ServiceTest { |
||||
fn handle_service<'a>(_graph_pattern: GraphPattern) -> Result<BindingsIterator<'a>> { |
||||
Err(format_err!("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(Box::new(ServiceTest)); |
||||
let results = do_query(triples, query, options).unwrap(); |
||||
let collected = results.into_values_iter().map(move |b| b.unwrap()).collect::<Vec<_>>(); |
||||
println!("Collected: {:?}", collected); |
||||
assert_eq!(collected.len(), 0); |
||||
} |
||||
|
||||
#[test] |
||||
fn non_silent_service_test() { |
||||
|
||||
#[derive(Clone,Copy)] |
||||
struct ServiceTest; |
||||
impl ServiceHandler for ServiceTest { |
||||
fn handle<'a>(&'a self, _named_node: NamedNode) -> Option<(fn(GraphPattern) -> Result<BindingsIterator<'a>>)> { |
||||
Some(ServiceTest::handle_service)
|
||||
} |
||||
} |
||||
|
||||
impl ServiceTest { |
||||
fn handle_service<'a>(_graph_pattern: GraphPattern) -> Result<BindingsIterator<'a>> { |
||||
Err(format_err!("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(Box::new(ServiceTest)); |
||||
let results = do_query(triples, query, options).unwrap(); |
||||
let result = results.into_values_iter().next(); |
||||
match result { |
||||
Some(Err(_)) => assert_eq!(true, true), |
||||
_ => assert_eq!(true, false, "This should have been an error since the service fails"), |
||||
} |
||||
} |
||||
|
||||
fn ex(id: String) -> Term { |
||||
Term::NamedNode(NamedNode::parse(format!("http://example.com/{}", &id)).unwrap()) |
||||
} |
||||
|
||||
fn mailto(id: String) -> Term { |
||||
Term::NamedNode(NamedNode::parse(format!("mailto:{}", &id)).unwrap()) |
||||
} |
||||
|
||||
fn literal(str: String) -> Term { |
||||
Term::Literal(Literal::new_simple_literal(str)) |
||||
} |
||||
|
||||
fn make_repository(reader: impl BufRead) -> Result<MemoryRepository> { |
||||
let repository = MemoryRepository::default(); |
||||
let mut connection = repository.connection()?; |
||||
connection.load_graph(reader, GraphSyntax::NTriples, None, None).unwrap(); |
||||
Ok(repository) |
||||
} |
||||
|
||||
fn query_repository<'a>(repository: MemoryRepository, query: String, options: QueryOptions<'a>) -> Result<BindingsIterator<'a>> { |
||||
let connection = repository.connection()?; |
||||
let prepared_query = connection.prepare_query(&query, None)?; |
||||
let result = prepared_query.exec(&options)?; |
||||
match result { |
||||
QueryResult::Bindings(iterator) => { |
||||
let (varaibles, iter) = iterator.destruct(); |
||||
let collected = iter.collect::<Vec<_>>(); |
||||
Ok(BindingsIterator::new(varaibles, Box::new(collected.into_iter()))) |
||||
}, |
||||
_ => Err(format_err!("Excpected bindings but got another QueryResult")) |
||||
} |
||||
}
|
||||
|
||||
fn pattern_repository<'a>(repository: MemoryRepository, pattern: GraphPattern, options: QueryOptions<'a>) -> Result<BindingsIterator<'a>> { |
||||
let connection = repository.connection()?; |
||||
let prepared_query = connection.prepare_query_from_pattern(&pattern, None)?; |
||||
let result = prepared_query.exec(&options)?; |
||||
match result { |
||||
QueryResult::Bindings(iterator) => { |
||||
let (varaibles, iter) = iterator.destruct(); |
||||
let collected = iter.collect::<Vec<_>>(); |
||||
Ok(BindingsIterator::new(varaibles, Box::new(collected.into_iter()))) |
||||
} |
||||
_ => Err(format_err!("Excpected bindings but got another QueryResult")) |
||||
} |
||||
} |
||||
|
||||
fn do_query<'a>(reader: impl BufRead, query: String, options: QueryOptions<'a>) -> Result<BindingsIterator<'a>> { |
||||
let repository = make_repository(reader)?; |
||||
query_repository(repository, query, options) |
||||
} |
||||
|
||||
fn do_pattern<'a>(reader: impl BufRead, pattern: GraphPattern, options: QueryOptions<'a>) -> Result<BindingsIterator<'a>> { |
||||
let repository = make_repository(reader)?; |
||||
pattern_repository(repository, pattern, options) |
||||
} |
||||
Loading…
Reference in new issue