Chris Aniszczyk
14 years ago
committed by
Code Review
4 changed files with 359 additions and 51 deletions
@ -0,0 +1,254 @@ |
|||||||
|
/* |
||||||
|
* 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(); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
Loading…
Reference in new issue