/*

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

 * @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.ArrayList;

import java.util.List;

import java.util.function.Consumer;

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;

@Test

public class ToolSimpleTest extends ReplToolTesting {

    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")

        );

    }

    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")

        );

    }

    public void oneLineOfError() {

        test(

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

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

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

        );

    }

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

        );

    }

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

        );

    }

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

        );

    }

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

        );

    }

    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")

        );

    }

    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)")

        );

    }

    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")

        );

    }

    public void testUnknownCommand() {

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

                "|  No such command or snippet id: /unknown\n" +

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

    }

    public void testEmptyClassPath() {

        test(after -> assertCommand(after, "/classpath", "|  The /classpath command requires a path argument."));

    }

    public void testNoArgument() {

        test(

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

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

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

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

        );

    }

    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")

        );

    }

    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; }", "()I", "b"),

                a -> dropMethod(a, "/drop 2", "b ()I", "|  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; }", "()I", "b"),

                a -> dropMethod(a, "/drop b", "b ()I", "|  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())

        );

    }

    public void testDropNegative() {

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

                a -> assertCommandOutputStartsWith(a, "/drop 0", "|  No such snippet: 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")

        );

    }

    public void testAmbiguousDrop() {

        Consumer<String> check = s -> {

            assertTrue(s.startsWith("|  The argument references more than one import, variable, method, or class"), s);

            int lines = s.split("\n").length;

            assertEquals(lines, 5, "Expected 3 ambiguous keys, but found: " + (lines - 2) + "\n" + s);

        };

        test(

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

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

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

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

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

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

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

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

        );

        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 -> assertCommandCheckOutput(a, "/drop a", check),

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

        );

    }

    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)

        );

    }

    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 /help", "/help <command>")

        );

    }

    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);

            }

        });

    }

    // Check that each line of output contains the corresponding string from the list

    private void checkLineToList(String in, List<String> match) {

        String trimmed = in.trim();

        String[] res = trimmed.isEmpty()

                ? new String[0]

                : trimmed.split("\n");

        assertEquals(res.length, match.size(), "Got: " + Arrays.asList(res));

        for (int i = 0; i < match.size(); ++i) {

            assertTrue(res[i].contains(match.get(i)));

        }

    }

    public void testListArgs() {

        String arg = "qqqq";

        List<String> startVarList = new ArrayList<>(START_UP);

        startVarList.add("int aardvark");

        test(

                a -> assertCommandCheckOutput(a, "/list -all",

                        s -> checkLineToList(s, START_UP)),

                a -> assertCommandOutputStartsWith(a, "/list " + arg,

                        "|  No such snippet: " + arg),

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

                a -> assertCommandOutputContains(a, "/list aardvark", "aardvark"),

                a -> assertCommandCheckOutput(a, "/list -start",

                        s -> checkLineToList(s, START_UP)),

                a -> assertCommandCheckOutput(a, "/list -all",

                        s -> checkLineToList(s, startVarList)),

                a -> assertCommandCheckOutput(a, "/list printf",

                        s -> assertTrue(s.contains("void printf"))),

                a -> assertCommandOutputStartsWith(a, "/list " + arg,

                        "|  No such snippet: " + arg)

        );

    }

    public void testVarsArgs() {

        String arg = "qqqq";

        List<String> startVarList = new ArrayList<>();

        test(

                a -> assertCommandCheckOutput(a, "/vars -all",

                        s -> checkLineToList(s, startVarList)),

                a -> assertCommand(a, "/vars " + arg,

                        "|  No such snippet: " + arg),

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

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

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

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

                a -> assertCommandOutputStartsWith(a, "/vars aardvark",

                        "|    int aardvark = 0"),

                a -> assertCommandCheckOutput(a, "/vars -start",

                        s -> checkLineToList(s, startVarList)),

                a -> assertCommandOutputStartsWith(a, "/vars -all",

                        "|    int aardvark = 0\n|    int a = "),

                a -> assertCommandOutputStartsWith(a, "/vars printf",

                        "|  This command does not accept the snippet 'printf'"),

                a -> assertCommand(a, "/var " + arg,

                        "|  No such snippet: " + arg)

        );

    }

    public void testMethodsArgs() {

        String arg = "qqqq";

        List<String> startMethodList = new ArrayList<>(START_UP_CMD_METHOD);

        test(

                a -> assertCommandCheckOutput(a, "/methods -all",

                        s -> checkLineToList(s, startMethodList)),

                a -> assertCommandCheckOutput(a, "/methods -start",

                        s -> checkLineToList(s, startMethodList)),

                a -> assertCommandCheckOutput(a, "/methods printf",

                        s -> checkLineToList(s, startMethodList)),

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

                        s -> checkLineToList(s, startMethodList)),

                a -> assertCommandOutputStartsWith(a, "/methods " + arg,

                        "|  No such snippet: " + arg),

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

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

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

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

                a -> assertCommandOutputStartsWith(a, "/methods " + arg,

                        "|  No such snippet: " + arg),

                a -> assertCommandOutputStartsWith(a, "/methods aardvark",

                        "|  This command does not accept the snippet 'aardvark' : int aardvark"),

                a -> assertCommandCheckOutput(a, "/methods -start",

                        s -> checkLineToList(s, startMethodList)),

                a -> assertCommandCheckOutput(a, "/methods printf",

                        s -> checkLineToList(s, startMethodList)),

                a -> assertCommandOutputStartsWith(a, "/methods g",

                        "|    g ()void"),

                a -> assertCommandOutputStartsWith(a, "/methods f",

                        "|    f ()int\n" +

                        "|    f (int)void")

        );

    }

    public void testTypesArgs() {

        String arg = "qqqq";

        List<String> startTypeList = new ArrayList<>();

        test(

                a -> assertCommandCheckOutput(a, "/types -all",

                        s -> checkLineToList(s, startTypeList)),

                a -> assertCommandCheckOutput(a, "/types -start",

                        s -> checkLineToList(s, startTypeList)),

                a -> assertCommandOutputStartsWith(a, "/types " + arg,

                        "|  No such snippet: " + arg),

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

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

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

                a -> assertCommandOutputStartsWith(a, "/types -all",

                        "|    class A\n" +

                        "|    interface A"),

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

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

                a -> assertCommand(a, "/types aardvark",

                        "|  This command does not accept the snippet 'aardvark' : int aardvark;"),

                a -> assertCommandOutputStartsWith(a, "/types A",

                        "|    interface A"),

                a -> assertCommandOutputStartsWith(a, "/types E",

                        "|    enum E"),

                a -> assertCommandOutputStartsWith(a, "/types B",

                        "|    @interface B"),

                a -> assertCommandOutputStartsWith(a, "/types " + arg,

                        "|  No such snippet: " + arg),

                a -> assertCommandCheckOutput(a, "/types -start",

                        s -> checkLineToList(s, startTypeList))

        );

    }

    public void defineClasses() {

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

        );

    }

    public void testCommandPrefix() {

        test(a -> assertCommandCheckOutput(a, "/s",

                      assertStartsWith("|  Command: '/s' is ambiguous: /save, /set")),

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

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

                      assertStartsWith("|    int var = 0")),

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

                      assertStartsWith("|  '/save' requires a filename argument.")));

    }

    public void testOptionQ() {

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

                (a) -> assertCommand(a, "1+1", "$1 ==> 2"),

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

        );

    }

    public void testOptionS() {

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

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

        );

    }

    public void testOptionV() {

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

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

                        "$1 ==> 2\n" +

                        "|  created scratch variable $1 : int")

        );

    }

    public void testOptionFeedback() {

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

                (a) -> assertCommand(a, "1+1", "$1 ==> 2"),

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

        );

    }

    public void testCompoundOptions() {

        Consumer<String> confirmNoStartup = s -> {

                    assertEquals(0, Stream.of(s.split("\n"))