Download zip feature.

James Moger

2011-05-24 9197d340db81a245193dff1ecb44889b8e0dfe31

 src/com/gitblit/tests/JGitUtilsTest.java

@@ -1,6 +1,7 @@
package com.gitblit.tests;



import java.io.File;

import java.io.FileOutputStream;

import java.util.Date;

import java.util.List;



@@ -15,9 +16,10 @@
import org.eclipse.jgit.storage.file.FileRepository;



import com.gitblit.utils.JGitUtils;

import com.gitblit.wicket.models.PathModel;

import com.gitblit.utils.JGitUtils.DiffOutputType;

import com.gitblit.wicket.models.PathModel.PathChangeModel;

import com.gitblit.wicket.models.RefModel;

import com.gitblit.wicket.models.TicGitTicket;

import com.gitblit.wicket.models.TicketModel;



public class JGitUtilsTest extends TestCase {



@@ -50,6 +52,14 @@
      r.close();

      assertTrue("Could not get last repository change date!", date != null);

   }

	
   public void testFirstCommit() throws Exception {

      Repository r = getRepository();

      RevCommit commit = JGitUtils.getFirstCommit(r, null);

      r.close();

      assertTrue("Could not get first commit!", commit != null);

      System.out.println(commit.getName() + " " + commit.getShortMessage());

   }



   public void testRetrieveRevObject() throws Exception {

      Repository r = getRepository();

@@ -72,9 +82,9 @@


   public void testTicGit() throws Exception {

      Repository r = new FileRepository(new File(repositoriesFolder, "ticgit") + "/" + Constants.DOT_GIT);

      RefModel ticgit = JGitUtils.getTicGitBranch(r);

      RefModel ticgit = JGitUtils.getTicketsBranch(r);

      assertTrue("Ticgit branch does not exist!", ticgit != null);

      List<TicGitTicket> tickets = JGitUtils.getTicGitTickets(r);

      List<TicketModel> tickets = JGitUtils.getTickets(r);

      assertTrue("No tickets found!", tickets.size() > 0);

      r.close();

   }

@@ -82,17 +92,39 @@
   public void testFilesInCommit() throws Exception {

      Repository r = getRepository();

      RevCommit commit = JGitUtils.getCommit(r, Constants.HEAD);

      List<PathModel> paths = JGitUtils.getFilesInCommit(r, commit);

      List<PathChangeModel> paths = JGitUtils.getFilesInCommit(r, commit);

      r.close();

      assertTrue("No changed paths found!", paths.size() > 0);

   }

	


   public void testCommitDiff() throws Exception {

      Repository r = getRepository();

      RevCommit commit = JGitUtils.getCommit(r, Constants.HEAD);

      String diff = JGitUtils.getCommitDiff(r, commit, false);

      String diff = JGitUtils.getCommitDiff(r, commit, DiffOutputType.PLAIN);

      r.close();

      System.out.println(diff);

   }

	
   public void testZip() throws Exception {

      Repository r = new FileRepository(new File(repositoriesFolder, "gitblit.git/" + Constants.DOT_GIT));

      FileOutputStream fos = null;

      try {

         File zipFile = new File("c:/output.zip");

         zipFile.delete();

         fos = new FileOutputStream(zipFile);

         if (JGitUtils.zip(r, "src", Constants.HEAD, fos)) {

            System.out.println("zip = " + zipFile.length() + " bytes");

         } else {

            System.err.println("failed to generate zip file?!");

         }

      } finally {

         if (fos != null) {

            try {

               fos.close();

            } catch (Throwable t) {

            }

         }

      }

   }



}