Add a test

2 files changed, 12 insertions, 0 deletions

diff --git a/client/src/main/java/ai/vespa/client/dsl/Rank.java b/client/src/main/java/ai/vespa/client/dsl/Rank.java
index aad70a29502..e1623aeec00 100644
--- a/client/src/main/java/ai/vespa/client/dsl/Rank.java
+++ b/client/src/main/java/ai/vespa/client/dsl/Rank.java
@@ -7,6 +7,8 @@ import java.util.Objects;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
+// TODO:
+// - make all items subclasses of "Query" rather than QueryChain and remove all content from QueryChain
 public class Rank extends QueryChain {
 
     private final List<QueryChain> queries = new ArrayList<>();
diff --git a/client/src/test/java/ai/vespa/client/dsl/QTest.java b/client/src/test/java/ai/vespa/client/dsl/QTest.java
index b3c26e3db6b..aae8b2c8923 100644
--- a/client/src/test/java/ai/vespa/client/dsl/QTest.java
+++ b/client/src/test/java/ai/vespa/client/dsl/QTest.java
@@ -176,6 +176,16 @@ class QTest {
     }
 
     @Test
+    void userInput_with_rank_2() {
+        String q = Q.select("doc_id")
+                    .from("ss")
+                    .where(Q.rank(Q.p("query").nearestNeighbor( "vector").and("doc_type").contains("pdf"),
+                                  Q.ui("@query"))).build();
+        assertEquals("yql=select doc_id from ss where rank(nearestNeighbor(query, vector) and doc_type contains \"pdf\", userInput(@query))",
+                     q);
+    }
+
+    @Test
     void dot_product() {
         String q = Q.select("*")
                 .from("sd1")