CommandCompletionTest.java revision 3314:97ec97671022 
1/*
2 * Copyright (c) 2015, 2016, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24/*
25 * @test
26 * @bug 8144095
27 * @summary Test Command Completion
28 * @modules jdk.compiler/com.sun.tools.javac.api
29 *          jdk.compiler/com.sun.tools.javac.main
30 *          jdk.jdeps/com.sun.tools.javap
31 *          jdk.jshell/jdk.internal.jshell.tool
32 * @library /tools/lib
33 * @build toolbox.ToolBox toolbox.JarTask toolbox.JavacTask
34 * @build ReplToolTesting TestingInputStream Compiler
35 * @run testng CommandCompletionTest
36 */
37
38import java.io.IOException;
39import java.nio.file.FileSystems;
40import java.nio.file.Files;
41import java.nio.file.Path;
42import java.nio.file.Paths;
43import java.util.Collections;
44import java.util.List;
45import java.util.function.Predicate;
46import java.util.stream.Collectors;
47import java.util.stream.Stream;
48
49import org.testng.annotations.Test;
50
51@Test
52public class CommandCompletionTest extends ReplToolTesting {
53
54    public void testCommand() {
55        assertCompletion("/deb|", false);
56        assertCompletion("/c|", false, "/classes ", "/classpath ");
57        assertCompletion("/h|", false, "/help ", "/history ");
58    }
59
60    public void testList() {
61        test(false, new String[] {"-nostartup"},
62                a -> assertCompletion(a, "/l|", false, "/list "),
63                a -> assertCompletion(a, "/list |", false, "all ", "history ", "start "),
64                a -> assertCompletion(a, "/list h|", false, "history "),
65                a -> assertCompletion(a, "/list q|", false),
66                a -> assertVariable(a, "int", "xray"),
67                a -> assertCompletion(a, "/list |", false, "1", "all ", "history ", "start ", "xray"),
68                a -> assertCompletion(a, "/list x|", false, "xray")
69        );
70    }
71
72    public void testDrop() {
73        test(false, new String[] {"-nostartup"},
74                a -> assertCompletion(a, "/d|", false, "/drop "),
75                a -> assertClass(a, "class cTest {}", "class", "cTest"),
76                a -> assertMethod(a, "int mTest() { return 0; }", "()I", "mTest"),
77                a -> assertVariable(a, "int", "fTest"),
78                a -> assertCompletion(a, "/drop |", false, "1", "2", "3", "cTest", "fTest", "mTest"),
79                a -> assertCompletion(a, "/drop f|", false, "fTest")
80        );
81    }
82
83    public void testEdit() {
84        test(false, new String[]{"-nostartup"},
85                a -> assertCompletion(a, "/e|", false, "/edit ", "/exit "),
86                a -> assertCompletion(a, "/ed|", false, "/edit "),
87                a -> assertClass(a, "class cTest {}", "class", "cTest"),
88                a -> assertMethod(a, "int mTest() { return 0; }", "()I", "mTest"),
89                a -> assertVariable(a, "int", "fTest"),
90                a -> assertCompletion(a, "/edit |", false, "1", "2", "3", "cTest", "fTest", "mTest"),
91                a -> assertCompletion(a, "/edit f|", false, "fTest")
92        );
93    }
94
95    public void testOpen() throws IOException {
96        Compiler compiler = new Compiler();
97        assertCompletion("/o|", false, "/open ");
98        List<String> p1 = listFiles(Paths.get(""));
99        FileSystems.getDefault().getRootDirectories().forEach(s -> p1.add(s.toString()));
100        Collections.sort(p1);
101        assertCompletion("/open |", false, p1.toArray(new String[p1.size()]));
102        Path classDir = compiler.getClassDir();
103        List<String> p2 = listFiles(classDir);
104        assertCompletion("/open " + classDir + "/|", false, p2.toArray(new String[p2.size()]));
105    }
106
107    public void testSave() throws IOException {
108        Compiler compiler = new Compiler();
109        assertCompletion("/s|", false, "/save ", "/set ");
110        List<String> p1 = listFiles(Paths.get(""));
111        Collections.addAll(p1, "all ", "history ", "start ");
112        FileSystems.getDefault().getRootDirectories().forEach(s -> p1.add(s.toString()));
113        Collections.sort(p1);
114        assertCompletion("/save |", false, p1.toArray(new String[p1.size()]));
115        Path classDir = compiler.getClassDir();
116        List<String> p2 = listFiles(classDir);
117        assertCompletion("/save " + classDir + "/|",
118                false, p2.toArray(new String[p2.size()]));
119        assertCompletion("/save all " + classDir + "/|",
120                false, p2.toArray(new String[p2.size()]));
121    }
122
123    public void testClassPath() throws IOException {
124        assertCompletion("/classp|", false, "/classpath ");
125        Compiler compiler = new Compiler();
126        Path outDir = compiler.getPath("testClasspathCompletion");
127        Files.createDirectories(outDir);
128        Files.createDirectories(outDir.resolve("dir"));
129        createIfNeeded(outDir.resolve("test.jar"));
130        createIfNeeded(outDir.resolve("test.zip"));
131        compiler.compile(outDir, "package pkg; public class A { public String toString() { return \"A\"; } }");
132        String jarName = "test.jar";
133        compiler.jar(outDir, jarName, "pkg/A.class");
134        compiler.getPath(outDir).resolve(jarName);
135        List<String> paths = listFiles(outDir, CLASSPATH_FILTER);
136        assertCompletion("/classpath " + outDir + "/|", false, paths.toArray(new String[paths.size()]));
137    }
138
139    public void testUserHome() throws IOException {
140        List<String> completions;
141        Path home = Paths.get(System.getProperty("user.home"));
142        try (Stream<Path> content = Files.list(home)) {
143            completions = content.filter(CLASSPATH_FILTER)
144                                 .map(file -> file.getFileName().toString() + (Files.isDirectory(file) ? "/" : ""))
145                                 .sorted()
146                                 .collect(Collectors.toList());
147        }
148        assertCompletion("/classpath ~/|", false, completions.toArray(new String[completions.size()]));
149    }
150
151    private void createIfNeeded(Path file) throws IOException {
152        if (!Files.exists(file))
153            Files.createFile(file);
154    }
155    private List<String> listFiles(Path path) throws IOException {
156        return listFiles(path, ACCEPT_ALL);
157    }
158
159    private List<String> listFiles(Path path, Predicate<? super Path> filter) throws IOException {
160        try (Stream<Path> stream = Files.list(path)) {
161            return stream.filter(filter)
162                         .map(p -> p.getFileName().toString() + (Files.isDirectory(p) ? "/" : ""))
163                         .sorted()
164                         .collect(Collectors.toList());
165        }
166    }
167
168    private static final Predicate<? super Path> ACCEPT_ALL =
169            (file) -> !file.endsWith(".") && !file.endsWith("..");
170
171    private static final Predicate<? super Path> CLASSPATH_FILTER =
172            (file) -> ACCEPT_ALL.test(file) &&
173                    (Files.isDirectory(file) ||
174                     file.getFileName().toString().endsWith(".jar") ||
175                     file.getFileName().toString().endsWith(".zip"));
176}
177