/*

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

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

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

            }

        });

    }

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

        }

    }

    @Test

    public void testListArgs() {

        String arg = "qqqq";

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

        startVarList.add("int aardvark");

        startVarList.add("int weevil");

        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 -> assertVariable(a, "int", "weevil"),

                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 -> assertCommandOutputStartsWith(a, "/list s3",

                        "s3 : import"),

                a -> assertCommandCheckOutput(a, "/list 1-2 s3",

                        s -> {

                            assertTrue(Pattern.matches(".*aardvark.*\\R.*weevil.*\\R.*s3.*import.*", s.trim()),

                                    "No match: " + s);

                        }),

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

                        "|  No such snippet: " + arg)

        );

    }

    @Test

    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 1-4",

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

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

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

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

                        "|  No such snippet: " + arg)

        );

    }