[jbosscache-commits] JBoss Cache SVN: r6053 - searchable/trunk/src/test/java/org/jboss/cache/search.

[jbosscache-commits at lists.jboss.org]

Author: manik.surtani at jboss.com
Date: 2008-06-26 07:31:23 -0400 (Thu, 26 Jun 2008)
New Revision: 6053

Modified:
   searchable/trunk/src/test/java/org/jboss/cache/search/QueryResultIteratorImplTest.java
Log:
Added a mock loader

Modified: searchable/trunk/src/test/java/org/jboss/cache/search/QueryResultIteratorImplTest.java
===================================================================
--- searchable/trunk/src/test/java/org/jboss/cache/search/QueryResultIteratorImplTest.java	2008-06-26 11:18:41 UTC (rev 6052)
+++ searchable/trunk/src/test/java/org/jboss/cache/search/QueryResultIteratorImplTest.java	2008-06-26 11:31:23 UTC (rev 6053)
@@ -6,7 +6,9 @@
 import org.testng.annotations.Test;
 
 import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * @author Navin Surtani  - navin at surtani.org
@@ -17,32 +19,48 @@
 @Test
 public class QueryResultIteratorImplTest
 {
-   CacheEntityLoader entityLoader;
-   List<CacheEntityId> list;
+   List<CacheEntityId> ids;
+   Map<CacheEntityId, Object> dummyResults;
+   QueryResultIterator iterator;
 
    @BeforeMethod
    public void SetUp()
    {
-      list = new ArrayList();
-      list.add(new CacheEntityId(Fqn.fromString("/a"), "key1"));
-      list.add(new CacheEntityId(Fqn.fromString("/b"), "key2"));
-      list.add(new CacheEntityId(Fqn.fromString("/c"), "key3"));
-      list.add(new CacheEntityId(Fqn.fromString("/d"), "key4"));
-      list.add(new CacheEntityId(Fqn.fromString("/e"), "key5"));
-      list.add(new CacheEntityId(Fqn.fromString("/f"), "key6"));
-      list.add(new CacheEntityId(Fqn.fromString("/g"), "key7"));
-      list.add(new CacheEntityId(Fqn.fromString("/h"), "key8"));
-      list.add(new CacheEntityId(Fqn.fromString("/i"), "key9"));
-      list.add(new CacheEntityId(Fqn.fromString("/j"), "key10"));
+      // create a set of dummy cache entity IDs
+      ids = new ArrayList();
+      ids.add(new CacheEntityId(Fqn.fromString("/a"), "key1"));
+      ids.add(new CacheEntityId(Fqn.fromString("/b"), "key2"));
+      ids.add(new CacheEntityId(Fqn.fromString("/c"), "key3"));
+      ids.add(new CacheEntityId(Fqn.fromString("/d"), "key4"));
+      ids.add(new CacheEntityId(Fqn.fromString("/e"), "key5"));
+      ids.add(new CacheEntityId(Fqn.fromString("/f"), "key6"));
+      ids.add(new CacheEntityId(Fqn.fromString("/g"), "key7"));
+      ids.add(new CacheEntityId(Fqn.fromString("/h"), "key8"));
+      ids.add(new CacheEntityId(Fqn.fromString("/i"), "key9"));
+      ids.add(new CacheEntityId(Fqn.fromString("/j"), "key10"));
 
-      QueryResultIteratorImpl iterator = new QueryResultIteratorImpl(list, entityLoader);
+      // create some dummy data
+      dummyResults = new HashMap<CacheEntityId, Object>();
+      int counter = 0;
 
+      for (CacheEntityId id : ids)
+      {
+         // for each cache entity ID, create a dummy result that will be returned when loading it.
+         dummyResults.put(id, "Result number " + counter);
+      }
+
+      // now create a dummy entity loader
+      CacheEntityLoader dummyLoader = new DummyEntityLoader(ids, dummyResults);
+
+      iterator = new QueryResultIteratorImpl(ids, dummyLoader);
    }
 
    @AfterMethod
    public void tearDown()
    {
-      list = null;
+      ids = null;
+      dummyResults = null;
+      iterator = null;
    }
 
    public void jumpToResultTest(int index) throws IndexOutOfBoundsException
@@ -134,4 +152,38 @@
    {
       //To change body of implemented methods use File | Settings | File Templates.
    }
+
+   public static class DummyEntityLoader extends CacheEntityLoader
+   {
+      private List<CacheEntityId> allKnownIds;
+      private Map<CacheEntityId, Object> dummyValues;
+
+      public DummyEntityLoader(List<CacheEntityId> allKnownIds, Map<CacheEntityId, Object> dummyValues)
+      {
+         // use a null as a cache since we won't ever need to refer to the cache
+         super(null);
+
+         this.allKnownIds = allKnownIds;
+         this.dummyValues = dummyValues;
+      }
+
+      @Override
+      public List<Object> load(List<CacheEntityId> ids)
+      {
+         List<Object> resultsToReturn = new ArrayList<Object>(ids.size());
+         // iterate through the list of ids we are looking for
+         for (CacheEntityId id : ids)
+         {
+            resultsToReturn.add(dummyValues.get(id));
+         }
+
+         return resultsToReturn;
+      }
+
+      @Override
+      public Object load(CacheEntityId id)
+      {
+         return dummyValues.get(id);
+      }
+   }
 }