changed example class to unit test class

Author: a-brandt

src/test/java/com/arangodb/bench/AbstractBenchmarkImporter.java

@@ -0,0 +1,26 @@
+package com.arangodb.bench;
+
+import java.util.List;
+
+import com.arangodb.ArangoDriver;
+
+public abstract class AbstractBenchmarkImporter {
+
+	protected final ArangoDriver driver;
+
+	protected final String collectionName;
+
+	public AbstractBenchmarkImporter(ArangoDriver driver, String collectionName) {
+		this.driver = driver;
+		this.collectionName = collectionName;
+	}
+
+	abstract protected void execute(List<?> values) throws Exception;
+
+	public long bench(List<?> values) throws Exception {
+		long t = System.currentTimeMillis();
+		execute(values);
+		return System.currentTimeMillis() - t;
+	}
+
+}

src/test/java/com/arangodb/bench/BenchmarkImport.java

@@ -18,95 +18,82 @@
 
 import java.util.List;
 
-import com.arangodb.ArangoConfigure;
+import org.junit.Assert;
+import org.junit.Before;
+import org.junit.Test;
+
 import com.arangodb.ArangoDriver;
 import com.arangodb.ArangoException;
 import com.arangodb.Station;
+import com.arangodb.example.document.BaseExample;
 import com.arangodb.util.TestUtils;
 
 /**
+ * Import a list of objects
+ * 
+ * 1. Import the data with importDocuments()
+ * 
+ * 2. Import the data with createDocument()
+ * 
  * @author tamtam180 - kirscheless at gmail.com
+ * @author a-brandt
  *
  */
-public class BenchmarkImport {
-
-	final static String COLLECTION_NAME = "bench-test";
+public class BenchmarkImport extends BaseExample {
 
-	public static void main(String[] args) throws Exception {
+	private static final String DATABASE_NAME = "BenchmarkImport";
 
-		final int max = 10;
-		ArangoConfigure configure = new ArangoConfigure();
-		configure.init();
-
-		try {
+	private static final String COLLECTION_NAME = "BenchmarkImportCollection";
 
-			ArangoDriver driver = new ArangoDriver(configure);
-			List<Station> stations = TestUtils.readStations();
+	public ArangoDriver arangoDriver;
 
-			// truncate collection
-			try {
-				driver.truncateCollection(COLLECTION_NAME);
-			} catch (ArangoException e) {
-			}
+	@Before
+	public void _before() {
+		removeTestDatabase(DATABASE_NAME);
 
-			// Bench import
-			BenchLogic logic1 = new BenchImport(driver);
-			BenchLogic logic2 = new BenchDocument(driver);
-
-			// Bench create document
-			long time1 = 0, time2 = 0;
-			for (int i = 0; i < max; i++) {
-				time1 += logic1.bench(stations);
-				time2 += logic2.bench(stations);
-			}
+		arangoDriver = getArangoDriver(getConfiguration());
+		createDatabase(arangoDriver, DATABASE_NAME);
+	}
 
-			System.out.println("import:" + time1);
-			System.out.println("document:" + time2);
+	@Test
+	public void BenchmarkImportTest() throws Exception {
 
-		} finally {
-			configure.shutdown();
-		}
+		final int max = 10;
 
-	}
+		//
+		printHeadline("read example data");
+		//
 
-	static private abstract class BenchLogic {
-		protected ArangoDriver driver;
+		List<Station> stations = TestUtils.readStations();
 
-		public BenchLogic(ArangoDriver driver) {
-			this.driver = driver;
+		// truncate collection
+		try {
+			arangoDriver.truncateCollection(COLLECTION_NAME);
+		} catch (ArangoException e) {
 		}
 
-		abstract protected void execute(List<?> values) throws Exception;
+		// create importer
+		AbstractBenchmarkImporter logic1 = new ImportDocumentBenchmarkImporter(arangoDriver, COLLECTION_NAME);
+		AbstractBenchmarkImporter logic2 = new SingleDocumentBenchmarkImporter(arangoDriver, COLLECTION_NAME);
 
-		public long bench(List<?> values) throws Exception {
-			long t = System.currentTimeMillis();
-			execute(values);
-			return System.currentTimeMillis() - t;
-		}
-	}
+		//
+		printHeadline("import data");
+		//
 
-	static class BenchImport extends BenchLogic {
-		public BenchImport(ArangoDriver driver) {
-			super(driver);
+		// Bench import and create document
+		long time1 = 0, time2 = 0;
+		for (int i = 0; i < max; i++) {
+			time1 += logic1.bench(stations);
+			time2 += logic2.bench(stations);
 		}
 
-		@Override
-		protected void execute(List<?> values) throws Exception {
-			driver.importDocuments(COLLECTION_NAME, true, values);
-		}
-	}
-
-	static class BenchDocument extends BenchLogic {
-		public BenchDocument(ArangoDriver driver) {
-			super(driver);
-		}
+		//
+		printHeadline("results");
+		//
+		System.out.println("importDocuments(): " + time1 + " ms");
+		System.out.println("createDocument():  " + time2 + " ms");
 
-		@Override
-		protected void execute(List<?> values) throws Exception {
-			for (Object value : values) {
-				driver.createDocument("bench-test", value, true, false);
-			}
-		}
+		Assert.assertTrue(time1 < time2);
 	}
 
 }

src/test/java/com/arangodb/bench/ImportDocumentBenchmarkImporter.java

@@ -0,0 +1,24 @@
+package com.arangodb.bench;
+
+import java.util.List;
+
+import com.arangodb.ArangoDriver;
+
+/**
+ * Import all values with importDocuments()
+ * 
+ * @author a-brandt
+ *
+ */
+public class ImportDocumentBenchmarkImporter extends AbstractBenchmarkImporter {
+
+	public ImportDocumentBenchmarkImporter(ArangoDriver driver, String collectionName) {
+		super(driver, collectionName);
+	}
+
+	@Override
+	protected void execute(List<?> values) throws Exception {
+		driver.importDocuments(collectionName, true, values);
+	}
+
+}

src/test/java/com/arangodb/bench/SingleDocumentBenchmarkImporter.java

@@ -0,0 +1,26 @@
+package com.arangodb.bench;
+
+import java.util.List;
+
+import com.arangodb.ArangoDriver;
+
+/**
+ * Import all values with createDocument()
+ * 
+ * @author a-brandt
+ *
+ */
+public class SingleDocumentBenchmarkImporter extends AbstractBenchmarkImporter {
+
+	public SingleDocumentBenchmarkImporter(ArangoDriver driver, String collectionName) {
+		super(driver, collectionName);
+	}
+
+	@Override
+	protected void execute(List<?> values) throws Exception {
+		for (Object value : values) {
+			driver.createDocument(collectionName, value, true, false);
+		}
+	}
+
+}

src/test/java/com/arangodb/example/ExamplesTestSuite.java

@@ -20,6 +20,7 @@
 import org.junit.runners.Suite;
 import org.junit.runners.Suite.SuiteClasses;
 
+import com.arangodb.bench.BenchmarkImport;
 import com.arangodb.example.document.DocumentExamplesTestSuite;
 import com.arangodb.example.graph.GraphExamplesTestSuite;
 
@@ -32,11 +33,13 @@
 @RunWith(Suite.class)
 @SuiteClasses({
 
-		TransactionExample.class,
-
 		DocumentExamplesTestSuite.class,
 
-		GraphExamplesTestSuite.class
+		GraphExamplesTestSuite.class,
+
+		BenchmarkImport.class,
+
+		TransactionExample.class
 
 })