1 /*

     2  * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 package jdk.nashorn.internal.runtime;

    26 

    27 import static org.testng.Assert.assertEquals;

    28 import static org.testng.Assert.assertFalse;

    29 import java.io.File;

    30 import java.io.IOException;

    31 import java.nio.file.DirectoryStream;

    32 import java.nio.file.FileSystems;

    33 import java.nio.file.Files;

    34 import java.nio.file.Path;

    35 import javax.script.ScriptEngine;

    36 import javax.script.ScriptException;

    37 import jdk.nashorn.api.scripting.NashornScriptEngineFactory;

    38 import org.testng.annotations.Test;

    39 

    40 /**

    41  * @test

    42  * @bug 8039185 8039403

    43  * @summary  Test for persistent code cache and path handling

    44  * @run testng jdk.nashorn.internal.runtime.CodeStoreAndPathTest

    45  */

    46 @SuppressWarnings("javadoc")

    47 public class CodeStoreAndPathTest {

    48 

    49     final String code1 = "var code1; var x = 'Hello Script'; var x1 = 'Hello Script'; "

    50                 + "var x2 = 'Hello Script'; var x3 = 'Hello Script'; "

    51                 + "var x4 = 'Hello Script'; var x5 = 'Hello Script';"

    52                 + "var x6 = 'Hello Script'; var x7 = 'Hello Script'; "

    53                 + "var x8 = 'Hello Script'; var x9 = 'Hello Script'; "

    54                 + "var x10 = 'Hello Script';"

    55                 + "function f() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    56                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    57                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    58                 + "var x10 = 'Hello Script';}"

    59                 + "function g() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    60                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    61                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    62                 + "var x10 = 'Hello Script';}"

    63                 + "function h() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    64                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    65                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    66                 + "var x10 = 'Hello Script';}"

    67                 + "function i() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    68                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    69                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    70                 + "var x10 = 'Hello Script';}";

    71     final String code2 = "var code2; var x = 'Hello Script'; var x1 = 'Hello Script'; "

    72                 + "var x2 = 'Hello Script'; var x3 = 'Hello Script'; "

    73                 + "var x4 = 'Hello Script'; var x5 = 'Hello Script';"

    74                 + "var x6 = 'Hello Script'; var x7 = 'Hello Script'; "

    75                 + "var x8 = 'Hello Script'; var x9 = 'Hello Script'; "

    76                 + "var x10 = 'Hello Script';"

    77                 + "function f() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    78                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    79                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    80                 + "var x10 = 'Hello Script';}"

    81                 + "function g() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    82                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    83                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    84                 + "var x10 = 'Hello Script';}"

    85                 + "function h() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    86                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    87                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    88                 + "var x10 = 'Hello Script';}"

    89                 + "function i() {x ='Bye Script'; x1 ='Bye Script'; x2='Bye Script';"

    90                 + "x3='Bye Script'; x4='Bye Script'; x5='Bye Script'; x6='Bye Script';"

    91                 + "x7='Bye Script'; x8='Bye Script'; var x9 = 'Hello Script'; "

    92                 + "var x10 = 'Hello Script';}";

    93     // Script size < Default minimum size for storing a compiled script class

    94     final String code3 = "var code3; var x = 'Hello Script'; var x1 = 'Hello Script'; ";

    95     final String codeCache = "build/nashorn_code_cache";

    96     final String oldUserDir = System.getProperty("user.dir");

    97 

    98     private static final String[] ENGINE_OPTIONS_OPT   = new String[]{"--persistent-code-cache", "--optimistic-types=true"};

    99     private static final String[] ENGINE_OPTIONS_NOOPT = new String[]{"--persistent-code-cache", "--optimistic-types=false"};

   100 

   101     @Test

   102     public void pathHandlingTest() {

   103         System.setProperty("nashorn.persistent.code.cache", codeCache);

   104         final NashornScriptEngineFactory fac = new NashornScriptEngineFactory();

   105 

   106         fac.getScriptEngine(ENGINE_OPTIONS_NOOPT);

   107 

   108         final Path expectedCodeCachePath = FileSystems.getDefault().getPath(oldUserDir + File.separator + codeCache);

   109         final Path actualCodeCachePath = FileSystems.getDefault().getPath(System.getProperty(

   110                             "nashorn.persistent.code.cache")).toAbsolutePath();

   111         // Check that nashorn code cache is created in current working directory

   112         assertEquals(actualCodeCachePath, expectedCodeCachePath);

   113         // Check that code cache dir exists and it's not empty

   114         final File file = new File(actualCodeCachePath.toUri());

   115         assertFalse(!file.isDirectory(), "No code cache directory was created!");

   116         assertFalse(file.list().length == 0, "Code cache directory is empty!");

   117     }

   118 

   119     @Test

   120     public void changeUserDirTest() throws ScriptException, IOException {

   121         System.setProperty("nashorn.persistent.code.cache", codeCache);

   122         final NashornScriptEngineFactory fac = new NashornScriptEngineFactory();

   123         final ScriptEngine e = fac.getScriptEngine(ENGINE_OPTIONS_NOOPT);

   124         final Path codeCachePath = getCodeCachePath(false);

   125         final String newUserDir = "build/newUserDir";

   126         // Now changing current working directory

   127         System.setProperty("user.dir", System.getProperty("user.dir") + File.separator + newUserDir);

   128         try {

   129             // Check that a new compiled script is stored in existing code cache

   130             e.eval(code1);

   131             final DirectoryStream<Path> stream = Files.newDirectoryStream(codeCachePath);

   132             checkCompiledScripts(stream, 1);

   133             // Setting to default current working dir

   134         } finally {

   135             System.setProperty("user.dir", oldUserDir);

   136         }

   137     }

   138 

   139     @Test

   140     public void codeCacheTest() throws ScriptException, IOException {

   141         System.setProperty("nashorn.persistent.code.cache", codeCache);

   142         final NashornScriptEngineFactory fac = new NashornScriptEngineFactory();

   143         final ScriptEngine e = fac.getScriptEngine(ENGINE_OPTIONS_NOOPT);

   144         final Path codeCachePath = getCodeCachePath(false);

   145         e.eval(code1);

   146         e.eval(code2);

   147         e.eval(code3);// less than minimum size for storing

   148         // adding code1 and code2.

   149         final DirectoryStream<Path> stream = Files.newDirectoryStream(codeCachePath);

   150         checkCompiledScripts(stream, 2);

   151     }

   152 

   153     @Test

   154     public void codeCacheTestOpt() throws ScriptException, IOException {

   155         System.setProperty("nashorn.persistent.code.cache", codeCache);

   156         final NashornScriptEngineFactory fac = new NashornScriptEngineFactory();

   157         final ScriptEngine e = fac.getScriptEngine(ENGINE_OPTIONS_OPT);

   158         final Path codeCachePath = getCodeCachePath(true);

   159         e.eval(code1);

   160         e.eval(code2);

   161         e.eval(code3);// less than minimum size for storing

   162         // adding code1 and code2.

   163         final DirectoryStream<Path> stream = Files.newDirectoryStream(codeCachePath);

   164         checkCompiledScripts(stream, 2);

   165     }

   166 

   167     private static Path getCodeCachePath(final boolean optimistic) {

   168         final String codeCache = System.getProperty("nashorn.persistent.code.cache");

   169         final Path codeCachePath = FileSystems.getDefault().getPath(codeCache).toAbsolutePath();

   170         final String[] files = codeCachePath.toFile().list();

   171         for (final String file : files) {

   172             if (file.endsWith("_opt") == optimistic) {

   173                 return codeCachePath.resolve(file);

   174             }

   175         }

   176         throw new AssertionError("Code cache path not found");

   177     }

   178 

   179     private static void checkCompiledScripts(final DirectoryStream<Path> stream, final int numberOfScripts) throws IOException {

   180         int n = numberOfScripts;

   181         for (@SuppressWarnings("unused") final Path file : stream) {

   182             n--;

   183         }

   184         stream.close();

   185         assertEquals(n, 0);

   186     }

   187 

   188 }