package com.atlassian.plugin.osgi.bridge;
   
   import com.atlassian.plugin.Plugin;
   import com.atlassian.plugin.PluginAccessor;
   import com.atlassian.plugin.osgi.bridge.external.PluginRetrievalService;
   import junit.framework.TestCase;
   import org.osgi.framework.Bundle;
   
   import java.util.Hashtable;
  
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  
  /**
   *
   */
  public class TestPluginRetrievalServiceFactory extends TestCase
  {
      public void testGetPlugin()
      {
          PluginAccessor pluginAccessor = mock(PluginAccessor.class);
          Bundle bundle = mock(Bundle.class);
          Hashtable headers = new Hashtable();
          headers.put("Atlassian-Plugin-Key", "foo");
          when(bundle.getHeaders()).thenReturn(headers);
          Plugin plugin = mock(Plugin.class);
          when(pluginAccessor.getPlugin("foo")).thenReturn(plugin);
  
          PluginRetrievalServiceFactory factory = new PluginRetrievalServiceFactory(pluginAccessor);
          assertEquals(plugin, ((PluginRetrievalService)factory.getService(bundle, null)).getPlugin());
      }
  
      public void testGetPluginButFrameworkBundle()
      {
          PluginAccessor pluginAccessor = mock(PluginAccessor.class);
          Bundle bundle = mock(Bundle.class);
          Hashtable headers = new Hashtable();
          headers.put("Atlassian-Plugin-Key", "foo");
          when(bundle.getHeaders()).thenReturn(headers);
  
          PluginRetrievalServiceFactory factory = new PluginRetrievalServiceFactory(pluginAccessor);
          assertNull(((PluginRetrievalService)factory.getService(bundle, null)).getPlugin());
      }
  }