added more tests

Author: a-brandt

src/test/java/com/arangodb/example/document/AqlQueryWithSpecialReturnTypesExample.java

@@ -0,0 +1,146 @@
+/*
+ * Copyright (C) 2015 ArangoDB GmbH
+ *
+ * Licensed 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 com.arangodb.example.document;
+
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
+import com.arangodb.ArangoDriver;
+import com.arangodb.ArangoException;
+import com.arangodb.CursorResult;
+import com.arangodb.util.AqlQueryOptions;
+
+public class AqlQueryWithSpecialReturnTypesExample extends BaseExample {
+
+	private static final String DATABASE_NAME = "SimplePersonAqlQueryWithLimitExample";
+
+	private static final String COLLECTION_NAME = "SimplePersonAqlQueryWithLimitExample";
+
+	public ArangoDriver arangoDriver;
+
+	@Before
+	public void _before() {
+		removeTestDatabase(DATABASE_NAME);
+
+		arangoDriver = getArangoDriver(getConfiguration());
+		createDatabase(arangoDriver, DATABASE_NAME);
+		createCollection(arangoDriver, COLLECTION_NAME);
+	}
+
+	@Test
+	public void simplePersonAqlWithLimitQuery() {
+		//
+		// You can find the ArangoDB Web interface here:
+		// http://127.0.0.1:8529/
+		//
+		// change the log level to "debug" in /src/test/resource/logback.xml to
+		// see the HTTP communication
+
+		try {
+			//
+			printHeadline("create example documents");
+			//
+			createExamples();
+
+			//
+			printHeadline("build query");
+			//
+
+			// bind @gender to WOMAN
+			HashMap<String, Object> bindVars = new HashMap<String, Object>();
+			bindVars.put("gender", FEMALE);
+
+			// query (count = true, batchSize = 5)
+			AqlQueryOptions aqlQueryOptions = new AqlQueryOptions().setCount(true).setBatchSize(5);
+
+			//
+			printHeadline("get query results in a map");
+			//
+
+			String queryString = "FOR t IN " + COLLECTION_NAME
+					+ " FILTER t.age >= 20 && t.age < 30 && t.gender == @gender RETURN t";
+			System.out.println(queryString);
+
+			@SuppressWarnings("rawtypes")
+			CursorResult<Map> cursor = arangoDriver.executeAqlQuery(queryString, bindVars, aqlQueryOptions, Map.class);
+			Assert.assertNotNull(cursor);
+
+			@SuppressWarnings("rawtypes")
+			Iterator<Map> iterator = cursor.iterator();
+			while (iterator.hasNext()) {
+				Map<?, ?> map = iterator.next();
+
+				Assert.assertNotNull(map);
+				Assert.assertNotNull(map.get("name"));
+				Assert.assertNotNull(map.get("gender"));
+				Assert.assertNotNull(map.get("age"));
+
+				System.out.printf("%15s (%5s): %s%n", map.get("name"), map.get("gender"), map.get("age").toString());
+			}
+
+			//
+			printHeadline("get query results in a list");
+			//
+
+			queryString = "FOR t IN " + COLLECTION_NAME
+					+ " FILTER t.age >= 20 && t.age < 30 && t.gender == @gender RETURN [t.name, t.gender, t.age]";
+			System.out.println(queryString);
+
+			@SuppressWarnings("rawtypes")
+			CursorResult<List> cursor2 = arangoDriver.executeAqlQuery(queryString, bindVars, aqlQueryOptions,
+				List.class);
+			Assert.assertNotNull(cursor2);
+
+			@SuppressWarnings("rawtypes")
+			Iterator<List> iterator2 = cursor2.iterator();
+			while (iterator2.hasNext()) {
+				List<?> list = iterator2.next();
+
+				Assert.assertNotNull(list);
+				Assert.assertNotNull(list.get(0));
+				Assert.assertNotNull(list.get(1));
+				Assert.assertNotNull(list.get(2));
+
+				System.out.printf("%15s (%5s): %s%n", list.get(0), list.get(1), list.get(2).toString());
+			}
+
+		} catch (ArangoException e) {
+			Assert.fail("Example failed. " + e.getMessage());
+		}
+
+	}
+
+	private void createExamples() throws ArangoException {
+		// create some persons
+		for (int i = 0; i < 100; i++) {
+			SimplePerson person = new SimplePerson();
+			person.setName("TestUser" + i);
+			person.setGender((i % 2) == 0 ? MALE : FEMALE);
+			person.setAge((int) (Math.random() * 100) + 10);
+
+			arangoDriver.createDocument(COLLECTION_NAME, person, true, null);
+		}
+
+	}
+
+}

src/test/java/com/arangodb/example/document/BaseExample.java

@@ -1,62 +1,97 @@
+/*
+ * Copyright (C) 2015 ArangoDB GmbH
+ *
+ * Licensed 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 com.arangodb.example.document;
 
+import org.junit.Assert;
+
 import com.arangodb.ArangoConfigure;
 import com.arangodb.ArangoDriver;
 import com.arangodb.ArangoException;
+import com.arangodb.entity.BooleanResultEntity;
 import com.arangodb.entity.CollectionEntity;
 
 public class BaseExample {
 
-	protected static ArangoDriver createDatabase(String name) {
+	protected static final String FEMALE = "female";
+
+	protected static final String MALE = "male";
 
+	protected ArangoConfigure getConfiguration() {
 		ArangoConfigure configure = new ArangoConfigure();
 		// configure.setUser("myUser");
 		// configure.setPassword("password");
 		// configuration file: src/test/resources/arangodb.properties
 		configure.init();
 
-		ArangoDriver arangoDriver = new ArangoDriver(configure);
+		return configure;
+	}
+
+	protected ArangoDriver getArangoDriver(ArangoConfigure configuration) {
+		return new ArangoDriver(configuration);
+	}
 
-		deleteDatabase(arangoDriver, name);
+	protected void removeTestDatabase(String name) {
+		ArangoDriver arangoDriver = getArangoDriver(getConfiguration());
+		try {
+			arangoDriver.deleteDatabase(name);
+		} catch (Exception e) {
+		}
+	}
 
+	protected void createDatabase(ArangoDriver arangoDriver, String name) {
 		try {
-			arangoDriver.createDatabase(name);
-			System.out.println("Database created: " + name);
+			BooleanResultEntity createDatabase = arangoDriver.createDatabase(name);
+			Assert.assertNotNull(createDatabase);
+			Assert.assertNotNull(createDatabase.getResult());
+			Assert.assertTrue(createDatabase.getResult());
 		} catch (Exception e) {
-			System.out.println("Failed to create database " + name + "; " + e.getMessage());
+			Assert.fail("Failed to create database " + name + "; " + e.getMessage());
 		}
 
 		arangoDriver.setDefaultDatabase(name);
-
-		return arangoDriver;
 	}
 
-	protected static void deleteDatabase(ArangoDriver arangoDriver, String name) {
+	protected void deleteDatabase(ArangoDriver arangoDriver, String name) {
 		try {
 			arangoDriver.deleteDatabase(name);
 		} catch (Exception e) {
 		}
 	}
 
-	protected static void createCollection(ArangoDriver arangoDriver, String name) {
+	protected void createCollection(ArangoDriver arangoDriver, String name) {
 		try {
 			CollectionEntity createCollection = arangoDriver.createCollection(name);
-			System.out.println(
-				"created collection '" + createCollection.getName() + "' with id '" + createCollection.getId() + "'");
+			Assert.assertNotNull(createCollection);
+			Assert.assertNotNull(createCollection.getName());
+			Assert.assertEquals(name, createCollection.getName());
 		} catch (ArangoException e) {
-			System.err.println(e.getMessage());
+			Assert.fail("create collection failed. " + e.getMessage());
 		}
 	}
 
-	protected static void printEntity(Object object) {
+	protected void printEntity(Object object) {
 		if (object == null) {
 			System.out.println("Document not found");
 		} else {
 			System.out.println(object);
 		}
 	}
 
-	protected static void printHeadline(String name) {
+	protected void printHeadline(String name) {
 		System.out.println("---------------------------------------------");
 		System.out.println(name);
 		System.out.println("---------------------------------------------");