lewismc commented on a change in pull request #179: GORA-532: Apache Gora
Benchmark initial pull request for review and comments
URL: https://github.com/apache/gora/pull/179#discussion_r310846993
##########
File path:
gora-benchmark/src/test/java/org/apache/gora/benchmark/GoraClientTest.java
##########
@@ -0,0 +1,201 @@
+package org.apache.gora.benchmark;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.HashSet;
+import java.util.List;
+import java.util.Properties;
+import java.util.Set;
+import java.util.Vector;
+
+import org.apache.gora.util.GoraException;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import com.yahoo.ycsb.ByteIterator;
+import com.yahoo.ycsb.StringByteIterator;
+import com.yahoo.ycsb.workloads.CoreWorkload;
+
+import generated.User;
+
+// TODO: Auto-generated Javadoc
+/**
+ * The Class GoraClientTest.
+ */
+public class GoraClientTest {
+
+
+ /** The Constant TABLE. */
+ private static final String TABLE = "users";
+
+ /** The client. */
+ private GoraBenchmarkClient client;
+
+ /** The data to insert. */
+ private static HashMap<String, ByteIterator> DATA_TO_INSERT;
+
+ /** The data to update. */
+ private static HashMap<String, ByteIterator> DATA_TO_UPDATE;
+
+ /** The data stores. */
+ private static List<String> dataStores = new ArrayList<String>();
+
+ /** The Constant NUMBER_OF_FIELDS. */
+ private static final int NUMBER_OF_FIELDS = 10;
+
+ /** The bmutils. */
+ private GoraBenchmarkUtils bmutils = new GoraBenchmarkUtils();
+
+ /**
+ * Sets the up.
+ *
+ * @throws Exception the exception
+ */
+ //Setup is executed before each test. Use @BeforeClass if you want to
execute a code block just once.
+ @Before
+ public void setUp() throws Exception {
+ //dataStores.add("mongodb");
+ //dataStores.add("hbase");
+ //bmutils.generateAvroSchema(NUMBER_OF_FIELDS);
+ //for(String dataStore: dataStores) {
+ //bmutils.generateMappingFile(dataStore);
+ //}
+ //bmutils.generateDataBeans();
+ DATA_TO_INSERT = new HashMap<>();
+ DATA_TO_UPDATE = new HashMap<>();
+ for(int i=0; i < NUMBER_OF_FIELDS; i++) {
+ DATA_TO_INSERT.put("field"+i, new StringByteIterator("value"+i));
+ DATA_TO_UPDATE.put("field"+i, new StringByteIterator("updated"+i));
+ }
+ Properties p = new Properties();
+ p.setProperty("key.class", "java.lang.String");
+ p.setProperty("persistent.class", "generated.User");
+ p.setProperty(CoreWorkload.FIELD_COUNT_PROPERTY, NUMBER_OF_FIELDS+"");
+ client = new GoraBenchmarkClient();
+ client.setProperties(p);
+ client.init();
+ }
+
+ /**
+ * Clean up.
+ *
+ * @throws Exception the exception
+ */
+ @After
+ public void cleanUp() throws Exception{
+ if(client != null)
+ client.cleanup();
+ client = null;
+ }
+
+ /**
+ * Read record.
+ *
+ * @param key the key
+ * @return the user
+ * @throws GoraException the gora exception
+ */
+ private User readRecord(String key) throws GoraException {
+ User u = client.dataStore.get(key);
+ return u;
+ }
+
+ /**
+ * Test client initialisation.
+ */
+ @Test
+ public void testClientInitialisation() {
+ assertNotNull(client.dataStore);
+ }
+
+
+ /**
+ * Test insert.
+ *
+ * @throws GoraException the gora exception
+ */
+ @Test
+ public void testInsert() throws GoraException{
+ int result1 = client.insert(TABLE, "key1", DATA_TO_INSERT);
+ int result2 = client.insert(TABLE, "key2", DATA_TO_INSERT);
+ int result3 = client.insert(TABLE, "key3", DATA_TO_INSERT);
+ assertEquals(0, result1);
+ assertEquals(0, result2);
+ assertEquals(0, result3);
+ }
+
+
+ /**
+ * Test read.
+ */
+ @Test
+ public void testRead() {
+ HashMap<String, ByteIterator> results = new HashMap<>();
+ Set<String> fields = new HashSet<>();//this could be null as well
+ //fields.add("field0");
+ int result = client.read(TABLE, "key1", fields, results);
+ assertEquals(0, result);
+ assertEquals(DATA_TO_INSERT.size(), results.size());
+ assertEquals(DATA_TO_INSERT.get("field0").toString(),
results.get("field0").toString());
+ assertEquals(DATA_TO_INSERT.get("field0").toString(), "value0");
+ }
+
+ /**
+ * Test scan.
+ */
+ @Test
+ public void testScan(){
+ Vector<HashMap<String, ByteIterator>> results = new Vector<HashMap<String,
ByteIterator>>();
+ Set<String> fields = new HashSet<>();
+ //fields.add("field0");
Review comment:
Remove
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
