/*

 * Copyright (c) 2016, 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

 * @summary Simple jshell tool tests

 * @modules jdk.compiler/com.sun.tools.javac.api

 *          jdk.compiler/com.sun.tools.javac.main

 *          jdk.jdeps/com.sun.tools.javap

 *          jdk.jshell/jdk.internal.jshell.tool

 * @build KullaTesting TestingInputStream

 * @run testng ToolSimpleTest

 */

import java.util.Arrays;

import java.util.List;

import java.util.Locale;

import java.util.function.Consumer;

import java.util.regex.Pattern;

import java.util.stream.Collectors;

import java.util.stream.Stream;

import org.testng.annotations.Test;

import static org.testng.Assert.assertEquals;

import static org.testng.Assert.assertTrue;

public class ToolSimpleTest extends ReplToolTesting {

    @Test

    public void testRemaining() {

        test(

                (a) -> assertCommand(a, "int z; z =", "z ==> 0"),

                (a) -> assertCommand(a, "5", "z ==> 5"),

                (a) -> assertCommand(a, "/*nada*/; int q =", ""),

                (a) -> assertCommand(a, "77", "q ==> 77"),

                (a) -> assertCommand(a, "//comment;", ""),

                (a) -> assertCommand(a, "int v;", "v ==> 0"),

                (a) -> assertCommand(a, "int v; int c",

                        "v ==> 0\n" +

                        "c ==> 0")

        );

    }

    @Test

    public void testOpenComment() {

        test(

                (a) -> assertCommand(a, "int z = /* blah", ""),

                (a) -> assertCommand(a, "baz */ 5", "z ==> 5"),

                (a) -> assertCommand(a, "/** hoge ", ""),

                (a) -> assertCommand(a, "baz **/", ""),

                (a) -> assertCommand(a, "int v", "v ==> 0")

        );

    }

    @Test

    public void testRawString() {

        test(false, new String[]{"--enable-preview", "--no-startup"},

                (a) -> assertCommand(a, "String s = `abc`", "s ==> \"abc\""),

                (a) -> assertCommand(a, "String a = `abc", ""),

                (a) -> assertCommand(a, "def`", "a ==> \"abc\\ndef\""),

                (a) -> assertCommand(a, "String bj = ``Hi, `Bob` and ```Jim```.``", "bj ==> \"Hi, `Bob` and ```Jim```.\""),

                (a) -> assertCommand(a, "String hw = ````````````", ""),

                (a) -> assertCommand(a, "Hello, world", ""),

                (a) -> assertCommand(a, "````````````;", "hw ==> \"\\nHello, world\\n\""),

                (a) -> assertCommand(a, "String uc = `\\u000d\\u000a`", "uc ==> \"\\\\u000d\\\\u000a\""),

                (a) -> assertCommand(a, "String es = `\\(.\\)\\1`", "es ==> \"\\\\(.\\\\)\\\\1\""),

                (a) -> assertCommand(a, "String end = `abc`+`def`+`ghi`", "end ==> \"abcdefghi\"")

        );

    }

    @Test

    public void testSwitchExpression() {

        test(false, new String[]{"--enable-preview", "--no-startup"},

                (a) -> assertCommand(a, "enum Day {MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY, SATURDAY, SUNDAY }", "|  created enum Day"),

                (a) -> assertCommand(a, "Day day = Day.FRIDAY;", "day ==> FRIDAY"),

                (a) -> assertCommand(a, "switch (day) {", ""),

                (a) -> assertCommand(a, "case MONDAY, FRIDAY, SUNDAY -> 6;", ""),

                (a) -> assertCommand(a, "case TUESDAY -> 7;", ""),

                (a) -> assertCommand(a, "case THURSDAY, SATURDAY -> 8;", ""),

                (a) -> assertCommand(a, "case WEDNESDAY -> 9;", ""),

                (a) -> assertCommandOutputContains(a, "}", " ==> 6")

                );

    }

    @Test

    public void testSwitchExpressionCompletion() {

        test(false, new String[]{"--enable-preview", "--no-startup"},

                (a) -> assertCommand(a, "enum Day {MONDAY, TUESDAY, WEDNESDAY, THURSDAY, FRIDAY, SATURDAY, SUNDAY }", "|  created enum Day"),

                (a) -> assertCommand(a, "Day day = Day.FRIDAY;", "day ==> FRIDAY"),

                (a) -> assertCommand(a, "switch (day) {", ""),

                (a) -> assertCommand(a, "case MONDAY, FRIDAY, SUNDAY -> 6;", ""),

                (a) -> assertCommand(a, "case TUESDAY -> 7;", ""),

                (a) -> assertCommand(a, "case THURSDAY, SATURDAY -> 8;", ""),

                (a) -> assertCommand(a, "case WEDNESDAY -> 9;", ""),

                (a) -> assertCommand(a, "} +", ""),

                (a) -> assertCommandOutputContains(a, "1000", " ==> 1006")

                );

    }

