Fix caching mechanism for Sagas

integrationtests/src/test/java/org/axonframework/integrationtests/cache/CachedSaga.java

@@ -0,0 +1,110 @@
+/*
+ * Copyright (c) 2010-2018. Axon Framework
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.axonframework.integrationtests.cache;
+
+import org.axonframework.modelling.saga.SagaEventHandler;
+import org.axonframework.modelling.saga.SagaLifecycle;
+import org.axonframework.modelling.saga.StartSaga;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Test saga used by the {@link CachingIntegrationTestSuite}.
+ *
+ * @author Steven van Beelen
+ */
+public class CachedSaga {
+
+    private String name;
+    private List<Object> state;
+
+    @StartSaga
+    @SagaEventHandler(associationProperty = "id")
+    public void on(SagaCreatedEvent event) {
+        this.name = event.name;
+        this.state = new ArrayList<>();
+    }
+
+    @SagaEventHandler(associationProperty = "id")
+    public void on(VeryImportantEvent event) {
+        state.add(event.stateEntry);
+    }
+
+    @SagaEventHandler(associationProperty = "id")
+    public void on(SagaEndsEvent event) {
+        if (event.shouldEnd) {
+            SagaLifecycle.end();
+        }
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public List<Object> getState() {
+        return state;
+    }
+
+    @SuppressWarnings({"FieldCanBeLocal", "unused"})
+    public static class SagaCreatedEvent {
+
+        private final String id;
+        private final String name;
+
+        public SagaCreatedEvent(String id, String name) {
+            this.id = id;
+            this.name = name;
+        }
+
+        public String getId() {
+            return id;
+        }
+    }
+
+    @SuppressWarnings({"FieldCanBeLocal", "unused"})
+    public static class VeryImportantEvent {
+
+        private final String id;
+        private final Object stateEntry;
+
+        public VeryImportantEvent(String id, Object stateEntry) {
+            this.id = id;
+            this.stateEntry = stateEntry;
+        }
+
+        public String getId() {
+            return id;
+        }
+    }
+
+    @SuppressWarnings({"FieldCanBeLocal", "unused"})
+    public static class SagaEndsEvent {
+
+        private final String id;
+        private final boolean shouldEnd;
+
+        public SagaEndsEvent(String id, boolean shouldEnd) {
+            this.id = id;
+            this.shouldEnd = shouldEnd;
+        }
+
+        public String getId() {
+            return id;
+        }
+    }
+}