/*
 * 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 8159111
 * @summary test wrappers and dependencies
 * @modules jdk.jshell/jdk.jshell
 * @build KullaTesting
 * @run testng WrapperTest
 */

import java.util.Collection;
import java.util.List;
import org.testng.annotations.Test;
import jdk.jshell.ErroneousSnippet;
import jdk.jshell.Snippet;
import jdk.jshell.Snippet.Kind;
import jdk.jshell.SourceCodeAnalysis.SnippetWrapper;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertTrue;
import static jdk.jshell.Snippet.Status.RECOVERABLE_DEFINED;
import static jdk.jshell.Snippet.Status.VALID;

@Test
public class WrapperTest extends KullaTesting {

    public void testMethod() {
        String src = "void glib() { System.out.println(\"hello\"); }";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.METHOD, "void", "glib", "println");
        assertPosition(swl.get(0), src, 0, 4);
        assertPosition(swl.get(0), src, 5, 4);
        assertPosition(swl.get(0), src, 15, 6);

        Snippet g = methodKey(assertEval(src, added(VALID)));
        SnippetWrapper swg = getState().sourceCodeAnalysis().wrapper(g);
        assertWrapperHas(swg, src, Kind.METHOD, "void", "glib", "println");
        assertPosition(swg, src, 0, 4);
        assertPosition(swg, src, 5, 4);
        assertPosition(swg, src, 15, 6);
    }

    @Test(enabled = false) // TODO 8159740
    public void testMethodCorralled() {
        String src = "void glib() { f(); }";
        Snippet g = methodKey(assertEval(src, added(RECOVERABLE_DEFINED)));
        SnippetWrapper swg = getState().sourceCodeAnalysis().wrapper(g);
        assertWrapperHas(swg, src, Kind.METHOD, "void", "glib");
        assertPosition(swg, src, 5, 4);
    }

    public void testMethodBad() {
        String src = "void flob() { ?????; }";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.METHOD, "void", "flob", "?????");
        assertPosition(swl.get(0), src, 9, 2);

        Snippet f = key(assertEvalFail(src));
        assertEquals(f.kind(), Kind.ERRONEOUS);
        assertEquals(((ErroneousSnippet)f).probableKind(), Kind.METHOD);
        SnippetWrapper sw = getState().sourceCodeAnalysis().wrapper(f);
        assertWrapperHas(sw, src, Kind.METHOD, "void", "flob", "?????");
        assertPosition(swl.get(0), src, 14, 5);
    }

    public void testVar() {
        String src = "int gx = 1234;";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.VAR, "int", "gx", "1234");
        assertPosition(swl.get(0), src, 4, 2);

        Snippet g = varKey(assertEval(src, added(VALID)));
        SnippetWrapper swg = getState().sourceCodeAnalysis().wrapper(g);
        assertWrapperHas(swg, src, Kind.VAR, "int", "gx", "1234");
        assertPosition(swg, src, 0, 3);
    }

    public void testVarBad() {
        String src = "double dd = ?????;";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.VAR, "double", "dd", "?????");
        assertPosition(swl.get(0), src, 9, 2);

        Snippet f = key(assertEvalFail(src));
        assertEquals(f.kind(), Kind.ERRONEOUS);
        assertEquals(((ErroneousSnippet)f).probableKind(), Kind.VAR);
        SnippetWrapper sw = getState().sourceCodeAnalysis().wrapper(f);
        assertWrapperHas(sw, src, Kind.VAR, "double", "dd", "?????");
        assertPosition(swl.get(0), src, 12, 5);
    }

    public void testImport() {
        String src = "import java.lang.*;";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.IMPORT, "import", "java.lang");
        assertPosition(swl.get(0), src, 7, 4);

        Snippet g = key(assertEval(src, added(VALID)));
        SnippetWrapper swg = getState().sourceCodeAnalysis().wrapper(g);
        assertWrapperHas(swg, src, Kind.IMPORT, "import", "java.lang");
        assertPosition(swg, src, 0, 6);
    }

    public void testImportBad() {
        String src = "import java.?????;";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.IMPORT, "import", "?????");
        assertPosition(swl.get(0), src, 7, 4);

        Snippet f = key(assertEvalFail(src));
        assertEquals(f.kind(), Kind.ERRONEOUS);
        assertEquals(((ErroneousSnippet)f).probableKind(), Kind.IMPORT);
        SnippetWrapper sw = getState().sourceCodeAnalysis().wrapper(f);
        assertWrapperHas(sw, src, Kind.IMPORT, "import", "?????");
        assertPosition(swl.get(0), src, 0, 6);
    }

    public void testErroneous() {
        String src = "@@@@@@@@@@";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 1, "unexpected list length");
        assertWrapperHas(swl.get(0), src, Kind.ERRONEOUS, "@@@@@@@@@@");
        assertPosition(swl.get(0), src, 0, 10);

        Snippet f = key(assertEvalFail(src));
        assertEquals(f.kind(), Kind.ERRONEOUS);
        assertEquals(((ErroneousSnippet)f).probableKind(), Kind.ERRONEOUS);
        SnippetWrapper sw = getState().sourceCodeAnalysis().wrapper(f);
        assertWrapperHas(sw, src, Kind.ERRONEOUS, "@@@@@@@@@@");
        assertPosition(swl.get(0), src, 0, 10);
    }

    public void testEmpty() {
        String src = "";
        List<SnippetWrapper> swl = getState().sourceCodeAnalysis().wrappers(src);
        assertEquals(swl.size(), 0, "expected empty list");
    }

    public void testDependencies() {
        Snippet a = key(assertEval("int aaa = 6;", added(VALID)));
        Snippet b = key(assertEval("class B { B(int x) { aaa = x; } }", added(VALID)));
        Snippet c = key(assertEval("B ccc() { return new B(aaa); }", added(VALID)));
        Collection<Snippet> dep;
        dep = getState().sourceCodeAnalysis().dependents(c);
        assertEquals(dep.size(), 0);
        dep = getState().sourceCodeAnalysis().dependents(b);
        assertEquals(dep.size(), 1);
        assertTrue(dep.contains(c));
        dep = getState().sourceCodeAnalysis().dependents(a);
        assertEquals(dep.size(), 2);
        assertTrue(dep.contains(c));
        assertTrue(dep.contains(b));
    }

    private void assertWrapperHas(SnippetWrapper sw, String source, Kind kind, String... has) {
        assertEquals(sw.source(), source);
        assertEquals(sw.kind(), kind);
        if (kind == Kind.IMPORT) {
            assertTrue(sw.wrapped().contains("import"));
        } else {
            String cn = sw.fullClassName();
            int idx = cn.lastIndexOf(".");
            assertTrue(sw.wrapped().contains(cn.substring(idx+1)));
            assertTrue(sw.wrapped().contains("class"));
        }
        for (String s : has) {
            assertTrue(sw.wrapped().contains(s));
        }
    }

    private void assertPosition(SnippetWrapper sw, String source, int start, int length) {
        int wpg = sw.sourceToWrappedPosition(start);
        assertEquals(sw.wrapped().substring(wpg, wpg+length),
                source.substring(start, start+length),
                "position " + wpg + " in " + sw.wrapped());
        assertEquals(sw.wrappedToSourcePosition(wpg), start);
    }
}