    @Test

    public void testLessThan() {

        test(

                (a) -> assertCommand(a, "45", "$1 ==> 45"),

                (a) -> assertCommand(a, "72", "$2 ==> 72"),

                (a) -> assertCommand(a, "$1 < $2", "$3 ==> true"),

                (a) -> assertCommand(a, "int a, b", "a ==> 0\n" +

                        "b ==> 0"),

                (a) -> assertCommand(a, "a < b", "$6 ==> false")

        );

    }

    @Test

    public void testChainedThrow() {

        test(

                (a) -> assertCommand(a, "void p() throws Exception { ((String) null).toString(); }",

                        "|  created method p()"),

                (a) -> assertCommand(a, "void n() throws Exception { try { p(); } catch (Exception ex) { throw new IOException(\"bar\", ex); }}",

                        "|  created method n()"),

                (a) -> assertCommand(a, "void m() { try { n(); } catch (Exception ex) { throw new RuntimeException(\"foo\", ex); }}",

                        "|  created method m()"),

                (a) -> assertCommand(a, "m()",

                          "|  Exception java.lang.RuntimeException: foo\n"

                        + "|        at m (#3:1)\n"

                        + "|        at (#4:1)\n"

                        + "|  Caused by: java.io.IOException: bar\n"

                        + "|        at n (#2:1)\n"

                        + "|        ...\n"

                        + "|  Caused by: java.lang.NullPointerException\n"

                        + "|        at p (#1:1)\n"

                        + "|        ..."),

                (a) -> assertCommand(a, "/drop p",

                        "|  dropped method p()"),

                (a) -> assertCommand(a, "m()",

                        "|  attempted to call method n() which cannot be invoked until method p() is declared")

        );

    }

    @Test

    public void oneLineOfError() {

        test(

                (a) -> assertCommand(a, "12+", null),

                (a) -> assertCommandCheckOutput(a, "  true", (s) ->

                        assertTrue(s.contains("12+") && !s.contains("true"), "Output: '" + s + "'"))

        );

    }

    @Test

