| /* |
| * * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.rya.mongodb; |
| |
| import static org.junit.Assert.assertEquals; |
| |
| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| import java.util.Collection; |
| import java.util.Map; |
| import java.util.Map.Entry; |
| |
| import org.apache.rya.api.RdfCloudTripleStoreUtils; |
| import org.apache.rya.api.domain.RyaStatement; |
| import org.apache.rya.api.domain.RyaStatement.RyaStatementBuilder; |
| import org.apache.rya.api.domain.RyaURI; |
| import org.junit.Test; |
| import org.openrdf.model.impl.URIImpl; |
| import org.openrdf.query.BindingSet; |
| import org.openrdf.query.impl.MapBindingSet; |
| |
| import com.google.common.collect.Lists; |
| |
| import info.aduna.iteration.CloseableIteration; |
| /** |
| * Integration tests the methods of {@link MongoDBQueryEngine}. |
| */ |
| public class MongoDBQueryEngineIT extends MongoITBase { |
| |
| private RyaStatement getStatement(final String s, final String p, final String o) { |
| final RyaStatementBuilder builder = new RyaStatementBuilder(); |
| if (s != null) { |
| builder.setSubject(new RyaURI(s)); |
| } |
| if (p != null) { |
| builder.setPredicate(new RyaURI(p)); |
| } |
| if (o != null) { |
| builder.setObject(new RyaURI(o)); |
| } |
| return builder.build(); |
| } |
| |
| public int size(final CloseableIteration<?, ?> iter) throws Exception { |
| int i = 0; |
| while (iter.hasNext()) { |
| i++; |
| iter.next(); |
| } |
| return i; |
| } |
| |
| @Test |
| public void statementQuery() throws Exception { |
| final MongoDBRyaDAO dao = new MongoDBRyaDAO(); |
| try(final MongoDBQueryEngine engine =new MongoDBQueryEngine()) { |
| engine.setConf(conf); |
| |
| // Add data. |
| dao.setConf(conf); |
| dao.init(); |
| dao.add(getStatement("u:a", "u:tt", "u:b")); |
| dao.add(getStatement("u:a", "u:tt", "u:c")); |
| |
| final RyaStatement s = getStatement("u:a", null, null); |
| assertEquals(2, size(engine.query(s, conf))); |
| } finally { |
| dao.destroy(); |
| } |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void batchbindingSetsQuery() throws Exception { |
| final MongoDBRyaDAO dao = new MongoDBRyaDAO(); |
| try(final MongoDBQueryEngine engine =new MongoDBQueryEngine()) { |
| engine.setConf(conf); |
| |
| // Add data. |
| dao.setConf(conf); |
| dao.init(); |
| dao.add(getStatement("u:a", "u:tt", "u:b")); |
| dao.add(getStatement("u:a", "u:tt", "u:c")); |
| |
| // Run the test. |
| final RyaStatement s1 = getStatement(null, null, "u:b"); |
| |
| final MapBindingSet bs1 = new MapBindingSet(); |
| bs1.addBinding("foo", new URIImpl("u:x")); |
| |
| final Map.Entry<RyaStatement, BindingSet> e1 = new RdfCloudTripleStoreUtils.CustomEntry<>(s1, bs1); |
| final Collection<Entry<RyaStatement, BindingSet>> stmts1 = Lists.newArrayList(e1); |
| assertEquals(1, size(engine.queryWithBindingSet(stmts1, conf))); |
| |
| |
| final MapBindingSet bs2 = new MapBindingSet(); |
| bs2.addBinding("foo", new URIImpl("u:y")); |
| |
| final RyaStatement s2 = getStatement(null, null, "u:c"); |
| |
| final Map.Entry<RyaStatement, BindingSet> e2 = new RdfCloudTripleStoreUtils.CustomEntry<>(s2, bs2); |
| |
| final Collection<Entry<RyaStatement, BindingSet>> stmts2 = Lists.newArrayList(e1, e2); |
| assertEquals(2, size(engine.queryWithBindingSet(stmts2, conf))); |
| |
| |
| final Map.Entry<RyaStatement, BindingSet> e3 = new RdfCloudTripleStoreUtils.CustomEntry<>(s2, bs1); |
| final Map.Entry<RyaStatement, BindingSet> e4 = new RdfCloudTripleStoreUtils.CustomEntry<>(s1, bs2); |
| |
| final Collection<Entry<RyaStatement, BindingSet>> stmts3 = Lists.newArrayList(e1, e2, e3, e4); |
| assertEquals(4, size(engine.queryWithBindingSet(stmts3, conf))); |
| } finally { |
| dao.destroy(); |
| } |
| } |
| |
| @SuppressWarnings("unchecked") |
| @Test |
| public void bindingSetsQuery() throws Exception { |
| final MongoDBRyaDAO dao = new MongoDBRyaDAO(); |
| try(final MongoDBQueryEngine engine =new MongoDBQueryEngine()) { |
| engine.setConf(conf); |
| |
| // Add data. |
| dao.setConf(conf); |
| dao.init(); |
| dao.add(getStatement("u:a", "u:tt", "u:b")); |
| dao.add(getStatement("u:a", "u:tt", "u:c")); |
| |
| // Run the test. |
| final RyaStatement s = getStatement("u:a", null, null); |
| |
| final MapBindingSet bs1 = new MapBindingSet(); |
| bs1.addBinding("foo", new URIImpl("u:x")); |
| |
| final Map.Entry<RyaStatement, BindingSet> e1 = new RdfCloudTripleStoreUtils.CustomEntry<>(s, bs1); |
| final Collection<Entry<RyaStatement, BindingSet>> stmts1 = Lists.newArrayList(e1); |
| assertEquals(2, size(engine.queryWithBindingSet(stmts1, conf))); |
| |
| |
| final MapBindingSet bs2 = new MapBindingSet(); |
| bs2.addBinding("foo", new URIImpl("u:y")); |
| |
| final Map.Entry<RyaStatement, BindingSet> e2 = new RdfCloudTripleStoreUtils.CustomEntry<>(s, bs2); |
| |
| final Collection<Entry<RyaStatement, BindingSet>> stmts2 = Lists.newArrayList(e1, e2); |
| assertEquals(4, size(engine.queryWithBindingSet(stmts2, conf))); |
| } finally { |
| dao.destroy(); |
| } |
| } |
| } |