feat(simulator-spring-boot): add MessageDirectionFilter and test

citrusframework · Jan 22, 2025 · 403700f · 403700f
1 parent bcb593c
commit 403700f
Show file tree

Hide file tree

Showing 3 changed files with 37 additions and 0 deletions.
diff --git a/...ot/src/main/java/org/citrusframework/simulator/service/ScenarioExecutionQueryService.java b/...ot/src/main/java/org/citrusframework/simulator/service/ScenarioExecutionQueryService.java
@@ -258,6 +258,15 @@ protected Specification<ScenarioExecution> createSpecification(ScenarioExecution
                         )
                     );
             }
+            if (nonNull(criteria.getScenarioMessagesDirection())) {
+                specification =
+                    specification.and(
+                        buildSpecification(
+                            criteria.getScenarioMessagesDirection(),
+                            root -> root.join(ScenarioExecution_.scenarioMessages, JoinType.LEFT).get(Message_.direction)
+                        )
+                    );
+            }
             if (nonNull(criteria.getScenarioParametersId())) {
                 specification =
                     specification.and(

diff --git a/...c/main/java/org/citrusframework/simulator/service/criteria/ScenarioExecutionCriteria.java b/...c/main/java/org/citrusframework/simulator/service/criteria/ScenarioExecutionCriteria.java
@@ -66,6 +66,8 @@ public class ScenarioExecutionCriteria implements Serializable, Criteria {
 
     private @Nullable LongFilter scenarioMessagesId;
 
+    private @Nullable IntegerFilter scenarioMessagesDirection;
+
     private @Nullable LongFilter scenarioParametersId;
 
     private @Nullable String headers;
@@ -85,6 +87,7 @@ public ScenarioExecutionCriteria(ScenarioExecutionCriteria other) {
         this.status = other.status == null ? null : other.status.copy();
         this.scenarioActionsId = other.scenarioActionsId == null ? null : other.scenarioActionsId.copy();
         this.scenarioMessagesId = other.scenarioMessagesId == null ? null : other.scenarioMessagesId.copy();
+        this.scenarioMessagesDirection = other.scenarioMessagesDirection == null ? null : other.scenarioMessagesDirection.copy();
         this.scenarioParametersId = other.scenarioParametersId == null ? null : other.scenarioParametersId.copy();
         this.headers = other.headers;
         this.scenarioMessagesPayload = other.scenarioMessagesPayload;
@@ -110,6 +113,7 @@ public boolean equals(Object o) {
             .append(status, scenarioExecutionCriteria.status)
             .append(scenarioActionsId, scenarioExecutionCriteria.scenarioActionsId)
             .append(scenarioMessagesId, scenarioExecutionCriteria.scenarioMessagesId)
+            .append(scenarioMessagesDirection, scenarioExecutionCriteria.scenarioMessagesDirection)
             .append(scenarioParametersId, scenarioExecutionCriteria.scenarioParametersId)
             .append(headers, scenarioExecutionCriteria.headers)
             .append(scenarioMessagesPayload, scenarioExecutionCriteria.scenarioMessagesPayload)
@@ -127,6 +131,7 @@ public int hashCode() {
             .append(status)
             .append(scenarioActionsId)
             .append(scenarioMessagesId)
+            .append(scenarioMessagesDirection)
             .append(scenarioParametersId)
             .append(headers)
             .append(scenarioMessagesPayload)

diff --git a/...oot/src/test/java/org/citrusframework/simulator/web/rest/ScenarioExecutionResourceIT.java b/...oot/src/test/java/org/citrusframework/simulator/web/rest/ScenarioExecutionResourceIT.java
@@ -398,6 +398,29 @@ void getAllScenarioExecutionsByScenarioMessagesIsEqualToSomething() throws Excep
         defaultScenarioExecutionShouldNotBeFound("scenarioMessagesId.equals=" + (scenarioMessagesId + 1));
     }
 
+    @Test
+    @Transactional
+    void getAllScenarioExecutionsByScenarioMessageDirectionIsEqualToSomething() throws Exception {
+        Message scenarioMessages;
+        if (TestUtil.findAll(entityManager, Message.class).isEmpty()) {
+            scenarioExecutionRepository.saveAndFlush(scenarioExecution);
+            scenarioMessages = MessageResourceIT.createEntity(entityManager);
+        } else {
+            scenarioMessages = TestUtil.findAll(entityManager, Message.class).get(0);
+        }
+        scenarioMessages.setDirection(Message.Direction.INBOUND);
+        entityManager.persist(scenarioMessages);
+        entityManager.flush();
+        scenarioExecution.addScenarioMessage(scenarioMessages);
+        scenarioExecutionRepository.saveAndFlush(scenarioExecution);
+        int scenarioMessageDirection = scenarioMessages.getDirection().getId();
+        // Get all the scenarioExecutionList where scenarioMessagesDirection equals to scenarioMessagesDirection
+        defaultScenarioExecutionShouldBeFound("scenarioMessagesDirection.equals=" + scenarioMessageDirection);
+
+        // Get all the scenarioExecutionList where scenarioMessagesDirection equals to (scenarioMessagesDirection + 1)
+        defaultScenarioExecutionShouldNotBeFound("scenarioMessagesDirection.equals=" + (Message.Direction.UNKNOWN.getId()));
+    }
+
     @Test
     @Transactional
     void getAllScenarioExecutionsByScenarioMessagesPayloadIsEqualToSomething() throws Exception {