water.junit.rules.tasks.ClearBeforeTestKeysTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-test-support Show documentation

null

The newest version!

package water.junit.rules.tasks;

import org.junit.Ignore;
import water.Key;
import water.MRTask;

import java.util.Arrays;

@Ignore
public class ClearBeforeTestKeysTask extends MRTask  {

  private final Key[] _retainedKeys;
  public ClearBeforeTestKeysTask(Key... retainedKeys) {
    _retainedKeys = retainedKeys;
  }

  @Override
  protected void setupLocal() {
    if (_retainedKeys.length > 0)
      LocalTestRuntime.beforeTestKeys.retainAll(Arrays.asList(_retainedKeys));
    else
      LocalTestRuntime.beforeTestKeys.clear();
  }
}