コード例 #1
0
  /**
   * Tests that an iterator can correctly iterate over a store.
   *
   * <p>There is no assumption about the order of iteration, it just makes sure all key-value pairs
   * are iterated.
   */
  @Test
  public void noOrderIteratorTest() throws Exception {
    List<TachyonURI> storeUris = Lists.newArrayList();
    List<List<KeyValuePair>> keyValuePairs = Lists.newArrayList();

    List<KeyValuePair> pairs = Lists.newArrayList();
    storeUris.add(createStoreOfSize(0, pairs));
    keyValuePairs.add(pairs);

    pairs = Lists.newArrayList();
    storeUris.add(createStoreOfSize(2, pairs));
    keyValuePairs.add(pairs);

    pairs = Lists.newArrayList();
    storeUris.add(createStoreOfMultiplePartitions(3, pairs));
    keyValuePairs.add(pairs);

    int numStoreUri = storeUris.size();
    for (int i = 0; i < numStoreUri; i++) {
      List<KeyValuePair> expectedPairs = keyValuePairs.get(i);
      List<KeyValuePair> iteratedPairs = Lists.newArrayList();
      mReader = sKeyValueSystem.openStore(storeUris.get(i));
      KeyValueIterator iterator = mReader.iterator();
      while (iterator.hasNext()) {
        iteratedPairs.add(iterator.next());
      }

      // If size is not the same, no need for the time-consuming list comparison below.
      Assert.assertEquals(expectedPairs.size(), iteratedPairs.size());
      // Sorts and then compares pairs and iteratedPairs.
      Collections.sort(expectedPairs);
      Collections.sort(iteratedPairs);
      Assert.assertEquals(expectedPairs, iteratedPairs);
    }
  }
コード例 #2
0
  private void testMergeStore(
      TachyonURI store1,
      List<KeyValuePair> keyValuePairs1,
      TachyonURI store2,
      List<KeyValuePair> keyValuePairs2)
      throws Exception {
    sKeyValueSystem.mergeStore(store1, store2);

    // store2 contains all key-value pairs in both store1 and store2.
    List<KeyValuePair> mergedPairs = Lists.newArrayList();
    mergedPairs.addAll(keyValuePairs1);
    mergedPairs.addAll(keyValuePairs2);

    List<KeyValuePair> store2Pairs = Lists.newArrayList();
    KeyValueIterator iterator = sKeyValueSystem.openStore(store2).iterator();
    while (iterator.hasNext()) {
      store2Pairs.add(iterator.next());
    }

    // If size is not the same, no need for the time-consuming list comparison below.
    Assert.assertEquals(mergedPairs.size(), store2Pairs.size());
    Collections.sort(mergedPairs);
    Collections.sort(store2Pairs);
    Assert.assertEquals(mergedPairs, store2Pairs);

    // store1 no longer exists, because it has been merged into store2.
    // TachyonException is expected to be thrown.
    try {
      sKeyValueSystem.openStore(store1);
    } catch (TachyonException e) {
      Assert.assertEquals(e.getMessage(), ExceptionMessage.PATH_DOES_NOT_EXIST.getMessage(store1));
      return;
    }
    Assert.assertTrue("The URI to the deleted key-value store still exists", false);
  }
コード例 #3
0
  /** Tests that an iterator for an empty store has no next elements. */
  @Test
  public void emptyStoreIteratorTest() throws Exception {
    mWriter = sKeyValueSystem.createStore(mStoreUri);
    mWriter.close();

    mReader = sKeyValueSystem.openStore(mStoreUri);
    KeyValueIterator iterator = mReader.iterator();
    Assert.assertFalse(iterator.hasNext());
  }