package com.atlassian.plugin.osgi.factory;
   
   import junit.framework.TestCase;
   import com.atlassian.plugin.osgi.container.OsgiContainerManager;
   import com.atlassian.plugin.PluginArtifact;
   
   import static org.hamcrest.Matchers.sameInstance;
   import static org.junit.Assert.assertThat;
   import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  import static org.mockito.Mockito.withSettings;
  
  import org.osgi.framework.Bundle;
  import org.osgi.framework.Constants;
  
  import java.io.File;
  import java.util.Hashtable;
  import java.util.Dictionary;
  
  import com.google.common.collect.ImmutableMap;
  
  public class TestOsgiPluginUninstalledHelper extends TestCase
  {
      private String key = "key";
      private OsgiContainerManager mgr;
      private OsgiPluginUninstalledHelper helper;
  
      @Override
       protected void setUp() throws Exception
      {
          super.setUp();
          PluginArtifact pluginArtifact = mock(PluginArtifact.class);
          mgr = mock(OsgiContainerManager.class);
          helper = new OsgiPluginUninstalledHelper(key, mgr, pluginArtifact);
      }
  
      public void testInstall()
      {
          Dictionary dict = new Hashtable();
          dict.put(OsgiPlugin.ATLASSIAN_PLUGIN_KEY, key);
          Bundle bundle = mock(Bundle.class);
          when(bundle.getHeaders()).thenReturn(dict);
          when(bundle.getSymbolicName()).thenReturn(key);
          when(mgr.installBundle(null)).thenReturn(bundle);
          assertEquals(bundle, helper.install());
  
      }
  
      public void testInstallDifferentSymbolicName()
      {
          Dictionary dict = new Hashtable();
          Bundle bundle = mock(Bundle.class);
          when(bundle.getHeaders()).thenReturn(dict);
          when(bundle.getSymbolicName()).thenReturn("bar");
          when(mgr.installBundle(null)).thenReturn(bundle);
          try
          {
              helper.install();
              fail();
          }
          catch (IllegalArgumentException ex)
          {
              //test passed
          }
      }
  
      public void testInstallDifferentSymbolicNameButAltassianKeyFound()
      {
          Dictionary dict = new Hashtable();
          dict.put(OsgiPlugin.ATLASSIAN_PLUGIN_KEY, key);
          Bundle bundle = mock(Bundle.class);
          when(bundle.getHeaders()).thenReturn(dict);
          when(bundle.getSymbolicName()).thenReturn("bar");
          when(mgr.installBundle(null)).thenReturn(bundle);
          try
          {
              helper.install();
              // test passed
          }
          catch (IllegalArgumentException ex)
          {
              fail();
          }
      }
  
      public void testPluginArtifactThatAllowsReferenceIsInstalledByReference()
      {
          final String bundleSymbolicName = getClass().getName() + ".testBundle";
          final String bundleVersion = "1.2";
          final String pluginKey = bundleSymbolicName + "-" + bundleVersion;
  
          final File jarFile = new File("some.jar");
          final PluginArtifact pluginArtifact = mock(
                  PluginArtifact.class,
                  withSettings().extraInterfaces(PluginArtifact.AllowsReference.class));
          when(pluginArtifact.toFile()).thenReturn(jarFile);
          when(((PluginArtifact.AllowsReference) pluginArtifact).allowsReference()).thenReturn(true);
          final Bundle expectedBundle = mock(Bundle.class);
          // Need a basic set of headers to pass the plugin key tests on install
         final Dictionary<String, String> headers = new Hashtable<String, String>(ImmutableMap.<String, String>builder()
                 .put(Constants.BUNDLE_SYMBOLICNAME, bundleSymbolicName)
                 .put(Constants.BUNDLE_VERSION, bundleVersion)
                 .build());
         when(expectedBundle.getSymbolicName()).thenReturn(bundleSymbolicName);
         when(expectedBundle.getHeaders()).thenReturn(headers);
         final OsgiContainerManager osgiContainerManager = mock(
                 OsgiContainerManager.class,
                 withSettings().extraInterfaces(OsgiContainerManager.AllowsReferenceInstall.class));
         // Have our mock OsgiContainerManager respond only to a reference install (the true in the next line)
         when(((OsgiContainerManager.AllowsReferenceInstall) osgiContainerManager).installBundle(jarFile, true))
                 .thenReturn(expectedBundle);
         final OsgiPluginUninstalledHelper osgiPluginUninstalledHelper =
                 new OsgiPluginUninstalledHelper(pluginKey, osgiContainerManager, pluginArtifact);
         final Bundle actualBundle = osgiPluginUninstalledHelper.install();
         assertThat(actualBundle, sameInstance(expectedBundle));
     }
 
 }