platform/platform-tests/testSrc/com/intellij/history/integration/BasicsTest.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;


import com.intellij.history.LocalHistory;
import com.intellij.history.core.revisions.Revision;
import com.intellij.history.utils.RunnableAdapter;
import com.intellij.openapi.command.CommandProcessor;
import com.intellij.openapi.vfs.JarFileSystem;
import com.intellij.openapi.vfs.LocalFileSystem;
import com.intellij.openapi.vfs.VirtualFile;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarOutputStream;

public class BasicsTest extends IntegrationTestCase {
  public void testProcessingCommands() throws Exception {
    final VirtualFile[] f = new VirtualFile[1];

    CommandProcessor.getInstance().executeCommand(myProject, new RunnableAdapter() {
      @Override
      public void doRun() throws IOException {
        f[0] = myRoot.createChildData(null, "f1.txt");
        f[0].setBinaryContent(new byte[]{1});
        f[0].setBinaryContent(new byte[]{2});
      }
    }, "name", null);

    assertEquals(2, getRevisionsFor(f[0]).size());
  }

  public void testPuttingUserLabel() throws Exception {
    VirtualFile f = myRoot.createChildData(null, "f.txt");

    LocalHistory.getInstance().putUserLabel(myProject, "global");

    assertEquals(3, getRevisionsFor(f).size());
    assertEquals(3, getRevisionsFor(myRoot).size());

    LocalHistory.getInstance().putUserLabel(myProject, "file");

    List<Revision> rr = getRevisionsFor(f);
    assertEquals(4, rr.size());
    assertEquals("file", rr.get(1).getLabel());
    assertEquals(-1, rr.get(1).getLabelColor());
    assertEquals("global", rr.get(2).getLabel());
    assertEquals(-1, rr.get(2).getLabelColor());
  }

  public void testPuttingSystemLabel() throws IOException {
    VirtualFile f = myRoot.createChildData(null, "file.txt");

    assertEquals(2, getRevisionsFor(f).size());
    assertEquals(2, getRevisionsFor(myRoot).size());

    LocalHistory.getInstance().putSystemLabel(myProject, "label");

    List<Revision> rr = getRevisionsFor(f);
    assertEquals(3, rr.size());
    assertEquals("label", rr.get(1).getLabel());

    rr = getRevisionsFor(myRoot);
    assertEquals(3, rr.size());
    assertEquals("label", rr.get(1).getLabel());
  }

  public void testPuttingLabelWithUnsavedDocuments() throws Exception {
    VirtualFile f = myRoot.createChildData(null, "f.txt");
    setContent(f, "1");

    setDocumentTextFor(f, "2");
    LocalHistory.getInstance().putSystemLabel(myProject, "label");

    setDocumentTextFor(f, "3");
    LocalHistory.getInstance().putUserLabel(myProject, "label");

    setDocumentTextFor(f, "4");
    LocalHistory.getInstance().putUserLabel(myProject, "label");

    List<Revision> rr = getRevisionsFor(f);
    assertEquals(9, rr.size()); // 5 changes + 3 labels
    assertEquals("4", new String(rr.get(0).findEntry().getContent().getBytes()));
    assertEquals("4", new String(rr.get(1).findEntry().getContent().getBytes()));
    assertEquals("3", new String(rr.get(2).findEntry().getContent().getBytes()));
    assertEquals("3", new String(rr.get(3).findEntry().getContent().getBytes()));
    assertEquals("2", new String(rr.get(4).findEntry().getContent().getBytes()));
    assertEquals("2", new String(rr.get(5).findEntry().getContent().getBytes()));
    assertEquals("1", new String(rr.get(6).findEntry().getContent().getBytes()));
    assertEquals("", new String(rr.get(7).findEntry().getContent().getBytes()));
  }

  public void testDoNotRegisterSameUnsavedDocumentContentTwice() throws Exception {
    VirtualFile f = myRoot.createChildData(null, "f.txt");
    setContent(f, "1");

    setDocumentTextFor(f, "2");
    LocalHistory.getInstance().putSystemLabel(myProject, "label");
    LocalHistory.getInstance().putUserLabel(myProject, "label");

    List<Revision> rr = getRevisionsFor(f);
    assertEquals(6, rr.size()); // 3 changes + 2 labels
    assertEquals("2", new String(rr.get(0).findEntry().getContent().getBytes()));
    assertEquals("2", new String(rr.get(1).findEntry().getContent().getBytes()));
    assertEquals("2", new String(rr.get(2).findEntry().getContent().getBytes()));
    assertEquals("1", new String(rr.get(3).findEntry().getContent().getBytes()));
    assertEquals("", new String(rr.get(4).findEntry().getContent().getBytes()));
  }

  public void testIsUnderControl() throws Exception {
    VirtualFile f1 = myRoot.createChildData(null, "file.txt");
    VirtualFile f2 = myRoot.createChildData(null, "file.hprof");

    assertTrue(LocalHistory.getInstance().isUnderControl(f1));
    assertFalse(LocalHistory.getInstance().isUnderControl(f2));
  }

  public void testDoNotRegisterChangesNotInLocalFS() throws Exception {
    File f = new File(myRoot.getPath(), "f.jar");
    JarOutputStream jar = new JarOutputStream(new FileOutputStream(f));

    jar.putNextEntry(new JarEntry("file.txt"));
    jar.write(1);
    jar.closeEntry();
    jar.close();

    VirtualFile vfile = LocalFileSystem.getInstance().refreshAndFindFileByIoFile(f);
    assertNotNull(vfile);

    VirtualFile jarRoot = JarFileSystem.getInstance().getJarRootForLocalFile(vfile);
    assertEquals(1, jarRoot.findChild("file.txt").contentsToByteArray()[0]);

    assertEquals(2, getRevisionsFor(myRoot).size());

    jar = new JarOutputStream(new FileOutputStream(f));

    JarEntry e = new JarEntry("file.txt");
    e.setTime(f.lastModified() + 10000);
    jar.putNextEntry(e);
    jar.write(2);
    jar.closeEntry();
    jar.close();
    f.setLastModified(f.lastModified() + 10000);

    LocalFileSystem.getInstance().refreshWithoutFileWatcher(false);
    JarFileSystem.getInstance().refreshWithoutFileWatcher(false);
    jarRoot = JarFileSystem.getInstance().getJarRootForLocalFile(vfile);
    assertEquals(2, jarRoot.findChild("file.txt").contentsToByteArray()[0]);

    assertEquals(2, getRevisionsFor(myRoot).size());
    assertEquals(1, getRevisionsFor(jarRoot).size());
  }
}