--- /dev/null
+/*
+ * Copyright (C) 2011, Mathias Kinzler <mathias.kinzler@sap.com>
+ * and other copyright owners as documented in the project's IP log.
+ *
+ * This program and the accompanying materials are made available
+ * under the terms of the Eclipse Distribution License v1.0 which
+ * accompanies this distribution, is reproduced below, and is
+ * available at http://www.eclipse.org/org/documents/edl-v10.php
+ *
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or
+ * without modification, are permitted provided that the following
+ * conditions are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ *
+ * - Redistributions in binary form must reproduce the above
+ * copyright notice, this list of conditions and the following
+ * disclaimer in the documentation and/or other materials provided
+ * with the distribution.
+ *
+ * - Neither the name of the Eclipse Foundation, Inc. nor the
+ * names of its contributors may be used to endorse or promote
+ * products derived from this software without specific prior
+ * written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
+ * CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
+ * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+package org.eclipse.jgit.api;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNull;
+import static org.junit.Assert.assertTrue;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.io.IOException;
+
+import org.eclipse.jgit.api.CreateBranchCommand.SetupUpstreamMode;
+import org.eclipse.jgit.api.RebaseResult.Status;
+import org.eclipse.jgit.lib.Constants;
+import org.eclipse.jgit.lib.RepositoryState;
+import org.eclipse.jgit.lib.RepositoryTestCase;
+import org.eclipse.jgit.lib.StoredConfig;
+import org.eclipse.jgit.storage.file.FileRepository;
+import org.eclipse.jgit.transport.RefSpec;
+import org.eclipse.jgit.transport.RemoteConfig;
+import org.eclipse.jgit.transport.URIish;
+import org.junit.Before;
+import org.junit.Test;
+
+public class PullCommandWithRebaseTest extends RepositoryTestCase {
+ /** Second Test repository */
+ protected FileRepository dbTarget;
+
+ private Git source;
+
+ private Git target;
+
+ private File sourceFile;
+
+ private File targetFile;
+
+ @Test
+ public void testPullFastForward() throws Exception {
+ PullResult res = target.pull().call();
+ // nothing to update since we don't have different data yet
+ assertTrue(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+ assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+
+ assertFileContentsEqual(targetFile, "Hello world");
+
+ // change the source file
+ writeToFile(sourceFile, "Another change");
+ source.add().addFilepattern("SomeFile.txt").call();
+ source.commit().setMessage("Some change in remote").call();
+
+ res = target.pull().call();
+
+ assertFalse(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+ assertEquals(Status.FAST_FORWARD, res.getRebaseResult().getStatus());
+ assertFileContentsEqual(targetFile, "Another change");
+ assertEquals(RepositoryState.SAFE, target.getRepository()
+ .getRepositoryState());
+
+ res = target.pull().call();
+ assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+ }
+
+ @Test
+ public void testPullConflict() throws Exception {
+ PullResult res = target.pull().call();
+ // nothing to update since we don't have different data yet
+ assertTrue(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+ assertTrue(res.getRebaseResult().getStatus().equals(Status.UP_TO_DATE));
+
+ assertFileContentsEqual(targetFile, "Hello world");
+
+ // change the source file
+ writeToFile(sourceFile, "Source change");
+ source.add().addFilepattern("SomeFile.txt").call();
+ source.commit().setMessage("Source change in remote").call();
+
+ // change the target file
+ writeToFile(targetFile, "Target change");
+ target.add().addFilepattern("SomeFile.txt").call();
+ target.commit().setMessage("Target change in local").call();
+
+ res = target.pull().call();
+
+ assertFalse(res.getFetchResult().getTrackingRefUpdates().isEmpty());
+ assertTrue(res.getRebaseResult().getStatus().equals(Status.STOPPED));
+ String result = "<<<<<<< OURS\nSource change\n=======\nTarget change\n>>>>>>> THEIRS\n";
+ assertFileContentsEqual(targetFile, result);
+ assertEquals(RepositoryState.REBASING_INTERACTIVE, target
+ .getRepository().getRepositoryState());
+ }
+
+ @Test
+ public void testPullLocalConflict() throws Exception {
+ target.branchCreate().setName("basedOnMaster").setStartPoint(
+ "refs/heads/master").setUpstreamMode(SetupUpstreamMode.NOTRACK)
+ .call();
+ StoredConfig config = target.getRepository().getConfig();
+ config.setString("branch", "basedOnMaster", "remote", ".");
+ config.setString("branch", "basedOnMaster", "rebase",
+ "refs/heads/master");
+ config.save();
+ target.getRepository().updateRef(Constants.HEAD).link(
+ "refs/heads/basedOnMaster");
+ PullResult res = target.pull().call();
+ // nothing to update since we don't have different data yet
+ assertNull(res.getFetchResult());
+ assertEquals(Status.UP_TO_DATE, res.getRebaseResult().getStatus());
+
+ assertFileContentsEqual(targetFile, "Hello world");
+
+ // change the file in master
+ target.getRepository().updateRef(Constants.HEAD).link(
+ "refs/heads/master");
+ writeToFile(targetFile, "Master change");
+ target.add().addFilepattern("SomeFile.txt").call();
+ target.commit().setMessage("Source change in master").call();
+
+ // change the file in slave
+ target.getRepository().updateRef(Constants.HEAD).link(
+ "refs/heads/basedOnMaster");
+ writeToFile(targetFile, "Slave change");
+ target.add().addFilepattern("SomeFile.txt").call();
+ target.commit().setMessage("Source change in based on master").call();
+
+ res = target.pull().call();
+
+ assertNull(res.getFetchResult());
+ assertEquals(Status.STOPPED, res.getRebaseResult().getStatus());
+ String result = "<<<<<<< OURS\nMaster change\n=======\nSlave change\n>>>>>>> THEIRS\n";
+ assertFileContentsEqual(targetFile, result);
+ assertEquals(RepositoryState.REBASING_INTERACTIVE, target
+ .getRepository().getRepositoryState());
+ }
+
+ @Override
+ @Before
+ public void setUp() throws Exception {
+ super.setUp();
+ dbTarget = createWorkRepository();
+ source = new Git(db);
+ target = new Git(dbTarget);
+
+ // put some file in the source repo
+ sourceFile = new File(db.getWorkTree(), "SomeFile.txt");
+ writeToFile(sourceFile, "Hello world");
+ // and commit it
+ source.add().addFilepattern("SomeFile.txt").call();
+ source.commit().setMessage("Initial commit for source").call();
+
+ // configure the target repo to connect to the source via "origin"
+ StoredConfig targetConfig = dbTarget.getConfig();
+ targetConfig.setString("branch", "master", "remote", "origin");
+ targetConfig
+ .setString("branch", "master", "merge", "refs/heads/master");
+ RemoteConfig config = new RemoteConfig(targetConfig, "origin");
+
+ config
+ .addURI(new URIish(source.getRepository().getWorkTree()
+ .getPath()));
+ config.addFetchRefSpec(new RefSpec(
+ "+refs/heads/*:refs/remotes/origin/*"));
+ config.update(targetConfig);
+ targetConfig.save();
+
+ targetFile = new File(dbTarget.getWorkTree(), "SomeFile.txt");
+ // make sure we have the same content
+ target.pull().call();
+ target.checkout().setStartPoint("refs/remotes/origin/master").setName(
+ "master").call();
+
+ targetConfig.setString("branch", "master", "rebase",
+ "refs/remotes/origin/master");
+ targetConfig.unset("branch", "master", "merge");
+ targetConfig.save();
+
+ assertFileContentsEqual(targetFile, "Hello world");
+ }
+
+ private void writeToFile(File actFile, String string) throws IOException {
+ FileOutputStream fos = null;
+ try {
+ fos = new FileOutputStream(actFile);
+ fos.write(string.getBytes("UTF-8"));
+ fos.close();
+ } finally {
+ if (fos != null)
+ fos.close();
+ }
+ }
+
+ private void assertFileContentsEqual(File actFile, String string)
+ throws IOException {
+ ByteArrayOutputStream bos = new ByteArrayOutputStream();
+ FileInputStream fis = null;
+ byte[] buffer = new byte[100];
+ try {
+ fis = new FileInputStream(actFile);
+ int read = fis.read(buffer);
+ while (read > 0) {
+ bos.write(buffer, 0, read);
+ read = fis.read(buffer);
+ }
+ String content = new String(bos.toByteArray(), "UTF-8");
+ assertEquals(string, content);
+ } finally {
+ if (fis != null)
+ fis.close();
+ }
+ }
+}
import java.text.MessageFormat;
import org.eclipse.jgit.JGitText;
+import org.eclipse.jgit.api.RebaseCommand.Operation;
import org.eclipse.jgit.api.errors.CanceledException;
import org.eclipse.jgit.api.errors.CheckoutConflictException;
import org.eclipse.jgit.api.errors.ConcurrentRefUpdateException;
import org.eclipse.jgit.api.errors.DetachedHeadException;
+import org.eclipse.jgit.api.errors.GitAPIException;
import org.eclipse.jgit.api.errors.InvalidConfigurationException;
import org.eclipse.jgit.api.errors.InvalidMergeHeadsException;
import org.eclipse.jgit.api.errors.InvalidRemoteException;
import org.eclipse.jgit.api.errors.JGitInternalException;
import org.eclipse.jgit.api.errors.NoHeadException;
import org.eclipse.jgit.api.errors.NoMessageException;
+import org.eclipse.jgit.api.errors.RefNotFoundException;
import org.eclipse.jgit.api.errors.WrongRepositoryStateException;
import org.eclipse.jgit.lib.AnyObjectId;
import org.eclipse.jgit.lib.Config;
String remoteBranchName = repoConfig.getString(
ConfigConstants.CONFIG_BRANCH_SECTION, branchName,
ConfigConstants.CONFIG_KEY_MERGE);
+ boolean doRebase = false;
if (remoteBranchName == null) {
// check if the branch is configured for pull-rebase
remoteBranchName = repoConfig.getString(
ConfigConstants.CONFIG_BRANCH_SECTION, branchName,
ConfigConstants.CONFIG_KEY_REBASE);
if (remoteBranchName != null) {
- // TODO implement pull-rebase
- throw new JGitInternalException(
- "Pull with rebase is not yet supported");
+ doRebase = true;
}
}
monitor.update(1);
- // we check the updates to see which of the updated branches corresponds
- // to the remote branch name
+ if (monitor.isCancelled())
+ throw new CanceledException(MessageFormat.format(
+ JGitText.get().operationCanceled,
+ JGitText.get().pullTaskName));
- AnyObjectId commitToMerge;
+ PullResult result;
+ if (doRebase) {
+ RebaseCommand rebase = new RebaseCommand(repo);
+ try {
+ RebaseResult rebaseRes = rebase.setUpstream(remoteBranchName)
+ .setProgressMonitor(monitor).setOperation(
+ Operation.BEGIN).call();
+ result = new PullResult(fetchRes, remote, rebaseRes);
+ } catch (NoHeadException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (RefNotFoundException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (JGitInternalException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (GitAPIException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ }
+ } else {
+ // we check the updates to see which of the updated branches
+ // corresponds
+ // to the remote branch name
+ AnyObjectId commitToMerge;
- if (isRemote) {
- Ref r = null;
- if (fetchRes != null) {
- r = fetchRes.getAdvertisedRef(remoteBranchName);
+ if (isRemote) {
+ Ref r = null;
+ if (fetchRes != null) {
+ r = fetchRes.getAdvertisedRef(remoteBranchName);
+ if (r == null)
+ r = fetchRes.getAdvertisedRef(Constants.R_HEADS
+ + remoteBranchName);
+ }
if (r == null)
- r = fetchRes.getAdvertisedRef(Constants.R_HEADS
- + remoteBranchName);
+ throw new JGitInternalException(MessageFormat.format(
+ JGitText.get().couldNotGetAdvertisedRef,
+ remoteBranchName));
+ else
+ commitToMerge = r.getObjectId();
+ } else {
+ try {
+ commitToMerge = repo.resolve(remoteBranchName);
+ } catch (IOException e) {
+ throw new JGitInternalException(
+ JGitText.get().exceptionCaughtDuringExecutionOfPullCommand,
+ e);
+ }
}
- if (r == null)
- throw new JGitInternalException(MessageFormat.format(JGitText
- .get().couldNotGetAdvertisedRef, remoteBranchName));
- else
- commitToMerge = r.getObjectId();
- } else {
+ MergeCommand merge = new MergeCommand(repo);
+ merge.include(
+ "branch \'" + remoteBranchName + "\' of " + remoteUri,
+ commitToMerge);
+ MergeResult mergeRes;
try {
- commitToMerge = repo.resolve(remoteBranchName);
- } catch (IOException e) {
- throw new JGitInternalException(
- JGitText.get().exceptionCaughtDuringExecutionOfPullCommand,
- e);
+ mergeRes = merge.call();
+ monitor.update(1);
+ result = new PullResult(fetchRes, remote, mergeRes);
+ } catch (NoHeadException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (ConcurrentRefUpdateException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (CheckoutConflictException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (InvalidMergeHeadsException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (WrongRepositoryStateException e) {
+ throw new JGitInternalException(e.getMessage(), e);
+ } catch (NoMessageException e) {
+ throw new JGitInternalException(e.getMessage(), e);
}
}
-
- if (monitor.isCancelled())
- throw new CanceledException(MessageFormat.format(
- JGitText.get().operationCanceled,
- JGitText.get().pullTaskName));
-
- MergeCommand merge = new MergeCommand(repo);
- merge.include("branch \'" + remoteBranchName + "\' of " + remoteUri,
- commitToMerge);
- MergeResult mergeRes;
- try {
- mergeRes = merge.call();
- monitor.update(1);
- } catch (NoHeadException e) {
- throw new JGitInternalException(e.getMessage(), e);
- } catch (ConcurrentRefUpdateException e) {
- throw new JGitInternalException(e.getMessage(), e);
- } catch (CheckoutConflictException e) {
- throw new JGitInternalException(e.getMessage(), e);
- } catch (InvalidMergeHeadsException e) {
- throw new JGitInternalException(e.getMessage(), e);
- } catch (WrongRepositoryStateException e) {
- throw new JGitInternalException(e.getMessage(), e);
- } catch (NoMessageException e) {
- throw new JGitInternalException(e.getMessage(), e);
- }
monitor.endTask();
- return new PullResult(fetchRes, remote, mergeRes);
+ return result;
}
}
projects / jgit.git / commitdiff