diff --git a/impl/persistence/tests/src/main/java/io/serverlessworkflow/impl/persistence/test/AbstractHandlerPersistenceTest.java b/impl/persistence/tests/src/main/java/io/serverlessworkflow/impl/persistence/test/AbstractHandlerPersistenceTest.java index bd57fc59..9cdb9a75 100644 --- a/impl/persistence/tests/src/main/java/io/serverlessworkflow/impl/persistence/test/AbstractHandlerPersistenceTest.java +++ b/impl/persistence/tests/src/main/java/io/serverlessworkflow/impl/persistence/test/AbstractHandlerPersistenceTest.java @@ -39,6 +39,7 @@ import java.time.Instant; import java.util.Map; import java.util.Optional; +import java.util.stream.Stream; import org.junit.jupiter.api.AfterAll; import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.BeforeAll; @@ -142,6 +143,9 @@ void testWorkflowInstance() throws InterruptedException { .writer() .taskCompleted(workflowContext, completedTaskContext(position2, completedMap)) .join(); + try (Stream stream = handlers.reader().scanAll(definition)) { + assertThat(stream.count()).isEqualTo(1); + } instance = (WorkflowPersistenceInstance) handlers.reader().find(definition, workflowInstance.id()).orElseThrow(); @@ -165,5 +169,8 @@ void testWorkflowInstance() throws InterruptedException { // workflow completed handlers.writer().completed(workflowContext).join(); assertThat(handlers.reader().find(definition, workflowInstance.id())).isEmpty(); + try (Stream stream = handlers.reader().scanAll(definition)) { + assertThat(stream.count()).isEqualTo(0); + } } }