javatests/com/google/gerrit/server/edit/tree/TreeCreatorTest.java

// Copyright (C) 2020 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package com.google.gerrit.server.edit.tree;

import static com.google.common.truth.Truth.assertThat;
import static com.google.gerrit.testing.GerritJUnit.assertThrows;
import static java.nio.charset.StandardCharsets.UTF_8;
import static org.eclipse.jgit.lib.Constants.OBJ_BLOB;

import com.google.common.collect.ImmutableList;
import com.google.gerrit.common.RawInputUtil;
import org.eclipse.jgit.internal.storage.dfs.DfsRepositoryDescription;
import org.eclipse.jgit.internal.storage.dfs.InMemoryRepository;
import org.eclipse.jgit.junit.TestRepository;
import org.eclipse.jgit.lib.ObjectId;
import org.eclipse.jgit.lib.ObjectReader;
import org.eclipse.jgit.lib.Repository;
import org.eclipse.jgit.revwalk.RevObject;
import org.eclipse.jgit.revwalk.RevTree;
import org.eclipse.jgit.revwalk.RevWalk;
import org.eclipse.jgit.treewalk.TreeWalk;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class TreeCreatorTest {

  private Repository repository;
  private TestRepository<?> testRepository;

  @Before
  public void setUp() throws Exception {
    repository = new InMemoryRepository(new DfsRepositoryDescription("Test Repository"));
    testRepository = new TestRepository<>(repository);
  }

  @After
  public void tearDown() throws Exception {
    if (testRepository != null) {
      testRepository.close();
    }
  }

  @Test
  public void fileContentModificationWorksWithEmptyTree() throws Exception {
    TreeCreator treeCreator = TreeCreator.basedOnEmptyTree();
    treeCreator.addTreeModifications(
        ImmutableList.of(
            new ChangeFileContentModification("file.txt", RawInputUtil.create("Line 1"))));
    ObjectId newTreeId = treeCreator.createNewTreeAndGetId(repository);

    String fileContent = getFileContent(newTreeId, "file.txt");
    assertThat(fileContent).isEqualTo("Line 1");
  }

  @Test
  public void renameFileModificationDoesNotComplainAboutEmptyTree() throws Exception {
    TreeCreator treeCreator = TreeCreator.basedOnEmptyTree();
    treeCreator.addTreeModifications(
        ImmutableList.of(new RenameFileModification("oldfileName", "newFileName")));
    ObjectId newTreeId = treeCreator.createNewTreeAndGetId(repository);

    assertThat(isEmptyTree(newTreeId)).isTrue();
  }

  @Test
  public void deleteFileModificationDoesNotComplainAboutEmptyTree() throws Exception {
    TreeCreator treeCreator = TreeCreator.basedOnEmptyTree();
    treeCreator.addTreeModifications(ImmutableList.of(new DeleteFileModification("file.txt")));
    ObjectId newTreeId = treeCreator.createNewTreeAndGetId(repository);

    assertThat(isEmptyTree(newTreeId)).isTrue();
  }

  @Test
  public void restoreFileModificationDoesNotComplainAboutEmptyTree() throws Exception {
    TreeCreator treeCreator = TreeCreator.basedOnEmptyTree();
    treeCreator.addTreeModifications(ImmutableList.of(new RestoreFileModification("file.txt")));
    ObjectId newTreeId = treeCreator.createNewTreeAndGetId(repository);

    assertThat(isEmptyTree(newTreeId)).isTrue();
  }

  @Test
  public void modificationsMustNotReferToSameFilePaths() {
    TreeCreator treeCreator = TreeCreator.basedOnEmptyTree();
    treeCreator.addTreeModifications(
        ImmutableList.of(
            new RenameFileModification("oldFileName", "newFileName"),
            new ChangeFileContentModification(
                "newFileName", RawInputUtil.create("Different content"))));
    IllegalStateException exception =
        assertThrows(
            IllegalStateException.class, () -> treeCreator.createNewTreeAndGetId(repository));

    assertThat(exception).hasMessageThat().contains("oldFileName");
    assertThat(exception).hasMessageThat().contains("newFileName");
  }

  @Test
  public void fileContentModificationRefersToModifiedFile() {
    ChangeFileContentModification contentModification =
        new ChangeFileContentModification("myFileName", RawInputUtil.create("Some content"));
    assertThat(contentModification.getFilePaths()).containsExactly("myFileName");
  }

  @Test
  public void renameFileModificationRefersToOldAndNewFilePath() {
    RenameFileModification fileModification =
        new RenameFileModification("oldFileName", "newFileName");
    assertThat(fileModification.getFilePaths()).containsExactly("oldFileName", "newFileName");
  }

  @Test
  public void deleteFileModificationRefersToDeletedFile() {
    DeleteFileModification fileModification = new DeleteFileModification("myFileName");
    assertThat(fileModification.getFilePaths()).containsExactly("myFileName");
  }

  @Test
  public void restoreFileModificationRefersToRestoredFile() {
    RestoreFileModification fileModification = new RestoreFileModification("myFileName");
    assertThat(fileModification.getFilePaths()).containsExactly("myFileName");
  }

  private String getFileContent(ObjectId treeId, String filePath) throws Exception {
    try (RevWalk revWalk = new RevWalk(repository);
        ObjectReader reader = revWalk.getObjectReader()) {
      RevTree revTree = revWalk.parseTree(treeId);
      RevObject revObject = testRepository.get(revTree, filePath);
      return new String(reader.open(revObject, OBJ_BLOB).getBytes(), UTF_8);
    }
  }

  private boolean isEmptyTree(ObjectId treeId) throws Exception {
    try (TreeWalk treeWalk = new TreeWalk(repository)) {
      treeWalk.reset(treeId);
      return !treeWalk.next();
    }
  }
}