pekko/api-java/src/test/java/se/scalablesolutions/akka/api/PersistentStateTest.java

/**
 * Copyright (C) 2009 Scalable Solutions.
 */

package se.scalablesolutions.akka.api;

import se.scalablesolutions.akka.annotation.*;
import se.scalablesolutions.akka.kernel.configuration.*;
import se.scalablesolutions.akka.kernel.InMemoryState;

import com.google.inject.Inject;
import com.google.inject.AbstractModule;
import com.google.inject.Scopes;

import junit.framework.TestCase;


public class PersistentStateTest extends TestCase {
  static String messageLog = "";

  final private ActiveObjectGuiceConfigurator conf = new ActiveObjectGuiceConfigurator();

  protected void setUp() {
    conf.configureActiveObjects(
        new RestartStrategy(new AllForOne(), 3, 5000), new Component[]{
            new Component(
                Stateful.class,
                StatefulImpl.class,
                new LifeCycle(new Permanent(), 1000),
                10000000),
            new Component(
                Failer.class,
                FailerImpl.class,
                new LifeCycle(new Permanent(), 1000),
                1000),
            new Component(
                Clasher.class,
                ClasherImpl.class,
                new LifeCycle(new Permanent(), 1000),
                100000)
        }).inject().supervise();

  }

  public void testShouldNotRollbackStateForStatefulServerInCaseOfSuccess() {
    Stateful stateful = conf.getActiveObject(Stateful.class);
    stateful.setState("stateful", "init"); // set init state
    stateful.success("stateful", "new state"); // transactional
    assertEquals("new state", stateful.getState("stateful"));
  }

  interface Stateful {
    // transactional
    @transactional
    public void success(String key, String msg);

    @transactional
    public void failure(String key, String msg, Failer failer);

    @transactional
    public void clashOk(String key, String msg, Clasher clasher);

    @transactional
    public void clashNotOk(String key, String msg, Clasher clasher);

    // non-transactional
    public String getState(String key);

    public void setState(String key, String value);
  }

  class StatefulImpl implements Stateful {
    @state
    private InMemoryState<String, Object> state = new InMemoryState<String, Object>();

    public String getState(String key) {
      return (String) state.get(key);
    }

    public void setState(String key, String msg) {
      state.put(key, msg);
    }

    public void success(String key, String msg) {
      state.put(key, msg);
    }

    public void failure(String key, String msg, Failer failer) {
      state.put(key, msg);
      failer.fail();
    }

    public void clashOk(String key, String msg, Clasher clasher) {
      state.put(key, msg);
      clasher.clash();
    }

    public void clashNotOk(String key, String msg, Clasher clasher) {
      state.put(key, msg);
      clasher.clash();
      clasher.clash();
    }
  }

  interface Failer {
    public void fail();
  }

  class FailerImpl implements Failer {
    public void fail() {
      throw new RuntimeException("expected");
    }
  }

  interface Clasher {
    public void clash();

    public String getState(String key);

    public void setState(String key, String value);
  }

  class ClasherImpl implements Clasher {
    @state
    private InMemoryState<String, Object> state = new InMemoryState<String, Object>();

    public String getState(String key) {
      return (String) state.get(key);
    }

    public void setState(String key, String msg) {
      state.put(key, msg);
    }

    public void clash() {
      state.put("clasher", "was here");
      // spend some time here
      for (long i = 0; i < 1000000000; i++) {
        for (long j = 0; j < 10000000; j++) {
          j += i;
        }
      }

      // FIXME: this statement gives me this error:
      // se.scalablesolutions.akka.kernel.ActiveObjectException:
      //      Unexpected message [!(scala.actors.Channel@c2b2f6,ErrRef[Right(null)])] to
      //      [GenericServer[se.scalablesolutions.akka.api.StatefulImpl]] from
      //      [GenericServer[se.scalablesolutions.akka.api.ClasherImpl]]]
      //try { Thread.sleep(1000); } catch (InterruptedException e) {}
    }
  }
}