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

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

package se.scalablesolutions.akka.api;

import se.scalablesolutions.akka.kernel.config.*;
import static se.scalablesolutions.akka.kernel.config.JavaConfig.*;
import se.scalablesolutions.akka.kernel.actor.*;
import se.scalablesolutions.akka.kernel.Kernel;

import junit.framework.TestCase;

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

  final private ActiveObjectManager conf = new ActiveObjectManager();

  protected void setUp() {
    PersistenceManager.init();
    conf.configure(
        new RestartStrategy(new AllForOne(), 3, 5000),
        new Component[] {
          new Component(PersistentStateful.class, new LifeCycle(new Permanent(), 1000), 10000000),
          new Component(PersistentFailer.class, new LifeCycle(new Permanent(), 1000), 1000)
          //new Component(PersistentClasher.class, new LifeCycle(new Permanent(), 1000), 100000) 
        }).supervise();
  }

  protected void tearDown() {
    conf.stop();
  }

  public void testShouldNotRollbackStateForStatefulServerInCaseOfSuccess() {
    PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
    stateful.setMapState("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "init"); // set init state
    stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state"); // transactionrequired
    assertEquals("new state", stateful.getMapState("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess"));
  }

  public void testMapShouldRollbackStateForStatefulServerInCaseOfFailure() {
   PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
   stateful.setMapState("testShouldRollbackStateForStatefulServerInCaseOfFailure", "init"); // set init state
   PersistentFailer failer = conf.getInstance(PersistentFailer.class);
   try {
     stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer); // call failing transactionrequired method
     fail("should have thrown an exception");
   } catch (RuntimeException e) {
   } // expected
   assertEquals("init", stateful.getMapState("testShouldRollbackStateForStatefulServerInCaseOfFailure")); // check that state is == init state
 }

  public void testVectorShouldNotRollbackStateForStatefulServerInCaseOfSuccess() {
    PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
    stateful.setVectorState("init"); // set init state
    stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state"); // transactionrequired
    assertEquals("init", stateful.getVectorState(0));
    assertEquals("new state", stateful.getVectorState(1));
  }

  public void testVectorShouldRollbackStateForStatefulServerInCaseOfFailure() {
   PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
   stateful.setVectorState("init"); // set init state
   PersistentFailer failer = conf.getInstance(PersistentFailer.class);
   try {
     stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer); // call failing transactionrequired method
     fail("should have thrown an exception");
   } catch (RuntimeException e) {
   } // expected
   assertEquals("init", stateful.getVectorState(0)); // check that state is == init state
 }

  public void testRefShouldNotRollbackStateForStatefulServerInCaseOfSuccess() {
    PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
    stateful.setRefState("init"); // set init state
    stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state"); // transactionrequired
    assertEquals("new state", stateful.getRefState());
  }

  public void testRefShouldRollbackStateForStatefulServerInCaseOfFailure() {
   PersistentStateful stateful = conf.getInstance(PersistentStateful.class);
   stateful.setRefState("init"); // set init state
   PersistentFailer failer = conf.getInstance(PersistentFailer.class);
   try {
     stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer); // call failing transactionrequired method
     fail("should have thrown an exception");
   } catch (RuntimeException e) {
   } // expected
   assertEquals("init", stateful.getRefState()); // check that state is == init state
 }
}