Added an option to create a fixture for a state stored aggregate

test/src/main/java/org/axonframework/test/aggregate/AggregateTestFixture.java

@@ -166,6 +166,16 @@ public final FixtureConfiguration<T> withSubtypes(Class<? extends T>... subtypes
         return this;
     }
 
+    @Override
+    public FixtureConfiguration<T> useStateStorage() {
+        return this.registerRepository(new InMemoryRepository<>(aggregateType,
+                                                                subtypes,
+                                                                eventStore,
+                                                                getParameterResolverFactory(),
+                                                                getHandlerDefinition(),
+                                                                getRepositoryProvider()));
+    }
+
     @Override
     public FixtureConfiguration<T> registerRepository(Repository<T> repository) {
         this.repository = new IdentifierValidatingRepository<>(repository);
@@ -314,12 +324,7 @@ public TestExecutor<T> givenState(Supplier<T> aggregate) {
         clearGivenWhenState();
         DefaultUnitOfWork.startAndGet(null).execute(() -> {
             if (repository == null) {
-                registerRepository(new InMemoryRepository<>(aggregateType,
-                                                            subtypes,
-                                                            eventStore,
-                                                            getParameterResolverFactory(),
-                                                            getHandlerDefinition(),
-                                                            getRepositoryProvider()));
+                this.useStateStorage();
             }
             try {
                 repository.newInstance(aggregate::get);

test/src/main/java/org/axonframework/test/aggregate/FixtureConfiguration.java

@@ -105,6 +105,14 @@ public interface FixtureConfiguration<T> {
      */
     FixtureConfiguration<T> withSubtypes(Class<? extends T>... subtypes);
 
+    /**
+     * Configures the fixture for state stored aggregates.
+     * This will register an {@link org.axonframework.test.aggregate.AggregateTestFixture.InMemoryRepository} with the fixture
+     *
+     * @return the current FixtureConfiguration, for fluent interfacing
+     */
+    FixtureConfiguration<T> useStateStorage();
+
     /**
      * Registers an arbitrary {@code repository} with the fixture. The repository must be wired
      * with the Event Store of this test fixture.

test/src/test/java/org/axonframework/test/aggregate/FixtureTest_StateStorage.java

@@ -61,6 +61,16 @@ void testCreateStateStoredAggregate() {
                .expectState(aggregate -> assertEquals("message2", aggregate.getMessage()));
     }
 
+    @Test
+    void testCreateStateStoredAggregateWithCommand() {
+        fixture
+                .useStateStorage()
+                .givenCommands(new InitializeCommand(AGGREGATE_ID, "message"))
+                .when(new SetMessageCommand(AGGREGATE_ID, "message2"))
+                .expectEvents(new StubDomainEvent())
+                .expectState(aggregate -> assertEquals("message2", aggregate.getMessage()));
+    }
+
     @Test
     void testEmittedEventsFromExpectStateAreNotStored() {
         fixture.givenState(() -> new StateStoredAggregate(AGGREGATE_ID, "message"))