ApplyPatchTest.java

/*
 * Copyright 2000-2017 JetBrains s.r.o.
 *
 * 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.intellij.openapi.diff.impl.patch;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.diff.impl.patch.formove.PatchApplier;
import com.intellij.openapi.fileTypes.ExtensionFileNameMatcher;
import com.intellij.openapi.fileTypes.FileTypeManager;
import com.intellij.openapi.fileTypes.FileTypes;
import com.intellij.openapi.vcs.changes.patch.ApplyPatchAction;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.openapi.vfs.VirtualFileFilter;
import com.intellij.testFramework.PlatformTestCase;
import com.intellij.testFramework.PlatformTestUtil;
import com.intellij.testFramework.PsiTestUtil;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class ApplyPatchTest extends PlatformTestCase {
  public void testAddLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testAddLastLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testOverlappingContext() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testAddFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testRemoveFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testMatchByContext() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testMultiFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testContextDiff() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testContextDiffAddLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testContextDiffRemoveLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testContextDiffMultiFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testEmptyLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testReversedNames() throws Exception {
    doTest(0, ApplyPatchStatus.SUCCESS, null);
  }

  public void testAlreadyApplied() throws Exception {
    doTest(1, ApplyPatchStatus.ALREADY_APPLIED, null);
  }

  public void testPartialApply() throws Exception {
    doTest(1, ApplyPatchStatus.PARTIAL, null);
  }

  public void testContextDiffSingleSpace() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testNoNewlineAtEof() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testContextNoNewlineAtEof() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testRenameFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testRenameDir() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, new VirtualFileFilter() {
      @Override
      public boolean accept(final VirtualFile file) {
        return !"empty".equals(file.getNameWithoutExtension());
      }
    });
  }

  public void testDeleteLastLineWithLineBreak() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testDeleteLineContentWithoutLineBreak() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testDeleteLastLineWithoutLineBreak() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyFileNoHunkAtEOF() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyFileRemoveLastEmptyLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyFileAddLastEmptyLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyFileLastLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyFileKeepLastEmptyLine() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testMoveFile() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testMoveFileWithoutRename() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testMoveAndRenameWithNameConflicts() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testIncorrectAlreadyAppliedDetection() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testOmittedChunkSize() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testWrongFileStartUnified() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testAddFileWithGitVersion() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testModifyLineWithGitVersion() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  public void testAddFileWithoutNewlineAtEOF() throws Exception {
    doTest(1, ApplyPatchStatus.SUCCESS, null);
  }

  private void doTest(final int skipTopDirs, final ApplyPatchStatus expectedStatus, final VirtualFileFilter fileFilter) throws Exception {
    ApplicationManager.getApplication()
      .runWriteAction(() -> FileTypeManager.getInstance().associate(FileTypes.PLAIN_TEXT, new ExtensionFileNameMatcher("old")));

    String testDataPath = PlatformTestUtil.getPlatformTestDataPath() + "diff/applyPatch/" + getTestName(true);
    String beforePath = testDataPath + "/before";
    String afterPath = testDataPath + "/after";
    VirtualFile afterDir = LocalFileSystem.getInstance().refreshAndFindFileByPath(afterPath.replace(File.separatorChar, '/'));

    VirtualFile patchedDir = PsiTestUtil.createTestProjectStructure(myProject, myModule, beforePath, myFilesToDelete);

    String patchPath = testDataPath + "/apply.patch";
    VirtualFile patchFile = LocalFileSystem.getInstance().refreshAndFindFileByPath(patchPath.replace(File.separatorChar, '/'));

    PatchReader reader = PatchVirtualFileReader.create(patchFile);
    List<FilePatch> patches = new ArrayList<>(reader.readTextPatches());

    ApplyPatchAction.applySkipDirs(patches, skipTopDirs);
    final PatchApplier.ApplyPatchTask applyPart =
      new PatchApplier<BinaryFilePatch>(myProject, patchedDir, patches, null, null, null).createApplyPart(false, false);
    applyPart.run();

    assertEquals(expectedStatus, applyPart.getStatus());

    PlatformTestUtil.assertDirectoriesEqual(patchedDir, afterDir, fileFilter);
  }
}