Java tutorial
/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you under the Apache License, Version 2.0 (the * "License"); you may not use this file except in compliance * with the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * KIND, either express or implied. See the License for the * specific language governing permissions and limitations * under the License. */ package org.apache.tinkerpop.gremlin.server; import org.apache.commons.lang3.RandomStringUtils; import org.apache.tinkerpop.gremlin.driver.ser.GraphSONMessageSerializerV1d0; import org.apache.tinkerpop.gremlin.driver.Tokens; import org.apache.tinkerpop.gremlin.driver.ser.GraphSONMessageSerializerV2d0; import org.apache.tinkerpop.gremlin.driver.ser.GraphSONMessageSerializerV3d0; import org.apache.tinkerpop.gremlin.jsr223.ScriptFileGremlinPlugin; import org.apache.tinkerpop.gremlin.server.auth.SimpleAuthenticator; import org.apache.tinkerpop.gremlin.server.channel.HttpChannelizer; import org.apache.tinkerpop.gremlin.server.handler.HttpBasicAuthenticationHandler; import org.apache.http.Consts; import org.apache.http.client.methods.CloseableHttpResponse; import org.apache.http.client.methods.HttpGet; import org.apache.http.client.methods.HttpPost; import org.apache.http.entity.StringEntity; import org.apache.http.impl.client.CloseableHttpClient; import org.apache.http.impl.client.HttpClients; import org.apache.http.util.EntityUtils; import org.apache.tinkerpop.gremlin.structure.io.graphson.GraphSONTokens; import org.apache.tinkerpop.shaded.jackson.databind.JsonNode; import org.apache.tinkerpop.shaded.jackson.databind.ObjectMapper; import org.junit.Test; import java.io.File; import java.time.Instant; import java.util.Base64; import java.util.Collections; import java.util.HashMap; import java.util.Map; import static org.hamcrest.MatcherAssert.assertThat; import static org.hamcrest.core.StringStartsWith.startsWith; import static org.junit.Assert.assertEquals; /** * Integration tests for server-side settings and processing. * * @author Stephen Mallette (http://stephen.genoprime.com) */ public class GremlinServerHttpIntegrateTest extends AbstractGremlinServerIntegrationTest { private final ObjectMapper mapper = new ObjectMapper(); private final Base64.Encoder encoder = Base64.getUrlEncoder(); /** * Configure specific Gremlin Server settings for specific tests. */ @Override public Settings overrideSettings(final Settings settings) { settings.channelizer = HttpChannelizer.class.getName(); final String nameOfTest = name.getMethodName(); switch (nameOfTest) { case "should413OnPostWithResultTooLarge": settings.maxContentLength = 31; break; case "should200OnGETWithGremlinQueryStringArgumentWithIteratorResult": case "should200OnPOSTWithGremlinJsonEndcodedBodyWithIteratorResult": case "should200OnPOSTWithGremlinJsonEndcodedBodyWithIteratorResultAndAliases": case "should200OnGETWithGremlinQueryStringArgumentWithIteratorResultAndAliases": final Map<String, Object> m = new HashMap<>(); m.put("files", Collections.singletonList("scripts/generate-classic.groovy")); settings.scriptEngines.get("gremlin-groovy").plugins.put(ScriptFileGremlinPlugin.class.getName(), m); break; case "should200OnPOSTTransactionalGraph": deleteDirectory(new File("/tmp/neo4j")); settings.graphs.put("graph", "conf/neo4j-empty.properties"); break; case "should200OnPOSTTransactionalGraphInStrictMode": settings.strictTransactionManagement = true; deleteDirectory(new File("/tmp/neo4j")); settings.graphs.put("graph", "conf/neo4j-empty.properties"); break; case "should200OnPOSTWithGraphSON1d0AcceptHeaderDefaultResultToJson": settings.serializers.clear(); final Settings.SerializerSettings serializerSettingsV1 = new Settings.SerializerSettings(); serializerSettingsV1.className = GraphSONMessageSerializerV1d0.class.getName(); settings.serializers.add(serializerSettingsV1); break; case "should200OnPOSTWithGraphSON2d0AcceptHeaderDefaultResultToJson": settings.serializers.clear(); final Settings.SerializerSettings serializerSettingsV2 = new Settings.SerializerSettings(); serializerSettingsV2.className = GraphSONMessageSerializerV2d0.class.getName(); settings.serializers.add(serializerSettingsV2); break; case "should200OnPOSTWithGraphSON3d0AcceptHeaderDefaultResultToJson": settings.serializers.clear(); final Settings.SerializerSettings serializerSettingsV3 = new Settings.SerializerSettings(); serializerSettingsV3.className = GraphSONMessageSerializerV3d0.class.getName(); settings.serializers.add(serializerSettingsV3); break; case "should401OnGETWithNoAuthorizationHeader": case "should401OnPOSTWithNoAuthorizationHeader": case "should401OnGETWithBadAuthorizationHeader": case "should401OnPOSTWithBadAuthorizationHeader": case "should401OnGETWithBadEncodedAuthorizationHeader": case "should401OnPOSTWithBadEncodedAuthorizationHeader": case "should401OnGETWithInvalidPasswordAuthorizationHeader": case "should401OnPOSTWithInvalidPasswordAuthorizationHeader": case "should200OnGETWithAuthorizationHeader": case "should200OnPOSTWithAuthorizationHeaderExplicitHandlerSetting": configureForAuthenticationWithHandlerClass(settings); break; case "should200OnPOSTWithAuthorizationHeader": configureForAuthentication(settings); break; } return settings; } private void configureForAuthentication(final Settings settings) { final Settings.AuthenticationSettings authSettings = new Settings.AuthenticationSettings(); authSettings.authenticator = SimpleAuthenticator.class.getName(); // use a credentials graph with one user in it: stephen/password final Map<String, Object> authConfig = new HashMap<>(); authConfig.put(SimpleAuthenticator.CONFIG_CREDENTIALS_DB, "conf/tinkergraph-credentials.properties"); authSettings.config = authConfig; settings.authentication = authSettings; } private void configureForAuthenticationWithHandlerClass(final Settings settings) { final Settings.AuthenticationSettings authSettings = new Settings.AuthenticationSettings(); authSettings.authenticator = SimpleAuthenticator.class.getName(); //Add basic auth handler to make sure the reflection code path works. authSettings.authenticationHandler = HttpBasicAuthenticationHandler.class.getName(); // use a credentials graph with one user in it: stephen/password final Map<String, Object> authConfig = new HashMap<>(); authConfig.put(SimpleAuthenticator.CONFIG_CREDENTIALS_DB, "conf/tinkergraph-credentials.properties"); authSettings.config = authConfig; settings.authentication = authSettings; } @Test public void should413OnPostWithResultTooLarge() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); final String bigPost = RandomStringUtils.random(32); httppost.setEntity( new StringEntity("{\"gremlin\":\"" + bigPost + "\", \"bindings\":{\"x\":\"10\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(413, response.getStatusLine().getStatusCode()); } } @Test public void should401OnGETWithNoAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnPOSTWithNoAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnGETWithBadAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Authorization", "not-base-64-encoded"); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnPOSTWithBadAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Authorization", "not-base-64-encoded"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnGETWithBadEncodedAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Authorization", "Basic: not-base-64-encoded"); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnPOSTWithBadEncodedAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Authorization", "Basic: not-base-64-encoded"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnGETWithInvalidPasswordAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Authorization", "Basic " + encoder.encodeToString("stephen:not-my-password".getBytes())); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should401OnPOSTWithInvalidPasswordAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Authorization", "Basic " + encoder.encodeToString("stephen:not-my-password".getBytes())); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(401, response.getStatusLine().getStatusCode()); } } @Test public void should200OnGETWithAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Authorization", "Basic " + encoder.encodeToString("stephen:password".getBytes())); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithAuthorizationHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Authorization", "Basic " + encoder.encodeToString("stephen:password".getBytes())); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithAuthorizationHeaderExplicitHandlerSetting() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Authorization", "Basic " + encoder.encodeToString("stephen:password".getBytes())); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnGETWithGremlinQueryStringArgumentWithBindingsAndFunction() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString( "?gremlin=addItUp(Integer.parseInt(x),Integer.parseInt(y))&bindings.x=10&bindings.y=10")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(20, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnGETWithGremlinQueryStringArgumentWithIteratorResult() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=g.V()")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } } @Test public void should200OnGETWithGremlinQueryStringArgumentWithIteratorResultAndAliases() throws Exception { // we can remove this first test when rebindings are completely removed final CloseableHttpClient httpclientLegacy = HttpClients.createDefault(); final HttpGet httpgetLegacy = new HttpGet( TestClientFactory.createURLString("?gremlin=g1.V()&rebindings.g1=g")); try (final CloseableHttpResponse response = httpclientLegacy.execute(httpgetLegacy)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=g1.V()&aliases.g1=g")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } } @Test public void should200OnGETWithGremlinQueryStringArgument() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").intValue()); } } @Test public void should200OnGETWithGremlinQueryStringArgumentReturningVertex() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet( TestClientFactory.createURLString("?gremlin=graph.addVertex('name','stephen')")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals("stephen", node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).get("properties").get("name").get(0) .get(GraphSONTokens.VALUEPROP).get(GraphSONTokens.VALUE).asText()); } } @Test public void should200OnGETWithGremlinQueryStringArgumentWithBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory .createURLString("?gremlin=Integer.parseInt(x)%2BInteger.parseInt(y)&bindings.x=10&bindings.y=10")); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(20, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should400OnGETWithNoGremlinQueryStringArgument() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString()); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(400, response.getStatusLine().getStatusCode()); } } @Test public void should200OnGETWithAnyAcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Accept", "*/*"); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).asInt()); } } @Test public void should400OnGETWithBadAcceptHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=1-1")); httpget.addHeader("Accept", "application/json+something-else-that-does-not-exist"); try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(400, response.getStatusLine().getStatusCode()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBody() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyForJavaTime() throws Exception { // basic test of java.time.* serialization over JSON from the server perspective. more complete tests // exist in gremlin-core final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"java.time.Instant.MAX\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(Instant.MAX, Instant.parse(node.get("result").get("data").get(GraphSONTokens.VALUEPROP) .get(0).get(GraphSONTokens.VALUEPROP).asText())); } } @Test public void should200OnPOSTTransactionalGraph() throws Exception { assumeNeo4jIsPresent(); final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"graph.addVertex('name','stephen');g.V().count()\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } final HttpGet httpget = new HttpGet(TestClientFactory.createURLString("?gremlin=g.V().count()")); httpget.addHeader("Accept", "application/json"); // execute this a bunch of times so that there's a good chance a different thread on the server processes // the request for (int ix = 0; ix < 100; ix++) { try (final CloseableHttpResponse response = httpclient.execute(httpget)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } } @Test public void should200OnPOSTTransactionalGraphInStrictMode() throws Exception { assumeNeo4jIsPresent(); // we can remove this first test when rebindings are completely removed final CloseableHttpClient httpclientLegacy = HttpClients.createDefault(); final HttpPost httppostLegacy = new HttpPost(TestClientFactory.createURLString()); httppostLegacy.addHeader("Content-Type", "application/json"); httppostLegacy.setEntity( new StringEntity("{\"gremlin\":\"g1.addV()\",\"rebindings\":{\"g1\":\"g\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclientLegacy.execute(httppostLegacy)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity( new StringEntity("{\"gremlin\":\"g1.addV()\",\"aliases\":{\"g1\":\"g\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyWithIteratorResult() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"g.V()\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyWithTinkerGraphResult() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity( "{\"gremlin\":\"org.apache.tinkerpop.gremlin.tinkergraph.structure.TinkerFactory.createModern()\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode resultJson = mapper.readTree(json); final JsonNode data = resultJson.get("result").get("data"); assertEquals(1, data.get(GraphSONTokens.VALUEPROP).size()); assertEquals(6, data.get(GraphSONTokens.VALUEPROP).get(0).get(GraphSONTokens.VALUEPROP) .get(GraphSONTokens.VERTICES).size()); assertEquals(6, data.get(GraphSONTokens.VALUEPROP).get(0).get(GraphSONTokens.VALUEPROP) .get(GraphSONTokens.EDGES).size()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyWithIteratorResultAndAliases() throws Exception { // we can remove this first test when rebindings are completely removed final CloseableHttpClient httpclientLegacy = HttpClients.createDefault(); final HttpPost httppostLegacy = new HttpPost(TestClientFactory.createURLString()); httppostLegacy.addHeader("Content-Type", "application/json"); httppostLegacy.setEntity( new StringEntity("{\"gremlin\":\"g1.V()\",\"rebindings\":{\"g1\":\"g\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclientLegacy.execute(httppostLegacy)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"g1.V()\",\"aliases\":{\"g1\":\"g\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(6, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).size()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity( new StringEntity("{\"gremlin\":\"x+y\", \"bindings\":{\"x\":10, \"y\":10}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(20, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndLongBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":10}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(10, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndDoubleBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":10.5}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(10.5d, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).doubleValue(), 0.0001); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndStringBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":\"10\"}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals("10", node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0).textValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndBooleanBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":true}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(true, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0).booleanValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndNullBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":null}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(true, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0).isNull()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndArrayBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":[1,2,3]}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(true, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).isArray()); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).intValue()); assertEquals(2, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(1) .get(GraphSONTokens.VALUEPROP).intValue()); assertEquals(3, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(2) .get(GraphSONTokens.VALUEPROP).intValue()); } } @Test public void should200OnPOSTWithGremlinJsonEndcodedBodyAndMapBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x\", \"bindings\":{\"x\":{\"y\":1}}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals("g:Map", node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0).get("@type").asText()); assertEquals(1, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).get(1).get(GraphSONTokens.VALUEPROP).asInt()); } } @Test public void should400OnPOSTWithGremlinJsonEndcodedBodyAndBadBindings() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"x+y\", \"bindings\":10}}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(400, response.getStatusLine().getStatusCode()); } } @Test public void should400OnPOSTWithGremlinJsonEndcodedBodyWithNoGremlinKey() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremadfadflin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(400, response.getStatusLine().getStatusCode()); } } @Test public void should400OnPOSTWithBadAcceptHeader() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Accept", "application/json+something-else-that-does-not-exist"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(400, response.getStatusLine().getStatusCode()); } } @Test public void should200OnPOSTWithAnyAcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.addHeader("Accept", "*/*"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).asInt()); } } @Test public void should200OnPOSTWithComplexAcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "*.*;q=0.8,application/xhtml"); httppost.addHeader("Accept", "*/*"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).asInt()); } } @Test public void should500OnGETWithGremlinEvalFailure() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.addHeader("Content-Type", "application/json"); httppost.setEntity(new StringEntity("{\"gremlin\":\"1/0\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(500, response.getStatusLine().getStatusCode()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals("java.lang.ArithmeticException", node.get(Tokens.STATUS_ATTRIBUTE_EXCEPTIONS).get(0).asText()); assertEquals(1, node.get(Tokens.STATUS_ATTRIBUTE_EXCEPTIONS).size()); assertThat(node.get(Tokens.STATUS_ATTRIBUTE_STACK_TRACE).asText(), startsWith( "java.lang.ArithmeticException: Division by zero\n\tat java.math.BigDecimal.divide(BigDecimal.java")); } } @Test public void should200OnPOSTWithGraphSON1d0AcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(0).asInt()); } } @Test public void should200OnPOSTWithGraphSON2d0AcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(0).asInt()); } } @Test public void should200OnPOSTWithGraphSON3d0AcceptHeaderDefaultResultToJson() throws Exception { final CloseableHttpClient httpclient = HttpClients.createDefault(); final HttpPost httppost = new HttpPost(TestClientFactory.createURLString()); httppost.setEntity(new StringEntity("{\"gremlin\":\"1-1\"}", Consts.UTF_8)); try (final CloseableHttpResponse response = httpclient.execute(httppost)) { assertEquals(200, response.getStatusLine().getStatusCode()); assertEquals("application/json", response.getEntity().getContentType().getValue()); final String json = EntityUtils.toString(response.getEntity()); final JsonNode node = mapper.readTree(json); assertEquals(0, node.get("result").get("data").get(GraphSONTokens.VALUEPROP).get(0) .get(GraphSONTokens.VALUEPROP).asInt()); } } }