@Test(groups = {"acceptance"}) public void createAndListTable() throws RqlDriverException { SecureRandom random = new SecureRandom(); String database = new BigInteger(130, random).toString(32); String table = new BigInteger(130, random).toString(32); RqlConnection r = RqlConnection.connect("localhost", 28015); r.run(r.db_create(database)); RqlCursor cursor = r.run(r.db(database).table_create(table)); assert Double.valueOf(1.0).equals(cursor.next().getAs("created")) : "Table was not created successfully "; cursor = r.run(r.db(database).table_list()); boolean found = false; for (Object o : cursor.next().getList()) { if (table.equals(o)) { found = true; break; } } assert found == true : "Table was not able to be listed"; cursor = r.run(r.db(database).table_drop(table)); assert Double.valueOf(1.0).equals(cursor.next().getAs("dropped")) : "Table was not dropped successfully "; r.run(r.db_drop(database)); r.close(); }
@SuppressWarnings({"unchecked", "rawtypes", "serial"}) @Test(groups = {"acceptance"}) public void insertAndRetrieveSingleRow() throws RqlDriverException { SecureRandom random = new SecureRandom(); String database = new BigInteger(130, random).toString(32); String table = new BigInteger(130, random).toString(32); RqlConnection r = RqlConnection.connect("localhost", 28015); r.run(r.db_create(database)); r.run(r.db(database).table_create(table)); RqlCursor cursor = r.run( r.db(database) .table(table) .insert( Arrays.asList( new HashMap() { { put("name", "Worf"); put("show", "Star Trek TNG"); } }, new HashMap() { { put("name", "Data"); put("show", "Star Trek TNG"); } }, new HashMap() { { put("name", "William Adama"); put("show", "Battlestar Galactica"); } }, new HashMap() { { put("name", "Homer Simpson"); put("show", "The Simpsons"); } }))); List generatedKeys = cursor.next().getAs("generated_keys"); Iterator it = generatedKeys.iterator(); while (it.hasNext()) { cursor = r.run(r.db(database).table(table).get(it.next())); for (RqlObject o : cursor) { assert o.toString() != null : "Failed to get single row"; } } r.run(r.db(database).table_drop(table)); r.run(r.db_drop(database)); r.close(); }
@SuppressWarnings({"unchecked", "rawtypes", "serial"}) @Test(groups = {"benchmark"}) public void largeDataSetTest() throws RqlDriverException { final double rowsPerIteration = 8192.0; final double numberOfIterations = 256.0; final SecureRandom random = new SecureRandom(); String database = new BigInteger(130, random).toString(32); String table = new BigInteger(130, random).toString(32); RqlConnection r = RqlConnection.connect("localhost", 28015); r.run(r.db_create(database)); r.run(r.db(database).table_create(table)); for (long j = 0; j < numberOfIterations; j++) { List<Object> l = new ArrayList<Object>(); System.out.println("Inserting Rows..."); for (long i = 0; i < rowsPerIteration; i++) { final long id = (i * (long) rowsPerIteration) + j; l.add( new HashMap() { { put("id", BigInteger.valueOf(id).toString(32)); put("name", new BigInteger(64, random).toString(32)); } }); } RqlCursor cursor = r.run(r.db(database).table(table).insert(l)); RqlObject obj = cursor.next(); Double result = obj.getAs("inserted"); assert result == rowsPerIteration : "Error inserting Data into Database on iteration " + j + " (" + result + " did not equal " + rowsPerIteration + ")"; } RqlCursor cursor = r.run(r.db(database).table(table).count()); assert Double.valueOf(numberOfIterations * rowsPerIteration).equals(cursor.next().getNumber()) : "Error getting large row count"; cursor = r.run(r.db(database).table(table)); long rowCount = 0; for (RqlObject o : cursor) { rowCount++; } assert rowCount == (rowsPerIteration * numberOfIterations) : "Error fetching all rows in large dataset"; System.out.println("We got " + rowCount + " results back "); r.run(r.db(database).table_drop(table)); r.run(r.db_drop(database)); r.close(); }
@SuppressWarnings({"unchecked", "rawtypes", "serial"}) @Test(groups = {"acceptance"}) public void insertAndRetrieveData() throws RqlDriverException { SecureRandom random = new SecureRandom(); String database = new BigInteger(130, random).toString(32); String table = new BigInteger(130, random).toString(32); RqlConnection r = RqlConnection.connect("localhost", 28015); r.run(r.db_create(database)); r.run(r.db(database).table_create(table)); RqlCursor cursor = r.run( r.db(database) .table(table) .insert( Arrays.asList( new HashMap() { { put("name", "Worf"); put("show", "Star Trek TNG"); } }, new HashMap() { { put("name", "Data"); put("show", "Star Trek TNG"); } }, new HashMap() { { put("name", "William Adama"); put("show", "Battlestar Galactica"); } }, new HashMap() { { put("name", "Homer Simpson"); put("show", "The Simpsons"); } }))); assert Double.valueOf(4.0).equals(cursor.next().getAs("inserted")) : "Error inserting Data into Database"; cursor = r.run( r.db(database) .table(table) .filter( new HashMap() { { put("show", "Star Trek TNG"); } })); // We Expect Two results int count = 0; for (RqlObject o : cursor) { Map<String, Object> m = o.getMap(); assert m.containsKey("name") : "Data that came back was malformed (missing \"name\")"; assert m.containsKey("show") : "Data that came back was malformed (missing \"show\")"; assert "Star Trek TNG".equals(m.get("show")) : "Data that came back was just plain wrong (\"show\" was not \"Star Trek TNG\");"; count++; } cursor = r.run( r.db(database) .table(table) .filter( new HashMap() { { put("name", "donald duck"); put("show", "Disney show"); } }) .is_empty()); Boolean isEmpty = null; for (RqlObject o : cursor) { isEmpty = o.getBoolean(); } assert isEmpty == true : "Failed at verifying query resultset is empty."; cursor = r.run(r.db(database).table(table).count()); double rowCount = 0; for (RqlObject o : cursor) { rowCount = o.getNumber(); } assert rowCount == 4.0 : "Failed at getting the correct row count."; r.run(r.db(database).table_drop(table)); r.run(r.db_drop(database)); r.close(); }