/*

 * 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

 * @bug 8153716 8143955 8151754 8150382 8153920 8156910 8131024 8160089 8153897 8167128 8154513 8170015 8170368 8172102 8172103  8165405 8173073 8173848 8174041 8173916 8174028 8174262 8174797 8177079 8180508 8177466 8172154 8192979 8191842 8198573 8198801 8210596 8210959 8215099

 * @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/othervm ToolSimpleTest

 */

import java.util.ArrayList;

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 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 testThrowWithPercent() {

        test(

                (a) -> assertCommandCheckOutput(a,

                        "URI u = new URI(\"http\", null, \"h\", -1, \"a\" + (char)0x04, null, null);", (s) ->

                                assertTrue(s.contains("URISyntaxException") && !s.contains("JShellTool"),

                                        "Output: '" + s + "'")),

                (a) -> assertCommandCheckOutput(a,

                        "throw new Exception(\"%z\")", (s) ->

                                assertTrue(s.contains("java.lang.Exception") && !s.contains("UnknownFormatConversionException"),

                                        "Output: '" + s + "'"))

        );

    }

    @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, "/help keys", "line", "Shift", "imports", "history"),

                (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() {

        test(

                (a) -> assertHelp(a, "/-<n>", "last snippet", "digits"),

                (a) -> assertHelp(a, "/<id>", "last snippet", "digits")

        );

    }

    private void assertHelp(boolean a, String command, String... find) {

        assertCommandCheckOutput(a, command, s -> {

            for (String f : find) {

                assertTrue(s.contains(f),

                        "Expected output of " + command + " to contain: " + f

                      + "\n" + s);

            }

        });

    }