feat(test-support): inMemory Builder and auditStore (#773)

Author: dieppa

build.gradle.kts

@@ -16,7 +16,7 @@ plugins {
 
 allprojects {
     group = "io.flamingock"
-    version = "1.0.0-beta.6"
+    version = "1.0.0-beta.7"
 
     repositories {
         mavenCentral()

core/flamingock-core/src/main/java/io/flamingock/internal/core/builder/AuditStoreReceiver.java

@@ -0,0 +1,24 @@
+/*
+ * Copyright 2025 Flamingock (https://www.flamingock.io)
+ *
+ * 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 io.flamingock.internal.core.builder;
+
+
+import io.flamingock.internal.core.store.CommunityAuditStore;
+
+public interface AuditStoreReceiver<RETURN_TYPE> {
+
+    RETURN_TYPE setAuditStore(CommunityAuditStore auditStore);
+}

core/flamingock-core/src/main/java/io/flamingock/internal/core/builder/CommunityChangeRunnerBuilder.java

@@ -15,31 +15,41 @@
  */
 package io.flamingock.internal.core.builder;
 
-import io.flamingock.internal.core.plan.community.CommunityExecutionPlanner;
-import io.flamingock.internal.core.store.CommunityAuditStore;
-import io.flamingock.internal.core.configuration.core.CoreConfiguration;
+import io.flamingock.internal.common.core.context.Context;
 import io.flamingock.internal.core.configuration.community.CommunityConfiguration;
 import io.flamingock.internal.core.configuration.community.CommunityConfigurator;
-import io.flamingock.internal.common.core.context.Context;
+import io.flamingock.internal.core.configuration.core.CoreConfiguration;
 import io.flamingock.internal.core.plan.ExecutionPlanner;
+import io.flamingock.internal.core.plan.community.CommunityExecutionPlanner;
 import io.flamingock.internal.core.plugin.PluginManager;
+import io.flamingock.internal.core.store.CommunityAuditStore;
 import io.flamingock.internal.util.id.RunnerId;
 
 public class CommunityChangeRunnerBuilder
         extends AbstractChangeRunnerBuilder<CommunityAuditStore, CommunityChangeRunnerBuilder>
-        implements CommunityConfigurator<CommunityChangeRunnerBuilder> {
+        implements CommunityConfigurator<CommunityChangeRunnerBuilder>,
+        AuditStoreReceiver<CommunityChangeRunnerBuilder> {
 
     private final CommunityConfiguration communityConfiguration;
 
 
     public CommunityChangeRunnerBuilder(CoreConfiguration coreConfiguration,
-                                           CommunityConfiguration communityConfiguration,
-                                           Context dependencyInjectableContext,
-                                           PluginManager pluginManager) {
+                                        CommunityConfiguration communityConfiguration,
+                                        Context dependencyInjectableContext,
+                                        PluginManager pluginManager) {
         super(coreConfiguration, dependencyInjectableContext, pluginManager);
         this.communityConfiguration = communityConfiguration;
     }
 
+    public CommunityChangeRunnerBuilder(CoreConfiguration coreConfiguration,
+                                        CommunityConfiguration communityConfiguration,
+                                        Context dependencyInjectableContext,
+                                        PluginManager pluginManager,
+                                        CommunityAuditStore auditStore) {
+        super(coreConfiguration, dependencyInjectableContext, pluginManager, auditStore);
+        this.communityConfiguration = communityConfiguration;
+    }
+
     @Override
     protected CommunityChangeRunnerBuilder getSelf() {
         return this;
@@ -62,6 +72,7 @@ protected ExecutionPlanner buildExecutionPlanner(RunnerId runnerId) {
                 .build();
     }
 
+    @Override
     public CommunityChangeRunnerBuilder setAuditStore(CommunityAuditStore auditStore) {
         this.auditStore = auditStore;
         return this;

core/flamingock-test-support/src/main/java/io/flamingock/internal/core/builder/BuilderAccessor.java

@@ -15,7 +15,9 @@
  */
 package io.flamingock.internal.core.builder;
 
+import io.flamingock.internal.core.runner.Runner;
 import io.flamingock.internal.core.store.AuditStore;
+import io.flamingock.internal.core.store.CommunityAuditStore;
 
 public class BuilderAccessor {
 
@@ -29,10 +31,17 @@ public AuditStore<?> getAuditStore(){
         return builder.auditStore;
     }
 
-
-    public void run() {
-        builder.build().run();
+    public Runner build() {
+        return builder.build();
     }
 
 
+    public void overrideAuditStore(CommunityAuditStore auditStore) {
+        if(!(builder instanceof AuditStoreReceiver)) {
+            String message = String.format("Builder of type[%s] does not allow AuditStore override operation", builder.getClass().getSimpleName());
+            throw new IllegalArgumentException(message);
+        }
+
+        ((AuditStoreReceiver<?>)builder).setAuditStore(auditStore);
+    }
 }

core/flamingock-test-support/src/main/java/io/flamingock/support/context/TestContext.java

@@ -18,7 +18,9 @@
 import io.flamingock.internal.common.core.audit.AuditReader;
 import io.flamingock.internal.common.core.audit.AuditWriter;
 import io.flamingock.internal.core.builder.BuilderAccessor;
+import io.flamingock.internal.core.runner.Runner;
 import io.flamingock.support.domain.AuditEntryDefinition;
+import io.flamingock.support.inmemory.InMemoryAuditStore;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -75,10 +77,12 @@ public AuditWriter getAuditWriter() {
     }
 
     /**
-     * Runs the change runner by building and executing it.
+     * Returns the change runner by building it from the builder.
+     *
+     * @return the built runner ready for execution
      */
-    public void run() {
-        builderAccessor.run();
+    public Runner build() {
+        return builderAccessor.build();
     }
 
     /**
@@ -89,4 +93,10 @@ public void run() {
     public List<AuditEntryDefinition> getPreconditions() {
         return Collections.unmodifiableList(preconditions);
     }
+
+    public void cleanUp() {
+        if(builderAccessor.getAuditStore() instanceof InMemoryAuditStore) {
+            ((InMemoryAuditStore)builderAccessor.getAuditStore()).cleanUp();
+        }
+    }
 }

core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditPersistence.java

@@ -0,0 +1,43 @@
+/*
+ * Copyright 2025 Flamingock (https://www.flamingock.io)
+ *
+ * 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 io.flamingock.support.inmemory;
+
+import io.flamingock.internal.common.core.audit.AuditEntry;
+import io.flamingock.internal.core.store.audit.community.CommunityAuditPersistence;
+import io.flamingock.internal.util.Result;
+
+import java.util.List;
+
+class InMemoryAuditPersistence implements CommunityAuditPersistence {
+    
+    private final InMemoryAuditWriter auditWriter;
+    private final InMemoyAuditReader auditReader;
+    
+    public InMemoryAuditPersistence(InMemoryAuditStorage auditStorage) {
+        this.auditWriter = new InMemoryAuditWriter(auditStorage);
+        this.auditReader = new InMemoyAuditReader(auditStorage);
+    }
+
+    @Override
+    public List<AuditEntry> getAuditHistory() {
+        return auditReader.getAuditHistory();
+    }
+
+    @Override
+    public Result writeEntry(AuditEntry auditEntry) {
+        return auditWriter.writeEntry(auditEntry);
+    }
+}

…e/kit/inmemory/InMemoryAuditStorage.java …pport/inmemory/InMemoryAuditStorage.javautils/test-util/src/main/java/io/flamingock/core/kit/inmemory/InMemoryAuditStorage.java renamed to core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditStorage.java utils/test-util/src/main/java/io/flamingock/core/kit/inmemory/InMemoryAuditStorage.java renamed to core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditStorage.java

@@ -13,9 +13,8 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package io.flamingock.core.kit.inmemory;
+package io.flamingock.support.inmemory;
 
-import io.flamingock.core.kit.audit.AuditStorage;
 import io.flamingock.internal.common.core.audit.AuditEntry;
 
 import java.util.ArrayList;
@@ -35,11 +34,9 @@
  * InMemoryAuditStorage storage = new InMemoryAuditStorage();
  * AuditTestHelper helper = new AuditTestHelper(storage);
  * }</pre>
- * 
- * <p><strong>For other storage implementations:</strong> Use this as a reference
- * when implementing AuditStorage for MongoDB, DynamoDB, etc.</p>
+ *
  */
-public class InMemoryAuditStorage implements AuditStorage {
+class InMemoryAuditStorage {
 
     private final List<AuditEntry> auditEntries = new ArrayList<>();
 

…kit/inmemory/InMemoryTestAuditStore.java …support/inmemory/InMemoryAuditStore.javautils/test-util/src/main/java/io/flamingock/core/kit/inmemory/InMemoryTestAuditStore.java renamed to core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditStore.java utils/test-util/src/main/java/io/flamingock/core/kit/inmemory/InMemoryTestAuditStore.java renamed to core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditStore.java

@@ -13,23 +13,27 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package io.flamingock.core.kit.inmemory;
+package io.flamingock.support.inmemory;
 
 import io.flamingock.internal.common.core.context.ContextResolver;
-import io.flamingock.internal.core.store.lock.community.CommunityLockService;
 import io.flamingock.internal.core.store.CommunityAuditStore;
+import io.flamingock.internal.core.store.lock.community.CommunityLockService;
 import io.flamingock.internal.util.id.RunnerId;
 
-public class InMemoryTestAuditStore implements CommunityAuditStore {
+public class InMemoryAuditStore implements CommunityAuditStore {
 
     private final InMemoryAuditStorage auditStorage;
     private final InMemoryLockStorage lockStorage;
-    private InMemoryTestAuditPersistence persistence;
+    private InMemoryAuditPersistence persistence;
     private RunnerId runnerId;
 
-    public InMemoryTestAuditStore(InMemoryAuditStorage auditStorage, InMemoryLockStorage lockStorage) {
-        this.auditStorage = auditStorage;
-        this.lockStorage = lockStorage;
+    public static InMemoryAuditStore create() {
+        return new InMemoryAuditStore();
+    }
+
+    private InMemoryAuditStore() {
+        this.auditStorage = new InMemoryAuditStorage();
+        this.lockStorage = new InMemoryLockStorage();
 
     }
 
@@ -38,12 +42,12 @@ public void initialize(ContextResolver contextResolver) {
         // Extract required components from context
         runnerId = contextResolver.getRequiredDependencyValue(RunnerId.class);
         // Create the test audit persistence with domain-separated storage
-        this.persistence = new InMemoryTestAuditPersistence(auditStorage);
+        this.persistence = new InMemoryAuditPersistence(auditStorage);
     }
 
 
     @Override
-    public InMemoryTestAuditPersistence getPersistence() {
+    public InMemoryAuditPersistence getPersistence() {
         if (persistence == null) {
             throw new IllegalStateException("AuditStore not initialized - call initialize first");
         }
@@ -54,4 +58,9 @@ public InMemoryTestAuditPersistence getPersistence() {
     public CommunityLockService getLockService() {
         return new InMemoryLockService(lockStorage, runnerId);
     }
+
+    public void cleanUp() {
+        auditStorage.clear();
+        lockStorage.clear();
+    }
 }

core/flamingock-test-support/src/main/java/io/flamingock/support/inmemory/InMemoryAuditWriter.java

@@ -0,0 +1,74 @@
+/*
+ * Copyright 2025 Flamingock (https://www.flamingock.io)
+ *
+ * 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 io.flamingock.support.inmemory;
+
+import io.flamingock.internal.common.core.audit.AuditEntry;
+import io.flamingock.internal.core.store.audit.LifecycleAuditWriter;
+import io.flamingock.internal.core.store.audit.domain.ExecutionAuditContextBundle;
+import io.flamingock.internal.core.store.audit.domain.RollbackAuditContextBundle;
+import io.flamingock.internal.core.store.audit.domain.StartExecutionAuditContextBundle;
+import io.flamingock.internal.util.Result;
+
+class InMemoryAuditWriter implements LifecycleAuditWriter {
+
+    private final InMemoryAuditStorage auditStorage;
+
+    public InMemoryAuditWriter(InMemoryAuditStorage auditStorage) {
+        this.auditStorage = auditStorage;
+    }
+    
+    @Override
+    public Result writeStartExecution(StartExecutionAuditContextBundle auditContextBundle) {
+        try {
+            AuditEntry auditEntry = auditContextBundle.toAuditEntry();
+            return writeEntry(auditEntry);
+        } catch (Exception e) {
+            return new Result.Error(e);
+        }
+    }
+    
+    @Override
+    public Result writeExecution(ExecutionAuditContextBundle auditContextBundle) {
+        try {
+            AuditEntry auditEntry = auditContextBundle.toAuditEntry();
+            return writeEntry(auditEntry);
+        } catch (Exception e) {
+            return new Result.Error(e);
+        }
+    }
+    
+    @Override
+    public Result writeRollback(RollbackAuditContextBundle auditContextBundle) {
+        try {
+            AuditEntry auditEntry = auditContextBundle.toAuditEntry();
+            return writeEntry(auditEntry);
+        } catch (Exception e) {
+            return new Result.Error(e);
+        }
+    }
+    
+    /**
+     * Write an audit entry to the storage
+     */
+    public Result writeEntry(AuditEntry auditEntry) {
+        try {
+            auditStorage.addAuditEntry(auditEntry);
+            return Result.OK();
+        } catch (Exception e) {
+            return new Result.Error(e);
+        }
+    }
+}