exceptioncontainer.java

来自「基于Junit的 功能和单元测试的的测试工具。只支持Swing.」· Java 代码 · 共 42 行

package org.uispec4j.utils;

public class ExceptionContainer {
  private RuntimeException exception;
  private Error error;

  public void set(Throwable e) {
    if (e instanceof RuntimeException) {
      exception = (RuntimeException)e;
    }
    else if (e instanceof Error) {
      error = (Error)e;
    }
    else {
      exception = new RuntimeException(e);
    }
  }

  public boolean isSet() {
    return (exception != null) || (error != null);
  }

  public void rethrowIfNeeded() {
    try {
      if (error != null) {
        throw error;
      }
      if (exception != null) {
        throw exception;
      }
    }
    finally {
      reset();
    }
  }

  public void reset() {
    exception = null;
    error = null;
  }
}