refactor(test-support): InMemoryFlamingockBulder and InMemoryAuditStore

Author: dieppa

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

@@ -0,0 +1,9 @@
+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,28 +15,28 @@
  */
 package io.flamingock.internal.core.builder;
 
-import io.flamingock.internal.core.plan.community.CommunityExecutionPlanner;
-import io.flamingock.internal.core.store.AuditStore;
-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;
     }
@@ -72,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

@@ -17,6 +17,7 @@
 
 import io.flamingock.internal.core.runner.Runner;
 import io.flamingock.internal.core.store.AuditStore;
+import io.flamingock.internal.core.store.CommunityAuditStore;
 
 public class BuilderAccessor {
 
@@ -35,4 +36,12 @@ public Runner 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/stages/GivenStage.java

@@ -15,6 +15,7 @@
  */
 package io.flamingock.support.stages;
 
+import io.flamingock.internal.core.store.CommunityAuditStore;
 import io.flamingock.support.FlamingockTestSupport;
 import io.flamingock.support.domain.AuditEntryDefinition;
 
@@ -72,6 +73,33 @@ public interface GivenStage {
      */
     GivenStage andExistingAudit(AuditEntryDefinition... definitions);
 
+    /**
+     * Overrides the audit store configured in the builder with a custom one.
+     *
+     * <p>Use this method when you need to use a different audit store than the one
+     * configured in the Flamingock builder. This is particularly useful for:</p>
+     * <ul>
+     *   <li>Testing with an in-memory audit store for isolation</li>
+     *   <li>Replacing a production audit store with a test-specific implementation</li>
+     * </ul>
+     *
+     * <h3>Example</h3>
+     * <pre>
+     * CommunityAuditStore testAuditStore = InMemoryAuditStore.create();
+     *
+     * FlamingockTestSupport.givenBuilder(builder)
+     *     .andOverrideAuditStore(testAuditStore)
+     *     .whenRun()
+     *     .thenExpectAuditFinalStateSequence(APPLIED(MyChange.class))
+     *     .verify();
+     * </pre>
+     *
+     * @param auditStore the audit store to use instead of the one configured in the builder
+     * @return this stage for method chaining
+     * @see CommunityAuditStore
+     */
+    GivenStage andOverrideAuditStore(CommunityAuditStore auditStore);
+
     /**
      * Completes the setup phase and transitions to the assertion phase.
      *

core/flamingock-test-support/src/main/java/io/flamingock/support/stages/GivenStageImpl.java

@@ -16,6 +16,7 @@
 package io.flamingock.support.stages;
 
 import io.flamingock.internal.core.builder.BuilderAccessor;
+import io.flamingock.internal.core.store.CommunityAuditStore;
 import io.flamingock.support.context.TestContext;
 import io.flamingock.support.domain.AuditEntryDefinition;
 
@@ -40,18 +41,16 @@ public GivenStage andExistingAudit(AuditEntryDefinition... definitions) {
         return this;
     }
 
+    @Override
+    public GivenStage andOverrideAuditStore(CommunityAuditStore auditStore) {
+        builderAccessor.overrideAuditStore(auditStore);
+        return this;
+    }
+
     @Override
     public WhenStage whenRun() {
         TestContext testContext = new TestContext(builderAccessor, existingAudit);
         return new WhenStageImpl(testContext);
     }
 
-    /**
-     * Returns the list of existing audit definitions.
-     *
-     * @return the existing audit definitions
-     */
-    public List<AuditEntryDefinition> getExistingAudit() {
-        return existingAudit;
-    }
 }