package com.atlassian.sal.core.upgrade;
   
   import com.atlassian.plugin.Plugin;
   import com.atlassian.plugin.PluginAccessor;
   import com.atlassian.plugin.event.PluginEventManager;
   import com.atlassian.sal.api.message.Message;
   import com.atlassian.sal.api.pluginsettings.PluginSettings;
   import com.atlassian.sal.api.pluginsettings.PluginSettingsFactory;
   import com.atlassian.sal.api.transaction.TransactionCallback;
  import com.atlassian.sal.api.transaction.TransactionTemplate;
  import com.atlassian.sal.api.upgrade.PluginUpgradeTask;
  import com.atlassian.sal.core.message.DefaultMessage;
  
  import org.hamcrest.Matchers;
  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 java.util.Arrays;
  import java.util.Collections;
  import java.util.List;
  
  import static com.atlassian.sal.core.upgrade.PluginUpgrader.BUILD;
  import static java.util.Collections.singletonList;
  import static org.hamcrest.Matchers.empty;
  import static org.hamcrest.Matchers.is;
  import static org.hamcrest.Matchers.not;
  import static org.junit.Assert.assertThat;
  import static org.mockito.Matchers.anyString;
  import static org.mockito.Mockito.never;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.when;
  
  @RunWith (MockitoJUnitRunner.class)
  public class DefaultPluginUpgradeManagerTest
  {
      @Mock private Plugin plugin;
      @Mock private Plugin secondPlugin;
      @Mock private PluginUpgradeTask pluginUpgradeTask;
      @Mock private PluginUpgradeTask secondPluginUpgradeTask;
      @Mock private PluginUpgradeTask thirdPluginUpgradeTask;
      @Mock private PluginAccessor pluginAccessor;
      @Mock private PluginSettingsFactory pluginSettingsFactory;
      @Mock private PluginEventManager pluginEventManager;
      @Mock private PluginSettings pluginSettings;
  
      private static final String PLUGIN_KEY = "com.atlassian.plugin.test.upgrade-test";
      private static final String SECOND_PLUGIN_KEY = "com.atlassian.plugin2.test.upgrade-test";
  
      private DefaultPluginUpgradeManager pluginUpgradeManager;
      private CountingTransactionTemplate transactionTemplate;
  
      @Before
      public void setUp() throws Exception
      {
          transactionTemplate = new CountingTransactionTemplate();
  
          when(plugin.getKey()).thenReturn(PLUGIN_KEY);
          when(secondPlugin.getKey()).thenReturn(SECOND_PLUGIN_KEY);
  
          when(pluginAccessor.getPlugin(PLUGIN_KEY)).thenReturn(plugin);
          when(pluginAccessor.getPlugin(SECOND_PLUGIN_KEY)).thenReturn(secondPlugin);
  
          when(pluginUpgradeTask.getPluginKey()).thenReturn(PLUGIN_KEY);
          when(pluginUpgradeTask.getBuildNumber()).thenReturn(1);
          when(pluginUpgradeTask.doUpgrade()).thenReturn(Collections.<Message>emptyList()); // default is success
  
          when(secondPluginUpgradeTask.getPluginKey()).thenReturn(PLUGIN_KEY);
          when(secondPluginUpgradeTask.getBuildNumber()).thenReturn(2);
          when(secondPluginUpgradeTask.doUpgrade()).thenReturn(Collections.<Message>emptyList()); // default is success
  
          when(thirdPluginUpgradeTask.getPluginKey()).thenReturn(SECOND_PLUGIN_KEY);
          when(thirdPluginUpgradeTask.getBuildNumber()).thenReturn(1);
          when(thirdPluginUpgradeTask.doUpgrade()).thenReturn(Collections.<Message>emptyList()); // default is success
  
          when(pluginSettingsFactory.createGlobalSettings()).thenReturn(pluginSettings);
      }
  
      @After
      public void tearDown()
      {
          transactionTemplate.resetTransactionCount();
      }
  
      @Test
      public void testUpgradeInternalIsSuccessful() throws Exception
      {
          pluginUpgradeManager = new DefaultPluginUpgradeManager(singletonList(pluginUpgradeTask), transactionTemplate, pluginAccessor, pluginSettingsFactory, pluginEventManager);
  
          when(pluginSettings.get(plugin.getKey() + BUILD)).thenReturn(Integer.toString(0)); // last upgrade task (data build number)
  
          List<Message> messages = pluginUpgradeManager.upgradeInternal(plugin);
  
          // A successful upgrade task should increment the build data number by a put on plugin settings
          verify(pluginSettings).put(PLUGIN_KEY + BUILD, "1");
  
         assertThat(messages, empty());
     }
 
     @Test
     public void testUpgradeInternalFailsIfUpgradeFails() throws Exception
     {
         pluginUpgradeManager = new DefaultPluginUpgradeManager(singletonList(pluginUpgradeTask), transactionTemplate, pluginAccessor, pluginSettingsFactory, pluginEventManager);
 
         when(pluginUpgradeTask.doUpgrade()).thenReturn(Collections.<Message>singletonList(new DefaultMessage("failed.upgrade")));
 
         List<Message> messages = pluginUpgradeManager.upgradeInternal(plugin);
 
         // A failed upgrade task should not increment the build data number in plugin settings
         verify(pluginSettings, never()).put(anyString(), anyString());
 
         assertThat(messages, not(empty()));
     }
 
     @Test
     public void testThatOnlyOneTransactionTemplateIsExecutedPerPluginUpgrade() throws Exception
     {
         pluginUpgradeManager = new DefaultPluginUpgradeManager(Arrays.asList(pluginUpgradeTask, secondPluginUpgradeTask), transactionTemplate, pluginAccessor, pluginSettingsFactory, pluginEventManager);
 
         List<Message> messages = pluginUpgradeManager.upgradeInternal(plugin);
 
         assertThat(messages, empty());
         assertThat(transactionTemplate.getTransactionCount(), is(1));
 
         verify(pluginSettings).put(PLUGIN_KEY + BUILD, "2"); // from pluginUpgradeTask, secondPluginUpgradeTask
     }
 
     @Test
     public void testThatOnlyOneTransactionTemplateIsExecutedPerPluginUpgradeWithMultiplePlugins() throws Exception
     {
         pluginUpgradeManager = new DefaultPluginUpgradeManager(Arrays.asList(pluginUpgradeTask, secondPluginUpgradeTask, thirdPluginUpgradeTask), transactionTemplate, pluginAccessor, pluginSettingsFactory, pluginEventManager);
 
         List<Message> messages = pluginUpgradeManager.upgradeInternal();
 
         verify(pluginSettings).put(PLUGIN_KEY + BUILD, "2"); // from pluginUpgradeTask, secondPluginUpgradeTask
         verify(pluginSettings).put(SECOND_PLUGIN_KEY + BUILD, "1"); // thirdPluginUpgradeTask
 
         assertThat(messages, empty());
         assertThat(transactionTemplate.getTransactionCount(), is(2));
     }
 
     private static class CountingTransactionTemplate implements TransactionTemplate
     {
         private int transactionCount = 0;
 
         @Override
         public <T> T execute(final TransactionCallback<T> action)
         {
             transactionCount++;
             return action.doInTransaction();
         }
 
         void resetTransactionCount()
         {
             transactionCount = 0;
         }
 
         int getTransactionCount()
         {
             return transactionCount;
         }
     }
 }