Unit tests for ssh daemon and keys dispatcher

James Moger

2014-04-07 521cb6022a9ee30bf3115a8dcb991aa5c7e420e3

 src/main/java/com/gitblit/manager/GitblitManager.java

@@ -33,9 +33,18 @@
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.eclipse.jgit.api.CloneCommand;
import org.eclipse.jgit.api.FetchCommand;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.lib.Ref;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.transport.RefSpec;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ro.fortsoft.pf4j.PluginClassLoader;
import ro.fortsoft.pf4j.PluginWrapper;
import ro.fortsoft.pf4j.RuntimeMode;

import com.gitblit.Constants;
import com.gitblit.Constants.AccessPermission;
@@ -52,6 +61,8 @@
import com.gitblit.models.GitClientApplication;
import com.gitblit.models.Mailing;
import com.gitblit.models.Metric;
import com.gitblit.models.PluginRegistry.PluginRegistration;
import com.gitblit.models.PluginRegistry.PluginRelease;
import com.gitblit.models.ProjectModel;
import com.gitblit.models.RegistrantAccessPermission;
import com.gitblit.models.RepositoryModel;
@@ -62,9 +73,11 @@
import com.gitblit.models.SettingModel;
import com.gitblit.models.TeamModel;
import com.gitblit.models.UserModel;
import com.gitblit.tickets.ITicketService;
import com.gitblit.transport.ssh.IPublicKeyManager;
import com.gitblit.transport.ssh.SshKey;
import com.gitblit.utils.ArrayUtils;
import com.gitblit.utils.HttpUtils;
import com.gitblit.utils.JGitUtils;
import com.gitblit.utils.JsonUtils;
import com.gitblit.utils.ObjectCache;
import com.gitblit.utils.StringUtils;
@@ -95,11 +108,15 @@

   protected final IRuntimeManager runtimeManager;

   protected final IPluginManager pluginManager;

   protected final INotificationManager notificationManager;

   protected final IUserManager userManager;

   protected final IAuthenticationManager authenticationManager;

   protected final IPublicKeyManager publicKeyManager;

   protected final IRepositoryManager repositoryManager;

@@ -109,18 +126,22 @@

   public GitblitManager(
         IRuntimeManager runtimeManager,
         IPluginManager pluginManager,
         INotificationManager notificationManager,
         IUserManager userManager,
         IAuthenticationManager authenticationManager,
         IPublicKeyManager publicKeyManager,
         IRepositoryManager repositoryManager,
         IProjectManager projectManager,
         IFederationManager federationManager) {

      this.settings = runtimeManager.getSettings();
      this.runtimeManager = runtimeManager;
      this.pluginManager = pluginManager;
      this.notificationManager = notificationManager;
      this.userManager = userManager;
      this.authenticationManager = authenticationManager;
      this.publicKeyManager = publicKeyManager;
      this.repositoryManager = repositoryManager;
      this.projectManager = projectManager;
      this.federationManager = federationManager;
@@ -158,7 +179,33 @@

      // clone the repository
      try {
         JGitUtils.cloneRepository(repositoryManager.getRepositoriesFolder(), cloneName, fromUrl, true, null);
         Repository canonical = getRepository(repository.name);
         File folder = new File(repositoryManager.getRepositoriesFolder(), cloneName);
         CloneCommand clone = new CloneCommand();
         clone.setBare(true);

         // fetch branches with exclusions
         Collection<Ref> branches = canonical.getRefDatabase().getRefs(Constants.R_HEADS).values();
         List<String> branchesToClone = new ArrayList<String>();
         for (Ref branch : branches) {
            String name = branch.getName();
            if (name.startsWith(Constants.R_TICKET)) {
               // exclude ticket branches
               continue;
            }
            branchesToClone.add(name);
         }
         clone.setBranchesToClone(branchesToClone);
         clone.setURI(fromUrl);
         clone.setDirectory(folder);
         Git git = clone.call();

         // fetch tags
         FetchCommand fetch  = git.fetch();
         fetch.setRefSpecs(new RefSpec("+refs/tags/*:refs/tags/*"));
         fetch.call();

         git.getRepository().close();
      } catch (Exception e) {
         throw new GitBlitException(e);
      }
@@ -483,6 +530,20 @@
      }
   }

   /**
    * Throws an exception if trying to get a ticket service.
    *
    */
   @Override
   public ITicketService getTicketService() {
      throw new RuntimeException("This class does not have a ticket service!");
   }

   @Override
   public IPublicKeyManager getPublicKeyManager() {
      return publicKeyManager;
   }

   /*
    * ISTOREDSETTINGS
    *
@@ -611,6 +672,12 @@
      }
      return user;
   }

   @Override
   public UserModel authenticate(String username, SshKey key) {
      return authenticationManager.authenticate(username, key);
   }

   @Override
   public UserModel authenticate(HttpServletRequest httpRequest, boolean requiresCertificate) {
      UserModel user = authenticationManager.authenticate(httpRequest, requiresCertificate);
@@ -1099,4 +1166,113 @@
   public boolean deletePendingFederationProposal(FederationProposal proposal) {
      return federationManager.deletePendingFederationProposal(proposal);
   }

   @Override
   public void closeAll() {
      repositoryManager.closeAll();
   }

   @Override
   public void close(String repository) {
      repositoryManager.close(repository);
   }

   @Override
   public boolean isIdle(Repository repository) {
      return repositoryManager.isIdle(repository);
   }

   /*
    * PLUGIN MANAGER
    */

   @Override
   public <T> List<T> getExtensions(Class<T> clazz) {
      return pluginManager.getExtensions(clazz);
   }

   @Override
   public PluginWrapper whichPlugin(Class<?> clazz) {
      return pluginManager.whichPlugin(clazz);
   }

   @Override
   public boolean deletePlugin(PluginWrapper wrapper) {
      return pluginManager.deletePlugin(wrapper);
   }

   @Override
   public boolean refreshRegistry() {
      return pluginManager.refreshRegistry();
   }

   @Override
   public boolean installPlugin(String url) {
      return pluginManager.installPlugin(url);
   }

   @Override
   public boolean installPlugin(PluginRelease pv) {
      return pluginManager.installPlugin(pv);
   }

   @Override
   public List<PluginRegistration> getRegisteredPlugins() {
      return pluginManager.getRegisteredPlugins();
   }

   @Override
   public PluginRegistration lookupPlugin(String idOrName) {
      return pluginManager.lookupPlugin(idOrName);
   }

   @Override
   public PluginRelease lookupRelease(String idOrName, String version) {
      return pluginManager.lookupRelease(idOrName, version);
   }

   @Override
   public List<PluginWrapper> getPlugins() {
      return pluginManager.getPlugins();
   }

   @Override
   public List<PluginWrapper> getResolvedPlugins() {
      return pluginManager.getResolvedPlugins();
   }

   @Override
   public List<PluginWrapper> getUnresolvedPlugins() {
      return pluginManager.getUnresolvedPlugins();
   }

   @Override
   public List<PluginWrapper> getStartedPlugins() {
      return pluginManager.getStartedPlugins();
   }

   @Override
   public void loadPlugins() {
      pluginManager.loadPlugins();
   }

   @Override
   public void startPlugins() {
      pluginManager.startPlugins();
   }

   @Override
   public void stopPlugins() {
      pluginManager.stopPlugins();
   }

   @Override
   public PluginClassLoader getPluginClassLoader(String pluginId) {
      return pluginManager.getPluginClassLoader(pluginId);
   }

   @Override
   public RuntimeMode getRuntimeMode() {
      return pluginManager.getRuntimeMode();
   }
}