/*******************************************************************************
* Copyright (C) 2010, 2013 Dariusz Luksza <dariusz@luksza.org> and others.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* which accompanies this distribution, and is available at
* http://www.eclipse.org/legal/epl-v10.html
*******************************************************************************/
package org.eclipse.egit.core.synchronize;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import java.io.File;
import java.io.IOException;
import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IResource;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.egit.core.synchronize.dto.GitSynchronizeData;
import org.eclipse.egit.core.synchronize.dto.GitSynchronizeDataSet;
import org.eclipse.egit.core.test.GitTestCase;
import org.eclipse.egit.core.test.TestRepository;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.revwalk.RevCommit;
import org.eclipse.team.core.TeamException;
import org.eclipse.team.core.variants.IResourceVariant;
import org.eclipse.team.core.variants.IResourceVariantTree;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
public class GitResourceVariantTreeSubscriberTest extends GitTestCase {
private static final String MASTER = Constants.R_HEADS + Constants.MASTER;
private static final String BRANCH = Constants.R_HEADS + "branch";
private TestRepository testRepo;
private RevCommit initialCommit;
private RevCommit commitBranch;
private RevCommit commitMaster;
private IFile changedFile;
@Before
public void setUp() throws Exception {
super.setUp();
testRepo = new TestRepository(gitDir);
testRepo.connect(project.getProject());
String fileName = "Main.java";
File file = testRepo.createFile(project.getProject(), fileName);
initialCommit = testRepo.appendContentAndCommit(project.getProject(),
file, "class Main {}", "initial commit");
changedFile = testRepo.getIFile(project.getProject(), file);
testRepo.createAndCheckoutBranch(Constants.HEAD, BRANCH);
commitBranch = testRepo.appendContentAndCommit(project.getProject(),
file, "// test 1", "second commit");
testRepo.checkoutBranch(MASTER);
commitMaster = testRepo.appendContentAndCommit(project.getProject(),
file, "// test 2", "third commit");
}
@After
public void clearGitResources() throws Exception {
testRepo.dispose();
super.tearDown();
}
@Test
public void testSyncLocalAndBranch() throws Exception {
// Note that HEAD is on master
GitResourceVariantTreeSubscriber grvts = createGitResourceVariantTreeSubscriber(
Constants.HEAD, BRANCH, true);
grvts.init(new NullProgressMonitor());
IResourceVariant actualSource = getSourceVariant(grvts, changedFile,
true);
IResourceVariant actualBase = getBaseVariant(grvts, changedFile);
IResourceVariant actualRemote = getRemoteVariant(grvts, changedFile);
assertVariantIsLocal(actualSource, changedFile);
assertVariantMatchCommit(actualBase, initialCommit);
assertVariantMatchCommit(actualRemote, commitBranch);
}
@Test
public void testSyncMasterAndBranch() throws Exception {
GitResourceVariantTreeSubscriber grvts = createGitResourceVariantTreeSubscriber(
MASTER, BRANCH, false);
grvts.init(new NullProgressMonitor());
IResourceVariant actualSource = getSourceVariant(grvts, changedFile,
false);
IResourceVariant actualBase = getBaseVariant(grvts, changedFile);
IResourceVariant actualRemote = getRemoteVariant(grvts, changedFile);
assertVariantMatchCommit(actualSource, commitMaster);
assertVariantMatchCommit(actualBase, initialCommit);
assertVariantMatchCommit(actualRemote, commitBranch);
}
@Test
public void testSyncBranchAndMaster() throws Exception {
GitResourceVariantTreeSubscriber grvts = createGitResourceVariantTreeSubscriber(
BRANCH, MASTER, false);
grvts.init(new NullProgressMonitor());
IResourceVariant actualSource = getSourceVariant(grvts, changedFile,
false);
IResourceVariant actualBase = getBaseVariant(grvts, changedFile);
IResourceVariant actualRemote = getRemoteVariant(grvts, changedFile);
assertVariantMatchCommit(actualSource, commitBranch);
assertVariantMatchCommit(actualBase, initialCommit);
assertVariantMatchCommit(actualRemote, commitMaster);
}
private void assertVariantIsLocal(IResourceVariant variant, IResource local) {
assertTrue(variant instanceof GitLocalResourceVariant);
assertEquals(local, ((GitLocalResourceVariant) variant).getResource());
}
private void assertVariantMatchCommit(IResourceVariant variant,
RevCommit commit) {
assertTrue(variant instanceof GitRemoteResource);
assertEquals(commit, ((GitRemoteResource) variant).getCommitId());
}
private GitResourceVariantTreeSubscriber createGitResourceVariantTreeSubscriber(
String src, String dst, boolean includeLocal) throws IOException {
GitSynchronizeData gsd = new GitSynchronizeData(
testRepo.getRepository(), src, dst, includeLocal);
GitSynchronizeDataSet gsds = new GitSynchronizeDataSet(gsd);
return new GitResourceVariantTreeSubscriber(gsds);
}
private IResourceVariant getSourceVariant(
GitResourceVariantTreeSubscriber subscriber, IResource resource,
boolean includeLocal) throws TeamException {
IResourceVariantTree tree = subscriber.getSourceTree();
assertNotNull(tree);
assertTrue(tree instanceof GitSourceResourceVariantTree);
IResourceVariant resourceVariant = tree.getResourceVariant(resource);
assertNotNull(resourceVariant);
if (includeLocal)
assertTrue(resourceVariant instanceof GitLocalResourceVariant);
else
assertTrue(resourceVariant instanceof GitRemoteResource);
return resourceVariant;
}
private IResourceVariant getBaseVariant(
GitResourceVariantTreeSubscriber subscriber, IResource resource)
throws TeamException {
IResourceVariantTree tree = subscriber.getBaseTree();
assertNotNull(tree);
assertTrue(tree instanceof GitBaseResourceVariantTree);
IResourceVariant resourceVariant = tree
.getResourceVariant(resource);
assertNotNull(resourceVariant);
assertTrue(resourceVariant instanceof GitRemoteResource);
return resourceVariant;
}
private IResourceVariant getRemoteVariant(
GitResourceVariantTreeSubscriber subscriber, IResource resource)
throws TeamException {
IResourceVariantTree tree = subscriber.getRemoteTree();
assertNotNull(tree);
assertTrue(tree instanceof GitRemoteResourceVariantTree);
IResourceVariant resourceVariant = tree.getResourceVariant(resource);
assertNotNull(resourceVariant);
assertTrue(resourceVariant instanceof GitRemoteResource);
return resourceVariant;
}
}