package com.atlassian.plugin.webresource;
   
   import java.util.Collections;
   
   import com.atlassian.plugin.ModuleDescriptor;
   import com.atlassian.plugin.Plugin;
   import com.atlassian.plugin.PluginAccessor;
   import com.atlassian.plugin.elements.ResourceLocation;
   import com.atlassian.plugin.servlet.DownloadableClasspathResource;
  import com.atlassian.plugin.servlet.DownloadableResource;
  import com.atlassian.plugin.servlet.ServletContextFactory;
  
  import com.google.common.collect.ImmutableMap;
  
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.mockito.Mock;
  import org.mockito.runners.MockitoJUnitRunner;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertNotNull;
  import static org.mockito.Mockito.when;
  
  @RunWith(MockitoJUnitRunner.class)
  public class TestSingleDownloadableResourceBuilder
  {
      public static final String MODULE_KEY = "test.plugin.key:module";
      public static final String PLUGIN_KEY = "test.plugin.key";
      @Mock
      private PluginAccessor mockPluginAccessor;
      @Mock
      private ServletContextFactory mockServletContextFactory;
      @Mock
      ModuleDescriptor mockModuleDescriptor;
  
      @Mock
      WebResourceIntegration mockWebResourceIntegration;
  
      SingleDownloadableResourceBuilder builder;
      Plugin plugin;
  
      @Before
      public void setUp() throws Exception
      {
          builder = new SingleDownloadableResourceBuilder(mockPluginAccessor, mockServletContextFactory, mockWebResourceIntegration);
  
          plugin = TestUtils.createTestPlugin(PLUGIN_KEY, "1.0");
  
          when(mockPluginAccessor.getEnabledPluginModule(MODULE_KEY)).thenReturn(mockModuleDescriptor);
          when(mockPluginAccessor.getPlugin(PLUGIN_KEY)).thenReturn(plugin);
      }
  
      @After
      public void tearDown() throws Exception
      {
          mockPluginAccessor = null;
          mockServletContextFactory = null;
      }
  
      @Test
      public void testParseWithSimpleName() throws Exception
      {
          final ResourceLocation location = new ResourceLocation("", "mydownload.jpg", "download", "image/jpeg", "", Collections.<String, String>emptyMap());
          when(mockModuleDescriptor.getResourceLocation("download", "mydownload.jpg")).thenReturn(location);
  
          DownloadableResource resource = builder.parse("/download/resources/test.plugin.key:module/mydownload.jpg", Collections.<String, String>emptyMap());
  
          assertNotNull(resource);
          assertEquals(location.getContentType(), resource.getContentType());
      }
  
      @Test
      public void testParseWithSlashesInName() throws Exception
      {
          final ResourceLocation location = new ResourceLocation("", "mydownload.swf", "download", "application/x-shockwave-flash", "", Collections.<String, String>emptyMap());
          when(mockModuleDescriptor.getResourceLocation("download", "path/to/mydownload.swf")).thenReturn(location);
  
          DownloadableResource resource = builder.parse("/download/resources/test.plugin.key:module/path/to/mydownload.swf", Collections.<String, String>emptyMap());
  
          assertNotNull(resource);
          assertEquals(location.getContentType(), resource.getContentType());
      }
  
      @Test
      public void testRoundTrip() throws Exception
      {
          SinglePluginResource resource = new SinglePluginResource("foo.css", MODULE_KEY, false);
          String url = resource.getUrl();
  
          final ResourceLocation location = new ResourceLocation("", "foo.css", "download", "text/css", "", Collections.<String, String>emptyMap());
          when(mockModuleDescriptor.getResourceLocation("download", "foo.css")).thenReturn(location);
  
          DownloadableResource parsedResource = builder.parse(url, Collections.<String, String>emptyMap());
  
          assertNotNull(parsedResource);
          assertEquals(location.getContentType(), parsedResource.getContentType());
      }
 
     @Test(expected = UrlParseException.class)
     public void testParseInvalidUrlThrowsException() throws UrlParseException
     {
         builder.parse("/download/resources/blah.png", Collections.<String, String>emptyMap());
     }
 
     @Test
     public void testVariableSubstitutionWithEmptyReplacement()
     {
         final ResourceLocation location = new ResourceLocation("${x}foo.css", "foo.css", "download", "text/css", "", Collections.<String, String>emptyMap());
         when(mockModuleDescriptor.getResourceLocation("download", "foo.css")).thenReturn(location);
 
         // subst: x == ""
         when(mockWebResourceIntegration.getResourceSubstitutionVariables()).thenReturn(ImmutableMap.of("x", ""));
         DownloadableClasspathResource resource = (DownloadableClasspathResource) builder.find(MODULE_KEY, "foo.css");
         assertNotNull(resource);
         assertEquals("foo.css", resource.getResourceLocation().getLocation());
     }
 
     @Test
     public void testVariableSubsitutionWithPrefixReplacement()
     {
         final ResourceLocation location = new ResourceLocation("${x}foo.css", "foo.css", "download", "text/css", "", Collections.<String, String>emptyMap());
         when(mockModuleDescriptor.getResourceLocation("download", "foo.css")).thenReturn(location);
 
         // subst: x == "old/"
         when(mockWebResourceIntegration.getResourceSubstitutionVariables()).thenReturn(ImmutableMap.of("x", "old/"));
         DownloadableClasspathResource resource = (DownloadableClasspathResource) builder.find(MODULE_KEY, "foo.css");
         assertNotNull(resource);
         assertEquals("old/foo.css", resource.getResourceLocation().getLocation());
     }
 
     @Test
     public void variableSubsitutionWithMissingVariablesLeavesTemplateInPlace()
     {
         final ResourceLocation location = new ResourceLocation("${x}foo.css", "foo.css", "download", "text/css", "", Collections.<String, String>emptyMap());
         when(mockModuleDescriptor.getResourceLocation("download", "foo.css")).thenReturn(location);
 
         when(mockWebResourceIntegration.getResourceSubstitutionVariables()).thenReturn(ImmutableMap.<String, String>of());
         DownloadableClasspathResource resource = (DownloadableClasspathResource) builder.find(MODULE_KEY, "foo.css");
         assertNotNull(resource);
         assertEquals("${x}foo.css", resource.getResourceLocation().getLocation());
     }
 }