package extra;

import static avian.Continuations.callWithCurrentContinuation;

import avian.Function;
import avian.Callback;

public class Continuations {
  private static void expect(boolean v) {
    if (! v) throw new RuntimeException();
  }

  public static void main(String[] args) throws Exception {
    expect
      (callWithCurrentContinuation
       (new Function<Callback<Integer>,Integer>() {
         public Integer call(Callback<Integer> continuation) {
           continuation.handleResult(42);
           throw new AssertionError();
         }
       }) == 42);

    expect
      (callWithCurrentContinuation
       (new Function<Callback<Integer>,Integer>() {
         public Integer call(Callback<Integer> continuation) {
           return 43;
         }
       }) == 43);

    try {
      callWithCurrentContinuation(new Function<Callback<Integer>,Integer>() {
          public Integer call(Callback<Integer> continuation) {
            continuation.handleException(new MyException());
            throw new AssertionError();
          }
        });
      throw new AssertionError();      
    } catch (MyException e) {
      e.printStackTrace();
    }

    try {
      callWithCurrentContinuation(new Function<Callback<Integer>,Integer>() {
          public Integer call(Callback<Integer> continuation)
            throws MyException
          {
            throw new MyException();
          }
        });
      throw new AssertionError();
    } catch (MyException e) {
      e.printStackTrace();
    }
  }

  private static class MyException extends Exception { }
}