initial draft of cassandra storage

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

@@ -0,0 +1,152 @@
+/**
+ * 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) {}
+    }
+  }
+}