Exemplo n.º 1
0
 @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();
 }
Exemplo n.º 2
0
  @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();
  }
Exemplo n.º 3
0
  @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();
  }
Exemplo n.º 4
0
  @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();
  }