UnlockTest.java

package water;

import org.junit.Assert;
import org.junit.Test;
import water.api.UnlockKeys;
import water.fvec.Frame;
import water.fvec.NFSFileVec;
import water.fvec.ParseDataset2;
import water.util.Log;

public class UnlockTest extends TestUtil {

  @Test
  public void run(){
    // Put chunks into KV store
    String dataset = "smalldata/logreg/syn_2659x1049.csv";
    Key file = NFSFileVec.make(find_test_file(dataset));
    Frame f = ParseDataset2.parse(Key.make(), new Key[]{file});
    // Create two lockable frames in KV store
    Frame fr1 = new Frame(Key.make(), f.names(), f.vecs());
    Frame fr2 = new Frame(Key.make(), f.names(), f.vecs());
    // Lock the frames against writes
    fr1.delete_and_lock(null);
    fr2.delete_and_lock(null);
    int i = 0;
    try {
      // try to delete the write-locked frames -> will throw an exception
      fr1.delete();
      fr2.delete(); // won't be reached
    } catch(Throwable t) {
      Log.info("Correctly unable to delete (was locked): " + t.getClass()); //either AssertionError if local or DistributedException if remote
      i++;
    } finally {
      // second attempt: will unlock and delete properly
      new UnlockKeys().serve(); // without this line, there will be a leak (and assertion won't be shown)
      fr1.delete();
      fr2.delete();
      f.delete();
      Log.info("Able to delete after unlocking.");
    }
    Assert.assertTrue(i == 1);
  }

}