/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2008, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags. See the copyright.txt file 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.classloading.vfs.metadata.test;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;

import junit.framework.Test;

import org.jboss.classloader.plugins.ClassLoaderUtils;
import org.jboss.classloading.spi.RealClassLoader;
import org.jboss.classloading.spi.metadata.ClassLoadingMetaDataFactory;
import org.jboss.classloading.spi.metadata.ExportAll;
import org.jboss.classloading.spi.vfs.metadata.VFSClassLoaderFactory;
import org.jboss.kernel.spi.deployment.KernelDeployment;
import org.jboss.test.classloading.vfs.metadata.VFSClassLoadingMicrocontainerTest;
import org.jboss.test.classloading.vfs.metadata.support.a.A;
import org.jboss.test.classloading.vfs.metadata.support.b.B;
import org.jboss.util.id.GUID;
import org.jboss.virtual.MemoryFileFactory;
import org.jboss.virtual.VFS;
import org.jboss.virtual.plugins.context.memory.MemoryContextFactory;

/**
 * DomainUnitTestCase.
 * 
 * @author <a href="adrian@jboss.com">Kabir Khan</a>
 * @version $Revision: 1.1 $
 */
public class GeneratedClassesUnitTestCase extends VFSClassLoadingMicrocontainerTest
{
   final static GeneratedClassInfo NEW_PACKAGE  = new GeneratedClassInfo("newpackage.GeneratedClass");
   final static GeneratedClassInfo OTHER_PACKAGE  = new GeneratedClassInfo("otherpackage.GeneratedClass");
   final static GeneratedClassInfo EXISTING_PACKAGE  = new GeneratedClassInfo("org.jboss.test.classloading.vfs.metadata.support.a.GeneratedClass");
   static
   {
      VFS.init();
   }
   
   public static Test suite()
   {
      return suite(GeneratedClassesUnitTestCase.class);
   }

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

   public void testImportAllGenerateClassInExistingPackage() throws Exception
   {
      runImportAllGenerateClass(EXISTING_PACKAGE, true);
   }
   
   public void testImportAllGenerateClassInGlobalIncludedPackage() throws Exception
   {
      runImportAllGenerateClass(NEW_PACKAGE, true);
   }
   
   public void testImportAllGenerateClassInOtherPackage() throws Exception
   {
      runImportAllGenerateClass(OTHER_PACKAGE, false);
   }
   
   private void runImportAllGenerateClass(GeneratedClassInfo info, boolean expectSuccess) throws Exception
   {
      ClassLoadingMetaDataFactory factory = ClassLoadingMetaDataFactory.getInstance();
      String dynamicClassRoot = getDynamicClassRoot();
      VFSClassLoaderFactory a = new VFSClassLoaderFactory("a");
      a.setImportAll(true);
      a.getRoots().add(getRoot(A.class));
      a.getRoots().add(dynamicClassRoot);
      a.getCapabilities().addCapability(factory.createPackage(A.class.getPackage().getName()));
      KernelDeployment depA = install(a);

      VFSClassLoaderFactory b = new VFSClassLoaderFactory("b");
      b.setExportAll(ExportAll.NON_EMPTY);
      b.setImportAll(true);
      b.getRoots().add(getRoot(B.class));
      b.getCapabilities().addCapability(factory.createPackage(B.class.getPackage().getName()));
      KernelDeployment depB = install(b);
      try
      {
         ClassLoader clA = assertClassLoader(a);
         ClassLoader clB = assertClassLoader(b);
         assertLoadClass(A.class, clA);
         assertLoadClass(B.class, clB);
         assertLoadClass(A.class, clB, clA);
         assertLoadClass(B.class, clA, clB);
         
         Class<?> clazz = generateClass(clA, dynamicClassRoot, info);
         Class<?> clazzA = assertLoadClass(info.getClassname(), clA);
         assertSame(clazz, clazzA);
         
         try
         {
            Class<?> clazzB = assertLoadClass(info.getClassname(), clB, clA);
            
            if (!expectSuccess)
            {
               fail("Should not have been able to load " + info.getClassname());
            }
            assertSame(clazz, clazzB);
         }
         catch(Throwable t)
         {
            if (expectSuccess)
            {
               fail("Should have been able to load class" + info.getClassname() + " " + t);
            }
         }
      }
      finally
      {
         undeploy(depB);
         undeploy(depA);
      }
   }

   private Class<?> generateClass(ClassLoader loader, String dynamicClassRoot, GeneratedClassInfo info) throws Exception
   {
      URL outputURL = new URL(dynamicClassRoot + "/" + info.getResourceName());
      MemoryContextFactory factory = MemoryContextFactory.getInstance();
      factory.putFile(outputURL, info.getClassBytes());
      
      if (loader instanceof RealClassLoader)
      {
         ((RealClassLoader)loader).clearBlackList(info.getResourceName());
      }
      return loader.loadClass(info.getClassname());
   }
   
   private String getDynamicClassRoot() throws Exception
   {
      URL dynamicClassRoot = new URL("vfsmemory", GUID.asString(), "");
      MemoryFileFactory.createRoot(dynamicClassRoot).getRoot();
      return dynamicClassRoot.toString();
   }

   private static class GeneratedClassInfo
   {
      String classname;
      String resourceName;
      byte[] classBytes;
      
      public GeneratedClassInfo(String classname)
      {
         this.classname = classname;
         resourceName = ClassLoaderUtils.classNameToPath(classname);
         loadClassBytes();
      }
      
      public String getClassname()
      {
         return classname;
      }

      public String getResourceName()
      {
         return resourceName;
      }

      public byte[] getClassBytes()
      {
         return classBytes;
      }

      private void loadClassBytes()
      {
         InputStream in = this.getClass().getClassLoader().getResourceAsStream("classes/" + resourceName);
         
         assertNotNull("Could not find inputstream for " + resourceName, in);
         try
         {
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            int i = in.read();
            while(i != -1)
            {
               out.write((byte)i);
               i = in.read();
            }
            classBytes = out.toByteArray();
         }
         catch(Exception e)
         {
            throw new RuntimeException(e);
         }
         finally
         {
            if (in != null)
            {
               try
               {
                  in.close();
               }
               catch (IOException e)
               {
                  // AutoGenerated
                  throw new RuntimeException(e);
               }
            }
         }
      }
   }
}