    public void defineVariables() {

        test(

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/vars", assertVariables()),

                (a) -> assertVariable(a, "int", "a"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/vars", assertVariables()),

                (a) -> assertVariable(a, "double", "a", "1", "1.0"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/vars", assertVariables()),

                (a) -> evaluateExpression(a, "double", "2 * a", "2.0"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/vars", assertVariables())

        );

    }

    @Test

    public void defineMethods() {

        test(

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/methods", assertMethods()),

                (a) -> assertMethod(a, "int f() { return 0; }", "()int", "f"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/methods", assertMethods()),

                (a) -> assertMethod(a, "void f(int a) { g(); }", "(int)void", "f"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/methods", assertMethods()),

                (a) -> assertMethod(a, "void g() {}", "()void", "g"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/methods", assertMethods())

        );

    }

    @Test

    public void defineTypes() {

        test(

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/types", assertClasses()),

                (a) -> assertClass(a, "class A { }", "class", "A"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/types", assertClasses()),

                (a) -> assertClass(a, "interface A { }", "interface", "A"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/types", assertClasses()),

                (a) -> assertClass(a, "enum A { }", "enum", "A"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/types", assertClasses()),

                (a) -> assertClass(a, "@interface A { }", "@interface", "A"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/types", assertClasses())

        );

    }

    @Test

    public void defineImports() {

        test(

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/imports", assertImports()),

                (a) -> assertImport(a, "import java.util.stream.Stream;", "", "java.util.stream.Stream"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/imports", assertImports()),

                (a) -> assertImport(a, "import java.util.stream.*;", "", "java.util.stream.*"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/imports", assertImports()),

                (a) -> assertImport(a, "import static java.lang.Math.PI;", "static", "java.lang.Math.PI"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/imports", assertImports()),

                (a) -> assertImport(a, "import static java.lang.Math.*;", "static", "java.lang.Math.*"),

                (a) -> assertCommandCheckOutput(a, "/list", assertList()),

                (a) -> assertCommandCheckOutput(a, "/imports", assertImports())

        );

    }

    @Test

    public void defineVar() {

        test(

                (a) -> assertCommand(a, "int x = 72", "x ==> 72"),

                (a) -> assertCommand(a, "x", "x ==> 72"),

                (a) -> assertCommand(a, "/vars", "|    int x = 72")

        );

    }

    @Test

    public void defineUnresolvedVar() {

        test(

                (a) -> assertCommand(a, "undefined x",

                        "|  created variable x, however, it cannot be referenced until class undefined is declared"),

                (a) -> assertCommand(a, "/vars", "|    undefined x = (not-active)")

        );

    }

    @Test

    public void testUnresolved() {

        test(

                (a) -> assertCommand(a, "int f() { return g() + x + new A().a; }",

                        "|  created method f(), however, it cannot be invoked until method g(), variable x, and class A are declared"),

                (a) -> assertCommand(a, "f()",

                        "|  attempted to call method f() which cannot be invoked until method g(), variable x, and class A are declared"),

                (a) -> assertCommandOutputStartsWith(a, "int g() { return x; }",

                        "|  created method g(), however, it cannot be invoked until variable x is declared"),

                (a) -> assertCommand(a, "g()", "|  attempted to call method g() which cannot be invoked until variable x is declared")

        );

    }

    @Test

    public void testUnknownCommand() {

        test((a) -> assertCommand(a, "/unknown",

                "|  Invalid command: /unknown\n" +

                "|  Type /help for help."));

    }

    @Test

    public void testEmptyClassPath() {

        test(after -> assertCommand(after, "/env --class-path", "|  Argument to class-path missing."));

    }

    @Test

    public void testInvalidClassPath() {

        test(

                a -> assertCommand(a, "/env --class-path snurgefusal",

                        "|  File 'snurgefusal' for '--class-path' is not found."),

                a -> assertCommand(a, "/env --class-path ?",

                        "|  File '?' for '--class-path' is not found.")

        );

    }

    @Test

    public void testNoArgument() {

        test(

                (a) -> assertCommand(a, "/save",

                        "|  '/save' requires a filename argument."),

                (a) -> assertCommand(a, "/open",

                        "|  '/open' requires a filename argument."),

                (a) -> assertCommandOutputStartsWith(a, "/drop",

                        "|  In the /drop argument, please specify an import, variable, method, or class to drop.")

        );

    }

    @Test

    public void testDebug() {

        test(

                (a) -> assertCommand(a, "/deb", "|  Debugging on"),

                (a) -> assertCommand(a, "/debug", "|  Debugging off"),

                (a) -> assertCommand(a, "/debug", "|  Debugging on"),

                (a) -> assertCommand(a, "/deb", "|  Debugging off")

        );

    }

    @Test

    public void testDrop() {

        test(false, new String[]{"--no-startup"},

                a -> assertVariable(a, "int", "a"),

                a -> dropVariable(a, "/drop 1", "int a = 0", "|  dropped variable a"),

                a -> assertMethod(a, "int b() { return 0; }", "()int", "b"),

                a -> dropMethod(a, "/drop 2", "int b()", "|  dropped method b()"),

                a -> assertClass(a, "class A {}", "class", "A"),

                a -> dropClass(a, "/drop 3", "class A", "|  dropped class A"),

                a -> assertImport(a, "import java.util.stream.*;", "", "java.util.stream.*"),

                a -> dropImport(a, "/drop 4", "import java.util.stream.*", ""),

                a -> assertCommand(a, "for (int i = 0; i < 10; ++i) {}", ""),

                a -> assertCommand(a, "/drop 5", ""),

                a -> assertCommand(a, "/list", ""),

                a -> assertCommandCheckOutput(a, "/vars", assertVariables()),

                a -> assertCommandCheckOutput(a, "/methods", assertMethods()),

                a -> assertCommandCheckOutput(a, "/types", assertClasses()),

                a -> assertCommandCheckOutput(a, "/imports", assertImports())

        );

        test(false, new String[]{"--no-startup"},

                a -> assertVariable(a, "int", "a"),

                a -> dropVariable(a, "/drop a", "int a = 0", "|  dropped variable a"),

                a -> assertMethod(a, "int b() { return 0; }", "()int", "b"),

                a -> dropMethod(a, "/drop b", "int b()", "|  dropped method b()"),

                a -> assertClass(a, "class A {}", "class", "A"),

                a -> dropClass(a, "/drop A", "class A", "|  dropped class A"),

                a -> assertCommandCheckOutput(a, "/vars", assertVariables()),

                a -> assertCommandCheckOutput(a, "/methods", assertMethods()),

                a -> assertCommandCheckOutput(a, "/types", assertClasses()),

                a -> assertCommandCheckOutput(a, "/imports", assertImports())

        );

    }

    @Test

    public void testDropRange() {

        test(false, new String[]{"--no-startup"},

                a -> assertVariable(a, "int", "a"),

                a -> assertMethod(a, "int b() { return 0; }", "()int", "b"),

                a -> assertClass(a, "class A {}", "class", "A"),

                a -> assertImport(a, "import java.util.stream.*;", "", "java.util.stream.*"),

                a -> assertCommand(a, "for (int i = 0; i < 10; ++i) {}", ""),

                a -> assertCommand(a, "/drop 3-5 b 1",

                        "|  dropped class A\n" +

                        "|  dropped method b()\n" +

                        "|  dropped variable a\n"),

                a -> assertCommand(a, "/list", "")

        );

    }

    @Test

    public void testDropNegative() {

        test(false, new String[]{"--no-startup"},

                a -> assertCommandOutputStartsWith(a, "/drop 0", "|  No snippet with ID: 0"),

                a -> assertCommandOutputStartsWith(a, "/drop a", "|  No such snippet: a"),

                a -> assertCommandCheckOutput(a, "/drop",

                        assertStartsWith("|  In the /drop argument, please specify an import, variable, method, or class to drop.")),

                a -> assertVariable(a, "int", "a"),

                a -> assertCommand(a, "a", "a ==> 0"),

                a -> assertCommand(a, "/drop 2", ""),

                a -> assertCommand(a, "/drop 2",

                        "|  This command does not accept the snippet '2' : a\n" +

                        "|  See /types, /methods, /vars, or /list")

        );

    }

    @Test

    public void testAmbiguousDrop() {

        test(

                a -> assertVariable(a, "int", "a"),

                a -> assertMethod(a, "int a() { return 0; }", "()int", "a"),

                a -> assertClass(a, "class a {}", "class", "a"),

                a -> assertCommand(a, "/drop a",

                        "|  dropped variable a\n" +

                        "|  dropped method a()\n" +

                        "|  dropped class a")

        );

        test(

                a -> assertMethod(a, "int a() { return 0; }", "()int", "a"),

                a -> assertMethod(a, "double a(int a) { return 0; }", "(int)double", "a"),

                a -> assertMethod(a, "double a(double a) { return 0; }", "(double)double", "a"),

                a -> assertCommand(a, "/drop a",

                        "|  dropped method a()\n" +

                        "|  dropped method a(int)\n" +

                        "|  dropped method a(double)\n")

        );

    }

    @Test

    public void testApplicationOfPost() {

        test(

                (a) -> assertCommand(a, "/set mode t normal -command", "|  Created new feedback mode: t"),

                (a) -> assertCommand(a, "/set feedback t", "|  Feedback mode: t"),

                (a) -> assertCommand(a, "/set format t post \"$%n\"", ""),

                (a) -> assertCommand(a, "/set prompt t \"+\" \"-\"", ""),

                (a) -> assertCommand(a, "/set prompt t", "|  /set prompt t \"+\" \"-\"$")

        );

    }

    @Test

    public void testHelpLength() {

        Consumer<String> testOutput = (s) -> {

            List<String> ss = Stream.of(s.split("\n"))

                    .filter(l -> !l.isEmpty())

                    .collect(Collectors.toList());

            assertTrue(ss.size() >= 10, "Help does not print enough lines:" + s);

        };

        test(

                (a) -> assertCommandCheckOutput(a, "/?", testOutput),

                (a) -> assertCommandCheckOutput(a, "/help", testOutput),

                (a) -> assertCommandCheckOutput(a, "/help /list", testOutput)

        );

    }

    @Test

    public void testHelp() {

        test(

                (a) -> assertHelp(a, "/?", "/list", "/help", "/exit", "intro"),

                (a) -> assertHelp(a, "/help", "/list", "/help", "/exit", "intro"),

                (a) -> assertHelp(a, "/help short", "shortcuts", "<tab>"),

                (a) -> assertHelp(a, "/? /li", "/list -all", "snippets"),

                (a) -> assertHelp(a, "/help /set prompt", "optionally contain '%s'", "quoted"),

                (a) -> assertHelp(a, "/help /help", "/help <command>"),

                (a) -> assertHelp(a, "/help li", "/list -start"),

                (a) -> assertHelp(a, "/help fe", "/set feedback -retain")

        );

    }

    @Test

    public void testHelpStart() {

        test(

                (a) -> assertCommandCheckOutput(a, "/help /exit",

                        s -> assertTrue(s.replaceAll("\\r\\n?", "\n").startsWith(

                                "|  \n" +

                                "|                                   /exit\n" +

                                "|                                   =====\n" +

                                "|  "

                        ))

                )

        );

    }

    @Test

    public void testHelpFormat() {

        test(

                (a) -> assertCommandCheckOutput(a, "/help", s -> {

                    String[] lines = s.split("\\R");

                    assertTrue(lines.length > 20,

                            "Too few lines of /help output: " + lines.length

                          + "\n" + s);

                    for (int i = 0; i < lines.length; ++i) {

                        String l = lines[i];

                        assertTrue(l.startsWith("| "),

                                "Expected /help line to start with | :\n" + l);

                        assertTrue(l.length() <= 80,

                                "/help line too long: " + l.length() + "\n" + l);

                    }

                 })

        );

    }

    @Test

    public void testConfusedUserPseudoCommands() {