VMOptionsTest.java

/*
 * Copyright 2000-2012 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.diagnostic;

import com.intellij.openapi.util.io.FileUtil;
import com.intellij.testFramework.UsefulTestCase;

import java.io.File;
import java.io.IOException;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class VMOptionsTest extends UsefulTestCase {
  private File myFile;

  @Override
  protected void setUp() throws Exception {
    super.setUp();

    myFile = FileUtil.createTempFile("vmoptions.", ".txt");
    writeFile("-Xmx512m");
    VMOptions.setTestFile(myFile.getPath());
  }

  @Override
  protected void tearDown() throws Exception {
    try {
      VMOptions.clearTestFile();
      //noinspection ResultOfMethodCallIgnored
      myFile.delete();
    }
    finally {
      super.tearDown();
    }
  }

  public void testRegExpr() throws Exception {
    Pattern p = Pattern.compile("-Xmx(\\d*)([a-zA-Z]*)");

    Matcher m = p.matcher("-option -Xmx128mb -option");
    assertTrue(m.find());

    assertEquals("128", m.group(1));
    assertEquals("mb", m.group(2));

    m = p.matcher("-option -Xmx -option");
    assertTrue(m.find());

    assertEquals("", m.group(1));
    assertEquals("", m.group(2));

    m = p.matcher("-option -Xxx -option");
    assertFalse(m.find());
  }

  public void testMacOsRegExpr() throws Exception {
    Pattern p = Pattern.compile("<key>" + VMOptions.MAC_ARCH_VM_OPTIONS + "</key>\\s*<string>(.*)</string>");

    Matcher m = p.matcher("<plist version=\"1.0\">\n" +
                          "  <dict>\n" +
                          "    <dict>\n" +
                          "      <key>" + VMOptions.MAC_ARCH_VM_OPTIONS + "</key>\n" +
                          "      <string>-option -Xmx128mb -option</string>\n" +
                          "   </dict>\n" +
                          "  </dict>\n" +
                          "</plist>");
    assertTrue(m.find());

    assertEquals("-option -Xmx128mb -option", m.group(1));
  }

  public void testReading() throws Exception {
    assertEquals(512, VMOptions.readXmx());
  }

  public void testReadingEmpty() throws Exception {
    writeFile("");

    assertEquals(-1, VMOptions.readXmx());
  }

  public void testReadingKilos() throws Exception {
    writeFile("-Xmx512000k\n" + "-XX:ReservedCodeCacheSize=256000K");

    assertEquals(512000 / 1024, VMOptions.readXmx());
    assertEquals(256000 / 1024, VMOptions.readCodeCache());
  }

  public void testReadingGigs() throws Exception {
    writeFile("-Xmx512g\n");

    assertEquals(512 * 1024, VMOptions.readXmx());
  }

  public void testReadingWithoutUnit() throws Exception {
    writeFile("-Xmx512");

    assertEquals(512, VMOptions.readXmx());
  }

  public void testWriting() throws Exception {
    VMOptions.writeXmx(1024);

    assertEquals("-Xmx1024m\n", readFile());
  }

  public void testWritingPreservingLocation() throws Exception {
    writeFile("-someOption\n" +
              "-Xmx512m\n" +
              "-anotherOption");

    VMOptions.writeXmx(1024);

    assertEquals("-someOption\n" +
                 "-Xmx1024m\n" +
                 "-anotherOption", readFile());
  }

  public void testWritingNew() throws Exception {
    writeFile("-someOption");

    VMOptions.writeXmx(1024);
    VMOptions.writeCodeCache(256);

    assertEquals("-someOption -Xmx1024m -XX:ReservedCodeCacheSize=256m", readFile());
  }

  public void testWritingReadOnlyFile() throws Exception {
    FileUtil.setReadOnlyAttribute(myFile.getPath(), true);

    VMOptions.writeXmx(1024);

    assertEquals("-Xmx1024m", readFile());
  }


  private String readFile() throws IOException {
    return FileUtil.loadFile(myFile);
  }

  private void writeFile(String content) throws IOException {
    FileUtil.writeToFile(myFile, content.getBytes());
  }
}