1 files changed, 15 insertions, 3 deletions

diff --git a/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/resource/RestApiTest.java b/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/resource/RestApiTest.java
index f098e00ec6d..316536b95bc 100644
--- a/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/resource/RestApiTest.java
+++ b/vespaclient-container-plugin/src/test/java/com/yahoo/document/restapi/resource/RestApiTest.java
@@ -271,8 +271,8 @@ public class RestApiTest {
         assertThat(rest, containsString(visit_response_part3));
     }
 
-    String visit_test_bad_uri = "/document/v1/namespace/document-type/group/abc?continuation=abc";
-    String visit_test_bad_response = "Visiting does not support setting value for group/value,";
+    String visit_test_bad_uri = "/document/v1/namespace/document-type/group/abc?continuation=abc&selection=foo";
+    String visit_test_bad_response = "Visiting does not support setting value for group/value in combination with expression";
 
 
     @Test
@@ -283,10 +283,22 @@ public class RestApiTest {
         assertThat(rest, containsString(visit_test_bad_response));
     }
 
+    String visit_test_bad_uri_selection_rewrite = "/document/v1/namespace/document-type/group/abc?continuation=abc";
+    String visit_test_bad_response_selection_rewrite = "doc selection: 'id.group='abc''";
+
+
+    @Test
+    public void testUseExpressionOnVisit() throws Exception {
+        Request request = new Request("http://localhost:" + getFirstListenPort() + visit_test_bad_uri_selection_rewrite);
+        HttpGet get = new HttpGet(request.getUri());
+        String rest = doRest(get);
+        assertThat(rest, containsString(visit_test_bad_response_selection_rewrite));
+    }
+
     private String doRest(HttpRequestBase request) throws IOException {
         HttpClient client = HttpClientBuilder.create().build();
         HttpResponse response = client.execute(request);
-        assertThat(response.getEntity().getContentType().getValue().toString(), is("application/json;charset=utf-8"));
+        assertThat(response.getEntity().getContentType().getValue().toString(), startsWith("application/json;"));
         HttpEntity entity = response.getEntity();
         return EntityUtils.toString(entity);
     }