Merged #6 "Support serving repositories over the SSH transport"

James Moger

2014-04-11 436bd3f0ecdee282c503a9eb0f7a240b7a68ff49

 src/test/java/com/gitblit/tests/SshDaemonTest.java

New file
@@ -0,0 +1,90 @@
/*
 * Copyright 2014 gitblit.com.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.gitblit.tests;

import java.io.File;
import java.text.MessageFormat;
import java.util.List;

import org.apache.sshd.ClientSession;
import org.apache.sshd.SshClient;
import org.eclipse.jgit.api.CloneCommand;
import org.eclipse.jgit.api.Git;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.jgit.transport.SshSessionFactory;
import org.eclipse.jgit.transport.UsernamePasswordCredentialsProvider;
import org.eclipse.jgit.util.FileUtils;
import org.junit.Test;

import com.gitblit.Constants;
import com.gitblit.Constants.AccessRestrictionType;
import com.gitblit.Constants.AuthorizationControl;
import com.gitblit.models.RepositoryModel;
import com.gitblit.utils.JGitUtils;

public class SshDaemonTest extends SshUnitTest {

   static File ticgitFolder = new File(GitBlitSuite.REPOSITORIES, "working/ticgit");

   String url = GitBlitSuite.sshDaemonUrl;

   @Test
   public void testPublicKeyAuthentication() throws Exception {
      SshClient client = getClient();
      ClientSession session = client.connect(username, "localhost", GitBlitSuite.sshPort).await().getSession();
      session.addPublicKeyIdentity(rwKeyPair);
      assertTrue(session.auth().await().isSuccess());
   }

   @Test
   public void testVersionCommand() throws Exception {
      String result = testSshCommand("version");
      assertEquals(Constants.getGitBlitVersion(), result);
   }

   @Test
   public void testCloneCommand() throws Exception {
      if (ticgitFolder.exists()) {
         GitBlitSuite.close(ticgitFolder);
         FileUtils.delete(ticgitFolder, FileUtils.RECURSIVE);
      }

      // set clone restriction
      RepositoryModel model = repositories().getRepositoryModel("ticgit.git");
      model.accessRestriction = AccessRestrictionType.CLONE;
      model.authorizationControl = AuthorizationControl.NAMED;
      repositories().updateRepositoryModel(model.name, model, false);

      JschConfigTestSessionFactory sessionFactory = new JschConfigTestSessionFactory(roKeyPair);
      SshSessionFactory.setInstance(sessionFactory);

      CloneCommand clone = Git.cloneRepository();
      clone.setCredentialsProvider(new UsernamePasswordCredentialsProvider(username, password));
      clone.setURI(MessageFormat.format("{0}/ticgit.git", url));
      clone.setDirectory(ticgitFolder);
      clone.setBare(false);
      clone.setCloneAllBranches(true);
      Git git = clone.call();
      List<RevCommit> commits = JGitUtils.getRevLog(git.getRepository(), 10);
      GitBlitSuite.close(git);
      assertEquals(10, commits.size());

      // restore anonymous repository access
      model.accessRestriction = AccessRestrictionType.NONE;
      model.authorizationControl = AuthorizationControl.NAMED;
      repositories().updateRepositoryModel(model.name, model, false);
   }
}