Add completion candidate suffix test, #425

jline · Nov 7, 2020 · 40c4d32 · 40c4d32
1 parent 404565b
commit 40c4d32
Showing 1 changed file with 31 additions and 3 deletions.
diff --git a/reader/src/test/java/org/jline/reader/impl/CompletionTest.java b/reader/src/test/java/org/jline/reader/impl/CompletionTest.java
@@ -11,10 +11,8 @@
 import java.io.IOException;
 import java.util.Arrays;
 
-import static java.util.Arrays.asList;
-import org.jline.reader.Completer;
+import org.jline.reader.*;
 import org.jline.reader.LineReader.Option;
-import org.jline.reader.Reference;
 import org.jline.reader.impl.completer.AggregateCompleter;
 import org.jline.reader.impl.completer.ArgumentCompleter;
 import org.jline.reader.impl.completer.NullCompleter;
@@ -134,6 +132,35 @@ public void testCompletePrefix() throws Exception {
         assertLine("read and nd", new TestBuffer("read and\033[D\033[D\t\n"));
     }
 
+    @Test
+    public void testSuffix() {
+        reader.setCompleter((reader, line, candidates) -> {
+                    candidates.add(new Candidate(
+                            /* value    = */ "range(",
+                            /* displ    = */ "range(",
+                            /* group    = */ null,
+                            /* descr    = */ null,
+                            /* suffix   = */ "(",
+                            /* key      = */ null,
+                            /* complete = */ false));
+                    candidates.add(new Candidate(
+                            /* value    = */ "strangeTest",
+                            /* displ    = */ "strangeTest",
+                            /* group    = */ null,
+                            /* descr    = */ null,
+                            /* suffix   = */ "Test",
+                            /* key      = */ null,
+                            /* complete = */ false));
+                }
+        );
+        //  DEFAULT_REMOVE_SUFFIX_CHARS = " \t\n;&|";
+
+        assertLine("range ;", new TestBuffer("r\t;\n"));
+        assertLine("range(1", new TestBuffer("r\t1\n"));
+        assertLine("strange ", new TestBuffer("s\t\n"));
+        assertLine("strangeTests", new TestBuffer("s\ts\n"));
+    }
+
     @Test
     public void testMenuOrder() {
         reader.setCompleter(new StringsCompleter(Arrays.asList("ae_helloWorld1", "ad_helloWorld12", "ac_helloWorld1234", "ab_helloWorld123", "aa_helloWorld12345")));
@@ -161,4 +188,5 @@ public void testParserEofOnEscapedNewLine() {
 
         assertLine("test ", new TestBuffer("test \\\t\n\n"));
     }
+
 }