PatchSetDetailRemoteFactoryTest.java

/*******************************************************************************
 * Copyright (c) 2014, 2015 Tasktop Technologies 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
 *
 * Contributors:
 *     Tasktop Technologies - initial API and implementation
 *******************************************************************************/

package org.eclipse.mylyn.internal.gerrit.core.remote;

import static org.eclipse.mylyn.internal.gerrit.core.remote.TestRemoteObserverConsumer.retrieveForLocalKey;
import static org.hamcrest.Matchers.is;
import static org.hamcrest.Matchers.notNullValue;
import static org.junit.Assert.assertThat;

import java.util.List;

import junit.framework.TestCase;

import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.jgit.api.CommitCommand;
import org.eclipse.mylyn.commons.sdk.util.CommonTestUtil.PrivilegeLevel;
import org.eclipse.mylyn.gerrit.tests.support.GerritFixture;
import org.eclipse.mylyn.internal.gerrit.core.client.GerritException;
import org.eclipse.mylyn.internal.gerrit.core.client.compat.ChangeDetailX;
import org.eclipse.mylyn.internal.gerrit.core.client.compat.PatchSetPublishDetailX;
import org.eclipse.mylyn.reviews.core.model.IReview;
import org.eclipse.mylyn.reviews.core.model.IReviewItemSet;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import com.google.common.collect.ImmutableList;
import com.google.gerrit.common.data.PatchSetDetail;
import com.google.gerrit.reviewdb.Change;
import com.google.gerrit.reviewdb.PatchSet;

public class PatchSetDetailRemoteFactoryTest extends TestCase {

	private static final String NON_DRAFT_BRANCH = "HEAD:refs/for/master";

	private static final String DRAFT_BRANCH = "HEAD:refs/drafts/master";

	private ReviewHarness reviewHarness;

	@Override
	@Before
	public void setUp() throws Exception {
		// sets who is signed-in to view the review (performs the retrieval)
		reviewHarness = new ReviewHarness();
		// set who makes the initial commit (and consequentially, becomes the review owner)
		reviewHarness.init(DRAFT_BRANCH, PrivilegeLevel.ADMIN, "testFile1.txt", false);
	}

	@Override
	@After
	public void tearDown() throws Exception {
		reviewHarness.dispose();
	}

	@Test
	public void testUserHasNoAccessToAdminDraft() throws Exception {
		createPatchSet(NON_DRAFT_BRANCH, PrivilegeLevel.ADMIN, ImmutableList.of("testFile2.txt", "testFile3.txt"));

		reviewHarness.retrieve();
		assertThat(reviewHarness.getReview().getSets().size(), is(1));
		assertThat(reviewHarness.getReview().getSets().get(0).getId(), is("2"));

		assertNull(retrievePatchSetDetail("1"));
		PatchSetDetail detail = retrievePatchSetDetail("2");
		assertThat(detail.getInfo().getKey().get(), is(2));
	}

	@Test
	public void testUserHasAccessToAdminDraft() throws Exception {
		createPatchSet(NON_DRAFT_BRANCH, PrivilegeLevel.ADMIN, ImmutableList.of("testFile2.txt", "testFile3.txt"));
		reviewHarness.getClient().addReviewers(reviewHarness.getShortId(),
				ImmutableList.of(GerritFixture.current().getCredentials(PrivilegeLevel.USER).getUserName()),
				new NullProgressMonitor());

		reviewHarness.retrieve();
		assertThat(reviewHarness.getReview().getSets().size(), is(2));
		assertThat(reviewHarness.getReview().getSets().get(0).getId(), is("1"));
		assertThat(reviewHarness.getReview().getSets().get(1).getId(), is("2"));

		PatchSetDetail detail = retrievePatchSetDetail("1");
		assertThat(detail.getInfo().getKey().get(), is(1));
		detail = retrievePatchSetDetail("2");
		assertThat(detail.getInfo().getKey().get(), is(2));

	}

	@Test
	public void testGetPatchSetPublishDetailOfDraftIffAdmin() throws Exception {
		int reviewId = Integer.parseInt(reviewHarness.getShortId());
		PatchSet.Id id = new PatchSet.Id(new Change.Id(reviewId), 1);
		PatchSetPublishDetailX patchSetDetail;
		try {
			patchSetDetail = reviewHarness.getClient().getPatchSetPublishDetail(id, new NullProgressMonitor());
			fail("Expected Gerrit Exception");
		} catch (GerritException e) {
		} // Needs admin client to view admin-created draft
		patchSetDetail = reviewHarness.getAdminClient().getPatchSetPublishDetail(id, new NullProgressMonitor());

		assertThat(patchSetDetail, notNullValue());
		// DRAFT is not correctly parsed for ChangeInfo since Change.Status does not define the corresponding enum field
		assertThat(patchSetDetail.getChange().getStatus(), is(Change.Status.NEW));
	}

	@Test
	public void testGetChangeDetailOfDraftIffAdmin() throws Exception {
		int reviewId = Integer.parseInt(reviewHarness.getShortId());
		ChangeDetailX changeDetail;
		try {
			changeDetail = reviewHarness.getClient().getChangeDetail(reviewId, new NullProgressMonitor());
			fail("Expected Gerrit Exception");
		} catch (GerritException e) {
		} // Needs admin client to view admin-created draft
		changeDetail = reviewHarness.getAdminClient().getChangeDetail(reviewId, new NullProgressMonitor());

		assertThat(changeDetail, notNullValue());
		// DRAFT is not correctly parsed for ChangeInfo since Change.Status does not define the corresponding enum field
		assertThat(changeDetail.getChange().getStatus(), is(Change.Status.NEW));
	}

	private void createPatchSet(String pushTo, PrivilegeLevel privilegeLevel, List<String> files) throws Exception {
		CommitCommand command = reviewHarness.createCommitCommand();
		for (String fileName : files) {
			reviewHarness.addFile(fileName);
		}
		reviewHarness.commitAndPush(command, pushTo, privilegeLevel);
	}

	private PatchSetDetail retrievePatchSetDetail(String patchSetId) {
		TestRemoteObserverConsumer<IReview, IReviewItemSet, String, PatchSetDetail, PatchSetDetail, String> itemSetObserver //
		= retrieveForLocalKey(reviewHarness.getProvider().getReviewItemSetFactory(), reviewHarness.getReview(),
				patchSetId, false);
		PatchSetDetail detail = itemSetObserver.getRemoteObject();
		return detail;
	}
}