package com.atlassian.plugin.osgi.factory.transform.stage;
   
   import com.atlassian.plugin.JarPluginArtifact;
   import com.atlassian.plugin.PluginAccessor;
   import com.atlassian.plugin.PluginParseException;
   import com.atlassian.plugin.osgi.container.OsgiContainerManager;
   import com.atlassian.plugin.osgi.external.SingleModuleDescriptorFactory;
   import com.atlassian.plugin.osgi.factory.transform.TransformContext;
   import com.atlassian.plugin.osgi.factory.transform.model.SystemExports;
  import com.atlassian.plugin.osgi.hostcomponents.HostComponentRegistration;
  import com.atlassian.plugin.test.PluginJarBuilder;
  import org.dom4j.DocumentException;
  import org.dom4j.DocumentHelper;
  import org.dom4j.Element;
  import org.junit.Test;
  import org.osgi.framework.ServiceReference;
  
  import java.io.File;
  import java.io.IOException;
  import java.util.Collections;
  
  import static org.junit.Assert.assertTrue;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  
  public class TestModuleTypeSpringStage {
  
      @Test
      public void testTransform() throws IOException, DocumentException {
          Element pluginRoot = DocumentHelper.createDocument().addElement("atlassian-plugin");
          Element moduleType = pluginRoot.addElement("module-type");
          moduleType.addAttribute("key", "foo");
          moduleType.addAttribute("class", "my.FooDescriptor");
  
          SpringTransformerTestHelper.transform(new ModuleTypeSpringStage(), pluginRoot,
                  "beans:bean[@id='springHostContainer' and @class='" + ModuleTypeSpringStage.SPRING_HOST_CONTAINER + "']",
                  "beans:bean[@id='moduleType-foo' and @class='" + SingleModuleDescriptorFactory.class.getName() + "']",
                  "osgi:service[@id='moduleType-foo_osgiService' and @auto-export='interfaces']");
      }
  
      @Test
      public void testTransformForOneApp() throws IOException, DocumentException {
          Element pluginRoot = DocumentHelper.createDocument().addElement("atlassian-plugin");
          Element moduleType = pluginRoot.addElement("module-type");
          moduleType.addAttribute("key", "foo");
          moduleType.addAttribute("class", "my.FooDescriptor");
          moduleType.addAttribute("application", "bar");
          SpringTransformerTestHelper.transform(new ModuleTypeSpringStage(), pluginRoot,
                  "not(beans:bean[@id='moduleType-foo' and @class='" + SingleModuleDescriptorFactory.class.getName() + "'])");
  
          pluginRoot = DocumentHelper.createDocument().addElement("atlassian-plugin");
          moduleType = pluginRoot.addElement("module-type");
          moduleType.addAttribute("key", "foo");
          moduleType.addAttribute("class", "my.FooDescriptor");
          moduleType.addAttribute("application", "foo");
          SpringTransformerTestHelper.transform(new ModuleTypeSpringStage(), pluginRoot,
                  "beans:bean[@id='moduleType-foo' and @class='" + SingleModuleDescriptorFactory.class.getName() + "']");
      }
  
      @Test(expected = PluginParseException.class)
      public void testTransformOfBadElement() throws IOException, DocumentException {
          Element pluginRoot = DocumentHelper.createDocument().addElement("atlassian-plugin");
          Element moduleType = pluginRoot.addElement("module-type");
          moduleType.addAttribute("key", "foo");
  
          SpringTransformerTestHelper.transform(new ModuleTypeSpringStage(), pluginRoot,
                  "beans:bean[@id='moduleType-foo' and @class='" + SingleModuleDescriptorFactory.class.getName() + "']",
                  "osgi:service[@id='moduleType-foo_osgiService' and @auto-export='interfaces']");
      }
  
      @Test(expected = PluginParseException.class)
      public void testTransformOfBadElementKey() throws IOException, DocumentException {
          Element pluginRoot = DocumentHelper.createDocument().addElement("atlassian-plugin");
          pluginRoot.addElement("module-type");
  
          SpringTransformerTestHelper.transform(new ModuleTypeSpringStage(), pluginRoot,
                  "beans:bean[@id='moduleType-foo' and @class='" + SingleModuleDescriptorFactory.class.getName() + "']",
                  "osgi:service[@id='moduleType-foo_osgiService' and @auto-export='interfaces']");
      }
  
      @Test
      public void testBeanTracking() throws Exception {
          OsgiContainerManager osgiContainerManager = mock(OsgiContainerManager.class);
          when(osgiContainerManager.getRegisteredServices()).thenReturn(new ServiceReference[0]);
  
          final File plugin = new PluginJarBuilder("plugin")
                  .addFormattedResource("atlassian-plugin.xml",
                          "<atlassian-plugin name='Test Bundle instruction plugin 2' key='test.plugin'>",
                          "    <plugin-info>",
                          "        <version>1.0</version>",
                          "    </plugin-info>",
                          "    <module-type key='mod1' class='my.FooDescriptor' />",
                          "</atlassian-plugin>")
                  .build();
  
          ModuleTypeSpringStage stage = new ModuleTypeSpringStage();
          final TransformContext context = new TransformContext(Collections.<HostComponentRegistration>emptyList(), SystemExports.NONE, new JarPluginArtifact(plugin),
                  null, PluginAccessor.Descriptor.FILENAME, osgiContainerManager);
          stage.execute(context);
 
         // springHostContainer should always exist if a module type is declared.
         assertTrue(context.beanExists("springHostContainer"));
         // these two names are generated from the base key, one for the module type itself and the other for service export.
         assertTrue(context.beanExists("moduleType-mod1"));
         assertTrue(context.beanExists("moduleType-mod1_osgiService"));
     }
 }