| /* |
| * 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. |
| */ |
| /* |
| * IUMJUnitTest.java |
| * |
| * @ TASK IUM 4 & IUM 3 Created on April 29, 2005, 10:14 AM |
| */ |
| package org.apache.geode.cache.query.functional; |
| |
| import static org.junit.Assert.fail; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Iterator; |
| import java.util.Set; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| import org.junit.experimental.categories.Category; |
| |
| import org.apache.geode.cache.Region; |
| import org.apache.geode.cache.query.CacheUtils; |
| import org.apache.geode.cache.query.Index; |
| import org.apache.geode.cache.query.IndexType; |
| import org.apache.geode.cache.query.Query; |
| import org.apache.geode.cache.query.QueryService; |
| import org.apache.geode.cache.query.SelectResults; |
| import org.apache.geode.cache.query.Struct; |
| import org.apache.geode.cache.query.data.Portfolio; |
| import org.apache.geode.cache.query.data.Position; |
| import org.apache.geode.cache.query.internal.QueryObserverAdapter; |
| import org.apache.geode.cache.query.internal.QueryObserverHolder; |
| import org.apache.geode.cache.query.types.StructType; |
| import org.apache.geode.test.junit.categories.OQLIndexTest; |
| |
| @Category({OQLIndexTest.class}) |
| public class IUMJUnitTest { |
| StructType resType1 = null; |
| StructType resType2 = null; |
| StructType resType3 = null; |
| |
| String[] strg1 = null; |
| String[] strg2 = null; |
| String[] strg3 = null; |
| |
| int resSize1 = 0; |
| int resSize2 = 0; |
| int resSize3 = 0; |
| |
| Object valPf1 = null; |
| Object valPos1 = null; |
| |
| Object valPf2 = null; |
| Object valPos2 = null; |
| |
| Object valPf3 = null; |
| Object valPos3 = null; |
| |
| Iterator itert1 = null; |
| Iterator itert2 = null; |
| Iterator itert3 = null; |
| |
| Set set1 = null; |
| Set set2 = null; |
| Set set3 = null; |
| |
| boolean isActive1 = false; |
| boolean isActive2 = false; |
| boolean isActive3 = true; |
| |
| @Before |
| public void setUp() throws java.lang.Exception { |
| CacheUtils.startCache(); |
| } |
| |
| @After |
| public void tearDown() throws java.lang.Exception { |
| CacheUtils.closeCache(); |
| } |
| |
| @Test |
| public void testComparisonBetnWithAndWithoutIndexCreation() throws Exception { |
| |
| Region region = CacheUtils.createRegion("pos", Portfolio.class); |
| |
| for (int i = 0; i < 4; i++) { |
| region.put("" + i, new Portfolio(i)); |
| } |
| QueryService qs; |
| qs = CacheUtils.getQueryService(); |
| String queries[] = {"SELECT DISTINCT * FROM /pos, positions.values where status='active'" |
| // TASK IUM4 |
| }; |
| SelectResults r[][] = new SelectResults[queries.length][2]; |
| for (int i = 0; i < queries.length; i++) { |
| Query q = null; |
| try { |
| q = CacheUtils.getQueryService().newQuery(queries[i]); |
| QueryObserverImpl observer = new QueryObserverImpl(); |
| QueryObserverHolder.setInstance(observer); |
| r[i][0] = (SelectResults) q.execute(); |
| |
| if (!observer.isIndexesUsed) { |
| CacheUtils.log("NO INDEX USED"); |
| } |
| } catch (Exception e) { |
| e.printStackTrace(); |
| fail(q.getQueryString()); |
| } |
| } |
| |
| // Create an Index on status and execute the same query again. |
| |
| qs = CacheUtils.getQueryService(); |
| qs.createIndex("statusIndex", IndexType.FUNCTIONAL, "status", "/pos"); |
| |
| for (int i = 0; i < queries.length; i++) { |
| Query q = null; |
| try { |
| q = CacheUtils.getQueryService().newQuery(queries[i]); |
| QueryObserverImpl observer2 = new QueryObserverImpl(); |
| QueryObserverHolder.setInstance(observer2); |
| r[i][1] = (SelectResults) q.execute(); |
| |
| if (observer2.isIndexesUsed) { |
| CacheUtils.log("YES INDEX IS USED!"); |
| } else { |
| fail("Index NOT Used"); |
| } |
| |
| } catch (Exception e) { |
| e.printStackTrace(); |
| fail(q.getQueryString()); |
| } |
| } |
| CacheUtils.compareResultsOfWithAndWithoutIndex(r, this); |
| // BUG : Types are not Equal in both the cases as when Indexes are used the Iterator Names |
| // are getting Overwritten as iter1,iter2 and so on instead of the complied values of the |
| // iterator names used in the Query. |
| |
| // if ((resType1).equals(resType2)){ |
| // CacheUtils.log("Both Search Results are of the same Type i.e.--> "+resType1); |
| // }else { |
| // fail("FAILED:Search result Type is different in both the cases"); |
| // } |
| // if (resSize1==resSize2 || resSize1 != 0 ){ |
| // CacheUtils.log("Both Search Results are non-zero and of Same Size i.e. Size= "+resSize1); |
| // }else { |
| // fail("FAILED:Search result Type is different in both the cases"); |
| // } |
| // |
| } |
| |
| @Test |
| public void testWithOutIndexCreatedMultiCondQueryTest() throws Exception { |
| Region region = CacheUtils.createRegion("portfolios", Portfolio.class); |
| for (int i = 0; i < 4; i++) { |
| region.put("" + i, new Portfolio(i)); |
| // CacheUtils.log(new Portfolio(i)); |
| } |
| CacheUtils.getQueryService(); |
| |
| String queries[] = { |
| "SELECT DISTINCT * from /portfolios pf , pf.positions.values pos where pos.getSecId = 'IBM' and status = 'inactive'" |
| // TASK IUM3 |
| }; |
| for (int i = 0; i < queries.length; i++) { |
| Query q = null; |
| try { |
| q = CacheUtils.getQueryService().newQuery(queries[i]); |
| Object r3 = q.execute(); |
| resType3 = (StructType) ((SelectResults) r3).getCollectionType().getElementType(); |
| resSize3 = (((SelectResults) r3).size()); |
| // CacheUtils.log(resType3); |
| strg3 = resType3.getFieldNames(); |
| // CacheUtils.log(strg3[0]); |
| // CacheUtils.log(strg2[1]); |
| |
| set3 = (((SelectResults) r3).asSet()); |
| Iterator iter = set3.iterator(); |
| while (iter.hasNext()) { |
| Struct stc3 = (Struct) iter.next(); |
| valPf2 = stc3.get(strg3[0]); |
| valPos2 = stc3.get(strg3[1]); |
| isActive3 = ((Portfolio) stc3.get(strg3[0])).isActive(); |
| // CacheUtils.log(valPf2); |
| // CacheUtils.log(valPos2); |
| } |
| |
| } catch (Exception e) { |
| e.printStackTrace(); |
| fail(q.getQueryString()); |
| } |
| } |
| |
| itert3 = set3.iterator(); |
| while (itert3.hasNext()) { |
| Struct stc3 = (Struct) itert3.next(); |
| if (!((Position) stc3.get(strg3[1])).secId.equals("IBM")) |
| fail("FAILED: secId found is not IBM"); |
| if (((Portfolio) stc3.get(strg3[0])).isActive() != false) |
| fail("FAILED:Portfolio in Search result is Active"); |
| } |
| |
| } |
| |
| class QueryObserverImpl extends QueryObserverAdapter { |
| boolean isIndexesUsed = false; |
| ArrayList indexesUsed = new ArrayList(); |
| |
| @Override |
| public void beforeIndexLookup(Index index, int oper, Object key) { |
| indexesUsed.add(index.getName()); |
| } |
| |
| @Override |
| public void afterIndexLookup(Collection results) { |
| if (results != null) { |
| isIndexesUsed = true; |
| } |
| } |
| } |
| } |