platform/util/testSrc/com/intellij/util/io/zip/ReorderJarsTest.java

/*
 * Copyright 2000-2014 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.util.io.zip;

import com.intellij.openapi.application.PathManager;
import com.intellij.openapi.util.io.FileUtil;
import com.intellij.util.lang.JarMemoryLoader;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import sun.misc.Resource;

import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;

import static org.junit.Assert.*;

/**
 * @author Dmitry Avdeev
 * @since 12/07/2011
 */
public class ReorderJarsTest {
  private File myTempDirectory;

  @Before
  public void setUp() throws Exception {
    myTempDirectory = FileUtil.createTempDirectory("__", "__");
  }

  @After
  public void tearDown() {
    FileUtil.delete(myTempDirectory);
  }

  private static String getTestDataPath() {
    String homePath = PathManager.getHomePath().replace(File.separatorChar, '/');
    if (new File(homePath + "/community/java/java-tests/testData").exists()) return homePath + "/community/java/java-tests/testData";
    return homePath + "/java/java-tests/testData";
  }

  @Test
  public void testReordering() throws IOException {
    String path = getTestDataPath() + "/ide/plugins/reorderJars";
    JBZipFile zipFile = null;
    try {
      zipFile = new JBZipFile(path + "/annotations.jar");
      List<JBZipEntry> entries = zipFile.getEntries();
      System.out.println(entries);
    }
    finally {
      if (zipFile != null) {
        zipFile.close();
      }
    }

    ReorderJarsMain.main(new String[]{path + "/order.txt", path, myTempDirectory.getPath()});

    File[] files = myTempDirectory.listFiles();
    assertNotNull(files);
    assertEquals(1, files.length);
    File file = files[0];
    assertEquals("annotations.jar", file.getName());
    byte[] data;
    try {
      zipFile = new JBZipFile(file);
      List<JBZipEntry> entries = zipFile.getEntries();
      System.out.println(entries);
      assertEquals(JarMemoryLoader.SIZE_ENTRY, entries.get(0).getName());
      JBZipEntry entry = entries.get(1);
      data = entry.getData();
      assertEquals(548, data.length);
      assertEquals("org/jetbrains/annotations/Nullable.class", entry.getName());
      assertEquals("org/jetbrains/annotations/NotNull.class", entries.get(2).getName());
      assertEquals("META-INF/", entries.get(3).getName());
    }
    finally {
      zipFile.close();
    }

    JarMemoryLoader loader = JarMemoryLoader.load(file, file.toURI().toURL());
    assertNotNull(loader);
    Resource resource = loader.getResource("org/jetbrains/annotations/Nullable.class");
    assertNotNull(resource);
    assertEquals(548, resource.getContentLength());
    byte[] bytes = resource.getBytes();
    assertTrue(Arrays.equals(data, bytes));
  }

  @Test
  public void testPluginXml() throws Exception {
    String path = getTestDataPath() + "/ide/plugins/reorderJars";

    ReorderJarsMain.main(new String[] { path + "/zkmOrder.txt", path, myTempDirectory.getPath() } );

    File[] files = myTempDirectory.listFiles();
    assertNotNull(files);
    File file = files[0];
    assertEquals("zkm.jar", file.getName());

    JBZipFile zipFile = new JBZipFile(file);
    try {
      List<JBZipEntry> entries = zipFile.getEntries();
      System.out.println(entries);
      assertEquals(JarMemoryLoader.SIZE_ENTRY, entries.get(0).getName());
      assertEquals("META-INF/plugin.xml", entries.get(1).getName());
    }
    finally {
      zipFile.close();
    }
  }
}