1 files changed, 101 insertions, 0 deletions
diff --git a/org.eclipse.egit.ui.test/src/org/eclipse/egit/ui/submodule/SubmoduleUpdateTest.java b/org.eclipse.egit.ui.test/src/org/eclipse/egit/ui/submodule/SubmoduleUpdateTest.java
new file mode 100644
index 0000000000..4f9dd9fe49
--- /dev/null
+++ b/org.eclipse.egit.ui.test/src/org/eclipse/egit/ui/submodule/SubmoduleUpdateTest.java
@@ -0,0 +1,101 @@
+/******************************************************************************
+ *  Copyright (c) 2012 GitHub Inc.
+ *  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
+ *
+ *  Contributors:
+ *    Kevin Sawicki (GitHub Inc.) - initial API and implementation
+ *****************************************************************************/
+package org.eclipse.egit.ui.submodule;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
+
+import java.io.File;
+
+import org.eclipse.egit.ui.Activator;
+import org.eclipse.egit.ui.JobFamilies;
+import org.eclipse.egit.ui.UIText;
+import org.eclipse.egit.ui.test.ContextMenuHelper;
+import org.eclipse.egit.ui.test.TestUtil;
+import org.eclipse.egit.ui.view.repositories.GitRepositoriesViewTestBase;
+import org.eclipse.jgit.api.SubmoduleAddCommand;
+import org.eclipse.jgit.lib.Constants;
+import org.eclipse.jgit.lib.ObjectId;
+import org.eclipse.jgit.lib.Ref;
+import org.eclipse.jgit.lib.Repository;
+import org.eclipse.jgit.storage.file.FileRepository;
+import org.eclipse.jgit.transport.URIish;
+import org.eclipse.swtbot.swt.finder.junit.SWTBotJunit4ClassRunner;
+import org.eclipse.swtbot.swt.finder.widgets.SWTBotTree;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+
+/**
+ * Unit tests for running a submodule update
+ */
+@RunWith(SWTBotJunit4ClassRunner.class)
+public class SubmoduleUpdateTest extends GitRepositoriesViewTestBase {
+
+	private static final String UPDATE_SUBMODULE_CONTEXT_MENU_LABEL = "SubmoduleUpdateCommand.label";
+
+	private static File repositoryFile;
+
+	@Before
+	public void before() throws Exception {
+		repositoryFile = createProjectAndCommitToRepository();
+	}
+
+	@Test
+	public void updateSubmodule() throws Exception {
+		deleteAllProjects();
+		assertProjectExistence(PROJ1, false);
+		clearView();
+		Activator.getDefault().getRepositoryUtil()
+				.addConfiguredRepository(repositoryFile);
+		shareProjects(repositoryFile);
+		assertProjectExistence(PROJ1, true);
+		refreshAndWait();
+		assertHasRepo(repositoryFile);
+		FileRepository repo = lookupRepository(repositoryFile);
+		ObjectId repoHead = repo.resolve(Constants.HEAD);
+
+		SubmoduleAddCommand command = new SubmoduleAddCommand(repo);
+		String path = "sub";
+		command.setPath(path);
+		String uri = new URIish(repo.getDirectory().toURI().toString())
+				.toString();
+		command.setURI(uri);
+		Repository subRepo = command.call();
+		assertNotNull(subRepo);
+
+		Ref head = subRepo.getRef(Constants.HEAD);
+		assertNotNull(head);
+		assertTrue(head.isSymbolic());
+		assertEquals(Constants.R_HEADS + Constants.MASTER, head.getLeaf()
+				.getName());
+		assertEquals(repoHead, head.getObjectId());
+
+		refreshAndWait();
+		SWTBotTree tree = getOrOpenView().bot().tree();
+		tree.getAllItems()[0]
+				.expand()
+				.expandNode(
+						UIText.RepositoriesViewLabelProvider_SubmodulesNodeText)
+				.select();
+		ContextMenuHelper.clickContextMenu(tree, myUtil
+				.getPluginLocalizedValue(UPDATE_SUBMODULE_CONTEXT_MENU_LABEL));
+		TestUtil.joinJobs(JobFamilies.SUBMODULE_UPDATE);
+		refreshAndWait();
+
+		head = subRepo.getRef(Constants.HEAD);
+		assertNotNull(head);
+		assertFalse(head.isSymbolic());
+		assertEquals(repoHead, head.getObjectId());
+	}
+}