test/langtools/jdk/jshell/HistoryTest.java

/*
 * Copyright (c) 2015, 2018, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

/*
 * @test
 * @bug 8166744
 * @summary Test Completion
 * @modules jdk.internal.le/jdk.internal.org.jline.reader
 *          jdk.jshell/jdk.internal.jshell.tool:+open
 * @build HistoryTest
 * @run testng HistoryTest
 */

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.Locale;
import java.util.logging.Level;
import java.util.logging.Logger;

import org.testng.annotations.Test;
import jdk.internal.jshell.tool.JShellTool;
import jdk.internal.jshell.tool.JShellToolBuilder;
import jdk.internal.org.jline.reader.History;
import static org.testng.Assert.*;
import org.testng.annotations.BeforeMethod;

public class HistoryTest extends ReplToolTesting {

    private JShellTool repl;

    @Override
    protected void testRawRun(Locale locale, String[] args) {
        // turn on logging of launch failures
        Logger.getLogger("jdk.jshell.execution").setLevel(Level.ALL);
        repl = ((JShellToolBuilder) builder(locale))
                .rawTool();
        try {
            repl.start(args);
        } catch (Exception ex) {
            fail("Repl tool died with exception", ex);
        }
    }

    @Test
    public void testHistory() {
        test(
             a -> {if (!a) setCommandInput("void test() {\n");},
             a -> {if (!a) setCommandInput("    System.err.println(1);\n");},
             a -> {if (!a) setCommandInput("    System.err.println(1);\n");},
             a -> {assertCommand(a, "} //test", "|  created method test()");},
             a -> {
                 if (!a) {
                     try {
                         previousAndAssert(getHistory(), "void test() {\n" +
                                                         "    System.err.println(1);\n" +
                                                         "    System.err.println(1);\n" +
                                                         "} //test");
                     } catch (Exception ex) {
                         throw new IllegalStateException(ex);
                     }
                 }
                 assertCommand(a, "int dummy;", "dummy ==> 0");
             });
        test(
             a -> {if (!a) setCommandInput("void test2() {\n");},
             a -> {assertCommand(a, "} //test2", "|  created method test2()");},
             a -> {
                 if (!a) {
                     try {
                         previousAndAssert(getHistory(), "void test2() {\n" +
                                                         "} //test2");
                         previousAndAssert(getHistory(), "/debug 0"); //added by test framework
                         previousAndAssert(getHistory(), "/exit");
                         previousAndAssert(getHistory(), "int dummy;");
                         previousAndAssert(getHistory(), "void test() {\n" +
                                                         "    System.err.println(1);\n" +
                                                         "    System.err.println(1);\n" +
                                                         "} //test");
                     } catch (Exception ex) {
                         throw new IllegalStateException(ex);
                     }
                 }
                 assertCommand(a, "int dummy;", "dummy ==> 0");
             });
    }

    @Test
    public void test8166744() {
        test(
             a -> {if (!a) setCommandInput("class C {\n");},
             a -> {if (!a) setCommandInput("void f() {\n");},
             a -> {if (!a) setCommandInput("}\n");},
             a -> {assertCommand(a, "}", "|  created class C");},
             a -> {
                 if (!a) {
                     try {
                         previousAndAssert(getHistory(), "class C {\n" +
                                                         "void f() {\n" +
                                                         "}\n" +
                                                         "}");
                         getHistory().add("class C {\n" +
                                          "void f() {\n" +
                                          "}\n" +
                                          "}");
                     } catch (Exception ex) {
                         throw new IllegalStateException(ex);
                     }
                 }
                 assertCommand(a, "int dummy;", "dummy ==> 0");
             });
        test(
             a -> {if (!a) setCommandInput("class C {\n");},
             a -> {if (!a) setCommandInput("void f() {\n");},
             a -> {if (!a) setCommandInput("}\n");},
             a -> {assertCommand(a, "}", "|  created class C");},
             a -> {
                 if (!a) {
                     try {
                         previousAndAssert(getHistory(), "class C {\n" +
                                                         "void f() {\n" +
                                                         "}\n" +
                                                         "}");
                         getHistory().add("class C {\n" +
                                          "void f() {\n" +
                                          "}\n" +
                                          "}");
                     } catch (Exception ex) {
                         throw new IllegalStateException(ex);
                     }
                 }
                 assertCommand(a, "int dummy;", "dummy ==> 0");
             });
    }

    @Test
    public void testReadExistingHistory() {
        prefsMap.put("HISTORY_LINE_0", "/debug 0");
        prefsMap.put("HISTORY_LINE_1", "void test() {\\");
        prefsMap.put("HISTORY_LINE_2", "    System.err.println(1);\\");
        prefsMap.put("HISTORY_LINE_3", "    System.err.println(`\\\\\\\\\\");
        prefsMap.put("HISTORY_LINE_4", "    \\\\\\");
        prefsMap.put("HISTORY_LINE_5", "`);\\");
        prefsMap.put("HISTORY_LINE_6", "} //test");
        test(
             a -> {assertCommand(a, "int i", "i ==> 0");},
             a -> {
                 if (!a) {
                     try {
                         previousAndAssert(getHistory(), "int i");
                         previousAndAssert(getHistory(), "/debug 0"); //added by test framework
                         previousAndAssert(getHistory(), "void test() {\n" +
                                                         "    System.err.println(1);\n" +
                                                         "    System.err.println(`\\\\\n" +
                                                         "    \\\n" +
                                                         "`);\n" +
                                                         "} //test");
                     } catch (Exception ex) {
                         throw new IllegalStateException(ex);
                     }
                 }
                  assertCommand(a, "/exit", "");
             });
        assertEquals(prefsMap.get("HISTORY_LINE_00"), "/debug 0");
        assertEquals(prefsMap.get("HISTORY_LINE_01"), "void test() {\\");
        assertEquals(prefsMap.get("HISTORY_LINE_02"), "    System.err.println(1);\\");
        assertEquals(prefsMap.get("HISTORY_LINE_03"), "    System.err.println(`\\\\\\\\\\");
        assertEquals(prefsMap.get("HISTORY_LINE_04"), "    \\\\\\");
        assertEquals(prefsMap.get("HISTORY_LINE_05"), "`);\\");
        assertEquals(prefsMap.get("HISTORY_LINE_06"), "} //test");
        assertEquals(prefsMap.get("HISTORY_LINE_07"), "/debug 0");
        assertEquals(prefsMap.get("HISTORY_LINE_08"), "int i");
        assertEquals(prefsMap.get("HISTORY_LINE_09"), "/exit");
        System.err.println("prefsMap: " + prefsMap);
    }

    private History getHistory() throws Exception {
        Field input = repl.getClass().getDeclaredField("input");
        input.setAccessible(true);
        Object console = input.get(repl);
        Method getHistory = console.getClass().getDeclaredMethod("getHistory");
        getHistory.setAccessible(true);
        return (History) getHistory.invoke(console);
    }

    private void previousAndAssert(History history, String expected) {
        assertTrue(history.previous());
        assertEquals(history.current().toString(), expected);
    }

    @BeforeMethod
    public void setUp() {
        super.setUp();
        System.setProperty("jshell.test.allow.incomplete.inputs", "false");
    }

}