pekko/akka-core/src/test/scala/InMemoryStateSpec.scala

/**
 * Copyright (C) 2009-2010 Scalable Solutions AB <http://scalablesolutions.se>
 */

package se.scalablesolutions.akka.actor

import org.scalatest.Spec
import org.scalatest.Assertions
import org.scalatest.matchers.ShouldMatchers
import org.scalatest.BeforeAndAfterAll
import org.scalatest.junit.JUnitRunner
import org.junit.runner.RunWith

import se.scalablesolutions.akka.config.Config
import se.scalablesolutions.akka.config._
import se.scalablesolutions.akka.config.ActiveObjectConfigurator
import se.scalablesolutions.akka.config.JavaConfig._
import se.scalablesolutions.akka.actor._

@RunWith(classOf[JUnitRunner])
class InMemoryStateSpec extends
  Spec with 
  ShouldMatchers with 
  BeforeAndAfterAll {  

  private val conf = new ActiveObjectConfigurator
  private var messageLog = ""
  
  override def beforeAll {
    Config.config
    conf.configure(
        new RestartStrategy(new AllForOne, 3, 5000, List(classOf[Exception]).toArray),
        List(
            new Component(classOf[InMemStateful],
                new LifeCycle(new Permanent),
                //new RestartCallbacks("preRestart", "postRestart")),
                10000),
            new Component(classOf[InMemFailer],
                new LifeCycle(new Permanent),
                10000)).toArray
        ).supervise
  }
  
  override def afterAll {
    conf.stop
  }

  describe("Transactional in-memory Active Object ") {

    it("map should not rollback state for stateful server in case of success") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setMapState("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "init")
      stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state")
      stateful.getMapState("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess") should equal("new state")
    }

    it("map should rollback state for stateful server in case of failure") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setMapState("testShouldRollbackStateForStatefulServerInCaseOfFailure", "init")
      val failer = conf.getInstance(classOf[InMemFailer])
      try {
        stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer) 
        fail("should have thrown an exception")
      } catch { case e => {} }
      stateful.getMapState("testShouldRollbackStateForStatefulServerInCaseOfFailure") should equal("init")
    }

    it("vector should rollback state for stateful server in case of failure") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setVectorState("init") // set init state
      val failer = conf.getInstance(classOf[InMemFailer])
      try {
        stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer)                    
        fail("should have thrown an exception")
      } catch { case e => {} }
      stateful.getVectorState should equal("init")
    }

    it("vector should not rollback state for stateful server in case of success") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setVectorState("init") // set init state
      stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state")
      stateful.getVectorState should equal("new state")
    }

    it("ref should rollback state for stateful server in case of failure") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setRefState("init") // set init state
      val failer = conf.getInstance(classOf[InMemFailer])
      try {
        stateful.failure("testShouldRollbackStateForStatefulServerInCaseOfFailure", "new state", failer) 
        fail("should have thrown an exception")
      } catch { case e => {} }
      stateful.getRefState should equal("init")
    }

    it("ref should not rollback state for stateful server in case of success") {
      val stateful = conf.getInstance(classOf[InMemStateful])
      stateful.init
      stateful.setRefState("init") // set init state
      stateful.success("testShouldNotRollbackStateForStatefulServerInCaseOfSuccess", "new state") 
      stateful.getRefState should equal("new state")
    }
  }
}