package com.atlassian.plugin;
   
   import junit.framework.TestCase;
   
   import java.io.File;
   import java.io.IOException;
   
   import org.apache.commons.io.FileUtils;
   import com.atlassian.plugin.test.PluginJarBuilder;
  
  import static com.atlassian.plugin.PluginArtifact.AllowsReference.ReferenceMode.FORBID_REFERENCE;
  import static com.atlassian.plugin.PluginArtifact.AllowsReference.ReferenceMode.PERMIT_REFERENCE;
  
  public class TestDefaultPluginArtifactFactory extends TestCase
  {
      File testDir;
      File spaceTestDir;
  
      public void setUp() throws IOException
      {
          testDir = makeTempDir("test");
          spaceTestDir = makeTempDir("space test");
      }
  
      private File makeTempDir(String prefix)
              throws IOException
      {
          File tempDir = File.createTempFile(prefix, "");
          tempDir.delete();
          tempDir.mkdir();
  
          return tempDir;
      }
  
      public void tearDown() throws IOException
      {
          FileUtils.deleteDirectory(testDir);
          FileUtils.deleteDirectory(spaceTestDir);
  
          testDir = null;
          spaceTestDir = null;
      }
  
      public void testCreate() throws IOException
      {
          doCreationTestInDirectory(testDir);
      }
  
      public void testCreateWithSpaceInArtifactPath() throws IOException
      {
          doCreationTestInDirectory(spaceTestDir);
      }
  
      private void doCreationTestInDirectory(File directory) throws IOException
      {
          File xmlFile = new File(directory, "foo.xml");
          FileUtils.writeStringToFile(xmlFile, "<xml/>");
          File jarFile = new PluginJarBuilder("jar").build(directory);
  
          DefaultPluginArtifactFactory factory = new DefaultPluginArtifactFactory();
  
          PluginArtifact jarArt = factory.create(jarFile.toURI());
          assertNotNull(jarArt);
          assertTrue(jarArt instanceof JarPluginArtifact);
  
          PluginArtifact xmlArt = factory.create(xmlFile.toURI());
          assertNotNull(xmlArt);
          assertTrue(xmlArt instanceof XmlPluginArtifact);
  
          try
          {
              factory.create(new File(testDir, "bob.jim").toURI());
              fail("Should have thrown exception");
          } catch (IllegalArgumentException ex)
          {
              // test passed
          }
      }
  
      public void testReferenceModeForbidReferenceDoesNotAllowReference()
      {
          final DefaultPluginArtifactFactory defaultPluginArtifactFactory = new DefaultPluginArtifactFactory(FORBID_REFERENCE);
          verifyAllowsReference(defaultPluginArtifactFactory, false);
      }
  
      public void testReferenceModePermitReferenceAllowsReference()
      {
          final DefaultPluginArtifactFactory defaultPluginArtifactFactory = new DefaultPluginArtifactFactory(PERMIT_REFERENCE);
          verifyAllowsReference(defaultPluginArtifactFactory, true);
      }
  
      public void testDefaultDoesNotAllowsReference()
      {
          final DefaultPluginArtifactFactory defaultPluginArtifactFactory = new DefaultPluginArtifactFactory();
          verifyAllowsReference(defaultPluginArtifactFactory, false);
      }
  
      private void verifyAllowsReference(DefaultPluginArtifactFactory defaultPluginArtifactFactory, boolean allowsReference)
      {
         final File file = new File(testDir, "some.jar");
         final PluginArtifact pluginArtifact = defaultPluginArtifactFactory.create(file.toURI());
         // The interface is always provided now
         assertTrue(pluginArtifact instanceof PluginArtifact.AllowsReference);
         // And the query should report what we specified
         assertEquals(((PluginArtifact.AllowsReference) pluginArtifact).allowsReference(), allowsReference);
         // And the convenience interface should likewise report this
         assertEquals(PluginArtifact.AllowsReference.Default.allowsReference(pluginArtifact), allowsReference);
     }
 }