Java tutorial
/* * JBoss, Home of Professional Open Source * Copyright 2009, Red Hat Middleware LLC, and individual contributors * by the @authors tag. See the copyright.txt in the distribution for a * full listing of individual contributors. * * This is free software; you can redistribute it and/or modify it * under the terms of the GNU Lesser General Public License as * published by the Free Software Foundation; either version 2.1 of * the License, or (at your option) any later version. * * This software is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this software; if not, write to the Free * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA * 02110-1301 USA, or see the FSF site: http://www.fsf.org. */ package org.infinispan.query.blackbox; import org.apache.lucene.index.Term; import org.apache.lucene.queryParser.ParseException; import org.apache.lucene.queryParser.QueryParser; import org.apache.lucene.search.Filter; import org.apache.lucene.search.PrefixFilter; import org.apache.lucene.search.Query; import org.apache.lucene.search.Sort; import org.apache.lucene.search.SortField; import org.apache.lucene.search.TermQuery; import org.infinispan.Cache; import org.infinispan.query.CacheQuery; import org.infinispan.query.QueryFactory; import org.infinispan.query.QueryIterator; import org.infinispan.query.backend.QueryHelper; import org.infinispan.query.test.AnotherGrassEater; import org.infinispan.query.test.Person; import org.infinispan.test.SingleCacheManagerTest; import org.testng.annotations.AfterMethod; import java.util.List; import static java.util.Arrays.asList; import static org.infinispan.query.helper.TestQueryHelperFactory.*; public abstract class AbstractLocalQueryTest extends SingleCacheManagerTest { protected Person person1; protected Person person2; protected Person person3; protected Person person4; protected Person person5; protected Person person6; protected AnotherGrassEater anotherGrassEater; protected QueryParser queryParser; protected Query luceneQuery; protected CacheQuery cacheQuery; protected List<Object> found; protected String key1 = "Navin"; protected String key2 = "BigGoat"; protected String key3 = "MiniGoat"; protected String anotherGrassEaterKey = "anotherGrassEaterKey"; protected Cache<Object, Object> cache; protected QueryHelper qh; @AfterMethod(alwaysRun = true) public void tearDown() { if (cache != null) cache.stop(); } public void testSimple() throws ParseException { cacheQuery = createCacheQuery(cache, qh, "blurb", "playing"); found = cacheQuery.list(); int elems = found.size(); assert elems == 1 : "Expected 1 but was " + elems; Object val = found.get(0); assert val.equals(person1) : "Expected " + person1 + " but was " + val; } public void testEagerIterator() throws ParseException { cacheQuery = createCacheQuery(cache, qh, "blurb", "playing"); QueryIterator found = cacheQuery.iterator(); assert found.isFirst(); assert found.isLast(); } public void testMultipleResults() throws ParseException { queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2; assert found.get(0) == person2; assert found.get(1) == person3; } public void testModified() throws ParseException { queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("playing"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 1; assert found.get(0).equals(person1); person1.setBlurb("Likes pizza"); cache.put(key1, person1); queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("pizza"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); System.out.println("Found size is: - " + found.size()); assert found.size() == 1; assert found.get(0).equals(person1); } public void testAdded() throws ParseException { queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2 : "Size of list should be 2"; assert found.contains(person2); assert found.contains(person3); assert !found.contains(person4) : "This should not contain object person4"; person4 = new Person(); person4.setName("Mighty Goat"); person4.setBlurb("Also eats grass"); cache.put("mighty", person4); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 3 : "Size of list should be 3"; assert found.contains(person2); assert found.contains(person3); assert found.contains(person4) : "This should now contain object person4"; } public void testRemoved() throws ParseException { queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2; assert found.contains(person2); assert found.contains(person3) : "This should still contain object person3"; cache.remove(key3); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 1; assert found.contains(person2); assert !found.contains(person3) : "The search should not return person3"; } public void testUpdated() throws ParseException { queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2 : "Size of list should be 2"; assert found.contains(person2) : "The search should have person2"; cache.put(key2, person1); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 1 : "Size of list should be 1"; assert !found.contains(person2) : "Person 2 should not be found now"; assert !found .contains(person1) : "Person 1 should not be found because it does not meet the search criteria"; } public void testSetSort() throws ParseException { person2.setAge(35); person3.setAge(12); Sort sort = new Sort(new SortField("age", SortField.STRING)); queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("Goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2; cacheQuery.setSort(sort); found = cacheQuery.list(); assert found.size() == 2; assert found.get(0).equals(person2); assert found.get(1).equals(person3); } public void testSetFilter() throws ParseException { queryParser = createQueryParser("name"); luceneQuery = queryParser.parse("goat"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2; Filter filter = new PrefixFilter(new Term("blurb", "cheese")); cacheQuery.setFilter(filter); found = cacheQuery.list(); assert found.size() == 1; } public void testLazyIterator() throws ParseException { queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("playing"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); QueryIterator found = cacheQuery.lazyIterator(); assert found.isFirst(); assert found.isLast(); } public void testGetResultSize() throws ParseException { queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("playing"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); assert cacheQuery.getResultSize() == 1; } public void testClear() throws ParseException { // Create a term that will return me everyone called Navin. Term navin = new Term("name", "navin"); // Create a term that I know will return me everything with name goat. Term goat = new Term("name", "goat"); Query[] queries = new Query[2]; queries[0] = new TermQuery(goat); queries[1] = new TermQuery(navin); luceneQuery = queries[0].combine(queries); // luceneQuery = new TermQuery(goat).combine(new Query[]{new TermQuery(navin)}); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); // We know that we've got all 3 hits. System.out.println("****** Res " + cacheQuery.list()); assert cacheQuery.getResultSize() == 3 : "Expected 3, got " + cacheQuery.getResultSize(); cache.clear(); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); assert cacheQuery.getResultSize() == 0; } public void testTypeFiltering() throws ParseException { queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("grass"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery); found = cacheQuery.list(); assert found.size() == 2; assert found.containsAll(asList(person2, anotherGrassEater)); queryParser = createQueryParser("blurb"); luceneQuery = queryParser.parse("grass"); cacheQuery = new QueryFactory(cache, qh).getQuery(luceneQuery, AnotherGrassEater.class); found = cacheQuery.list(); assert found.size() == 1; assert found.get(0).equals(anotherGrassEater); } }