/*
 * JBoss, Home of Professional Open Source
 * Copyright 2005, JBoss Inc., and individual contributors as indicated
 * 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.jboss.test.cmp2.simple;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Properties;
import javax.management.MBeanServer;
import javax.management.MBeanServerFactory;
import javax.management.ObjectName;
import javax.naming.InitialContext;

import junit.framework.Test;
import org.jboss.ejb.EntityContainer;
import org.jboss.ejb.plugins.cmp.ejbql.Catalog;
import org.jboss.ejb.plugins.cmp.jdbc.JDBCStoreManager;
import org.jboss.ejb.plugins.cmp.jdbc.ReadAheadCache;
import org.jboss.ejb.plugins.cmp.jdbc.bridge.JDBCEntityBridge;
import org.jboss.mx.server.ServerConstants;
import org.jboss.mx.server.registry.MBeanEntry;
import org.jboss.mx.server.registry.MBeanRegistry;
import org.jboss.mx.util.MBeanProxyExt;
import org.jboss.test.JBossTestCase;
import org.jboss.test.util.ejb.EJBTestCase;

public class PageSizeUnitTestCase extends EJBTestCase
{
   private JDBCStoreManager jdbcStoreManager;
   private List pkList;

   public static Test suite() throws Exception
   {
      return JBossTestCase.getDeploySetup(
            PageSizeUnitTestCase.class, "cmp2-simple.jar");
   }

   public PageSizeUnitTestCase(String name)
   {
      super(name);
   }

   private SimpleHome getSimpleHome()
   {
      try
      {
         InitialContext jndiContext = new InitialContext();
         return (SimpleHome) jndiContext.lookup("cmp2/simple/Simple");
      }
      catch (Exception e)
      {
         fail("Exception in getSimpleHome: " + e.getMessage());
      }
      return null;
   }

   public void testOnLoad() throws Exception
   {
      SimpleHome simpleHome = getSimpleHome();
      Iterator simpleIter = simpleHome.findAll().iterator();
      Simple simple = (Simple) simpleIter.next();
      Object pk = simple.getPrimaryKey();
      ReadAheadCache cache = jdbcStoreManager.getReadAheadCache();
      ReadAheadCache.EntityReadAheadInfo info = cache.getEntityReadAheadInfo(pk);
      assertEquals(pkList.subList(0, 4), info.getLoadKeys());

      for (int i = 0; i < 4; i++)
      {
         Object o = pkList.get(i);
         assertNull(cache.getPreloadDataMap(o, false));
      }

      simple.getStringValue(); // test0

      assertNull(cache.getPreloadDataMap("test0", false));
      for (int i = 1; i < 4; i++)
      {
         Object o = pkList.get(i);
         assertNotNull(cache.getPreloadDataMap(o, false));
      }
      assertNull(cache.getPreloadDataMap("test4", false));

      simple = (Simple) simpleIter.next(); // test1
      simple.getStringValue();
      assertNull(cache.getPreloadDataMap("test1", false));
      simple = (Simple) simpleIter.next(); // test2
      simple.getStringValue();
      simple = (Simple) simpleIter.next(); // test3
      simple.getStringValue();
      for (int i = 0; i < 4; i++)
      {
         Object o = pkList.get(i);
         assertNull(cache.getPreloadDataMap(o, false));
      }

      simple = (Simple) simpleIter.next(); // test4
      simple.getStringValue();
      for (int i = 5; i < 8; i++)
      {
         Object o = pkList.get(i);
         assertNotNull(cache.getPreloadDataMap(o, false));
      }
   }

   public void setUpEJB(Properties props) throws Exception
   {
      super.setUpEJB(props);
      SimpleHome simpleHome = getSimpleHome();
      pkList = new ArrayList();

      for (int i = 0; i < 10; i++)
      {
         Simple simple = simpleHome.create("test" + i);
         pkList.add(simple.getPrimaryKey());
         simple.setIntegerPrimitive(i);
      }

      MBeanServer server = (MBeanServer) MBeanServerFactory.findMBeanServer(null).get(0);
      ObjectName name = new ObjectName("jboss.j2ee:jndiName=cmp2/simple/Simple,service=EJB");
      MBeanRegistry registry = (MBeanRegistry) MBeanProxyExt.create(MBeanRegistry.class, ServerConstants.MBEAN_REGISTRY, server);
      MBeanEntry entry = registry.get(name);
      EntityContainer container = (EntityContainer) entry.getResourceInstance();
      Catalog catalog = (Catalog) container.getEjbModule().getModuleData("CATALOG");
      JDBCEntityBridge bridge = (JDBCEntityBridge) catalog.getEntityByEJBName("SimpleEJB");
      jdbcStoreManager = (JDBCStoreManager)bridge.getManager();
   }

   public void tearDownEJB(Properties props) throws Exception
   {
      SimpleHome simpleHome = getSimpleHome();
      Collection c = simpleHome.findAll();
      for (Iterator iterator = c.iterator(); iterator.hasNext();)
      {
         Simple simple = (Simple) iterator.next();
         simple.remove();
      }
      super.tearDownEJB(props);
   }
}

