@Test(groups = {"acceptance"})
public void createAndListDb() throws RqlDriverException {
SecureRandom random = new SecureRandom();
String database = new BigInteger(130, random).toString(32);
RqlConnection r = RqlConnection.connect("localhost", 28015);
RqlCursor cursor = r.run(r.db_create(database));
RqlObject obj = cursor.next();
assert Double.valueOf(1.0).equals(obj.getAs("created"))
: "Database was not created successfully ";
cursor = r.run(r.db_list());
obj = cursor.next();
boolean found = false;
for (Object o : obj.getList()) {
if (database.equals(o)) {
found = true;
break;
}
}
assert found == true : "Database was not able to be listed";
cursor = r.run(r.db_drop(database));
obj = cursor.next();
assert Double.valueOf(1.0).equals(obj.getAs("dropped"))
: "Database was not dropped successfully ";
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();
}
@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 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();
}