protected volatile boolean throwException = false;

  protected final ListenableFutureTask<Integer> task =
      ListenableFutureTask.create(
          new Callable<Integer>() {
            @Override
            public Integer call() throws Exception {
              runLatch.countDown();
              taskLatch.await();
              if (throwException) {
                throw new IllegalStateException("Fail");

        }

        public void setPurgeCallCount(long count) {
            this.purgeCallCount = count;
        }

        public void setThrowException(boolean throwException) {
            this.throwException = throwException;
        }

        public void setExceptionMessage(String message) {
            this.exceptionMessage = message;
        }

        public long getPurgeExpiry() {

  protected volatile boolean throwException = false;

  protected final ListenableFutureTask<Integer> task =
      ListenableFutureTask.create(
          new Callable<Integer>() {
            @Override
            public Integer call() throws Exception {
              runLatch.countDown();
              taskLatch.await();
              if (throwException) {
                throw new IllegalStateException("Fail");

        queryRescorer = new QueryRescorer() {
            @Override
            public RescorerBuilder<?> evaluate(Map<String, Object> params) {
                if (params != null && params.containsKey("throwException")) {
                    throw new RuntimeException("Test exception");
                }
                return createMockRescorerBuilder();
            }
        };

        // Test normal case