platform/platform-tests/testSrc/com/intellij/history/integration/ui/models/SelectionCalculatorTest.java

/*
 * Copyright 2000-2009 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.history.integration.ui.models;

import com.intellij.diff.Block;
import com.intellij.history.core.InMemoryLocalHistoryFacade;
import com.intellij.history.core.LocalHistoryFacade;
import com.intellij.history.core.LocalHistoryTestCase;
import com.intellij.history.core.revisions.Revision;
import com.intellij.history.core.tree.RootEntry;
import com.intellij.history.integration.IdeaGateway;
import com.intellij.util.diff.FilesTooBigForDiffException;
import org.jetbrains.annotations.NotNull;
import org.junit.Test;

import java.util.List;

import static org.easymock.classextension.EasyMock.*;

public class SelectionCalculatorTest extends LocalHistoryTestCase {
  IdeaGateway gw = new MyIdeaGateway();
  LocalHistoryFacade vcs = new InMemoryLocalHistoryFacade();

  @Test
  public void testSelectionWasNotChanged() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("abc\ndef\nghi", "abc1\ndef1\nghi1");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 2);

    Block b0 = c.getSelectionFor(rr.get(0), new NullProgress());
    Block b1 = c.getSelectionFor(rr.get(1), new NullProgress());

    assertBlock(0, 2, "abc1\ndef1\nghi1", b0);
    assertBlock(0, 2, "abc\ndef\nghi", b1);
  }

  @Test
  public void testSelectionWasMoved() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("abc\ndef\nghi", "def\nghi");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 1);

    Block b0 = c.getSelectionFor(rr.get(0), new NullProgress());
    Block b1 = c.getSelectionFor(rr.get(1), new NullProgress());

    assertBlock(0, 1, "def\nghi", b0);
    assertBlock(1, 2, "def\nghi", b1);
  }

  @Test
  public void testSelectionForVeryOldRevisionTakenBackward() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("ghi\nabc\ndef", "abc\nghi\ndef", "abc\ndef\nghi");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 1);

    Block b2 = c.getSelectionFor(rr.get(2), new NullProgress());
    Block b1 = c.getSelectionFor(rr.get(1), new NullProgress());
    Block b0 = c.getSelectionFor(rr.get(0), new NullProgress());

    assertBlock(0, 1, "abc\ndef", b0);
    assertBlock(0, 2, "abc\nghi\ndef", b1);
    assertBlock(1, 2, "abc\ndef", b2);
  }

  @Test
  public void testNormalizingLineEnds() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("abc\ndef\nghi", "abc\r\ndef\r\nghi");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 1);

    Block b0 = c.getSelectionFor(rr.get(0), new NullProgress());
    Block b1 = c.getSelectionFor(rr.get(1), new NullProgress());

    assertBlock(0, 1, "abc\ndef", b0);
    assertBlock(0, 1, "abc\ndef", b1);
  }

  @Test
  public void testProgressOnGetSelection() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("one", "two", "three", "four");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 0);

    Progress p = createStrictMock(Progress.class);
    p.processed(25);
    p.processed(50);
    p.processed(75);
    p.processed(100);
    replay(p);

    c.getSelectionFor(rr.get(3), p);

    verify(p);
  }

  @Test
  public void testProgressOnCanCalculate() throws FilesTooBigForDiffException {
    List<Revision> rr = createRevisions("one", "two");
    SelectionCalculator c = new SelectionCalculator(gw, rr, 0, 0);

    Progress p = createMock(Progress.class);
    p.processed(50);
    p.processed(100);
    replay(p);

    c.canCalculateFor(rr.get(1), p);

    verify(p);
  }

  private List<Revision> createRevisions(String... contents) {
    RootEntry r = new RootEntry();
    vcs.addChangeInTests(createFile(r, "f", contents[0], -1, false));
    for (int i = 1; i < contents.length; i++) {
      vcs.addChangeInTests(changeContent(r, "f", contents[i], i));
    }
    return collectRevisions(vcs, r, "f", null, null);
  }

  private void assertBlock(int from, int to, String content, Block b) {
    assertEquals(from, b.getStart());
    assertEquals(to, b.getEnd());
    assertEquals(content, b.getBlockContent());
  }

  private static class MyIdeaGateway extends IdeaGateway {
    @Override
    public String stringFromBytes(@NotNull byte[] bytes, @NotNull String path) {
      return new String(bytes);
    }
  }
}