update EnforcementFlowTest tests to supports updated cache retrieval …

…for _policy enrichment Signed-off-by: Aleksandar Stanchev <[email protected]>
thfries · Jan 15, 2024 · 242aa1d · 242aa1d
1 parent 96cdbd5
commit 242aa1d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/...rg/eclipse/ditto/thingsearch/service/persistence/write/streaming/EnforcementFlowTest.java b/...rg/eclipse/ditto/thingsearch/service/persistence/write/streaming/EnforcementFlowTest.java
@@ -253,6 +253,8 @@ public void computeThingCacheValueFromThingEvents() {
             sourceProbe.sendNext(inputMap);
             sourceProbe.sendComplete();
 
+            thingsProbe.expectMsgClass(Duration.apply(30, TimeUnit.SECONDS), SudoRetrieveThing.class);
+            thingsProbe.reply(SudoRetrieveThingResponse.of(JsonObject.empty(), DittoHeaders.empty()));
             // WHEN: policy is retrieved with up-to-date revisions
             policiesProbe.expectMsgClass(Duration.apply(30, TimeUnit.SECONDS), SudoRetrievePolicy.class);
             final var policy = Policy.newBuilder(policyId).setRevision(1).build();
@@ -319,6 +321,10 @@ public void computeThingCacheValueFromThingEventsWhenLastEventWasDeleted() {
             sourceProbe.sendNext(inputMap);
             sourceProbe.sendComplete();
 
+            // WHEN
+            thingsProbe.expectMsgClass(Duration.apply(30, TimeUnit.SECONDS), SudoRetrieveThing.class);
+            thingsProbe.reply(SudoRetrieveThingResponse.of(JsonObject.empty(), DittoHeaders.empty()));
+
             // THEN: the write model contains up-to-date revisions
             final AbstractWriteModel deleteModel = sinkProbe.expectNext().get(0);
             sinkProbe.expectComplete();