From f3ccfa54b37fb123c9e4762976197c1b4b2792b4 Mon Sep 17 00:00:00 2001
From: Johan Haleby <johan.haleby@gmail.com>
Date: Fri, 3 May 2024 15:18:52 +0200
Subject: [PATCH] Improved CatchupSubscriptionModelTest

---
 .../catchup/CatchupSubscriptionModelTest.java | 19 ++++++-------------
 1 file changed, 6 insertions(+), 13 deletions(-)

diff --git a/subscription/util/blocking/catchup-subscription/src/test/java/org/occurrent/subscription/blocking/durable/catchup/CatchupSubscriptionModelTest.java b/subscription/util/blocking/catchup-subscription/src/test/java/org/occurrent/subscription/blocking/durable/catchup/CatchupSubscriptionModelTest.java
index bb3626fa9..569b044ba 100644
--- a/subscription/util/blocking/catchup-subscription/src/test/java/org/occurrent/subscription/blocking/durable/catchup/CatchupSubscriptionModelTest.java
+++ b/subscription/util/blocking/catchup-subscription/src/test/java/org/occurrent/subscription/blocking/durable/catchup/CatchupSubscriptionModelTest.java
@@ -153,25 +153,18 @@ void catchup_subscription_reads_events_that_were_persisted_during_catchup_phase(
         CountDownLatch waitForNewEventsToBePublished = new CountDownLatch(1);
 
         new Thread(() -> {
-            try {
-                waitBeforePublishingNewEvents.await(1, TimeUnit.MINUTES);
-                mongoEventStore.write("3", 0, serialize(nameDefined3));
-                mongoEventStore.write("4", 0, serialize(nameDefined4));
-                waitForNewEventsToBePublished.countDown();
-            } catch (InterruptedException e) {
-                throw new RuntimeException(e);
-            }
+            awaitLatch(waitBeforePublishingNewEvents);
+            awaitLatch(waitBeforePublishingNewEvents);
+            mongoEventStore.write("3", 0, serialize(nameDefined3));
+            mongoEventStore.write("4", 0, serialize(nameDefined4));
+            waitForNewEventsToBePublished.countDown();
         }).start();
 
         // When
         subscription.subscribe(UUID.randomUUID().toString(), StartAt.subscriptionPosition(TimeBasedSubscriptionPosition.beginningOfTime()), e -> {
             if (state.size() == 1) {
                 waitBeforePublishingNewEvents.countDown();
-                try {
-                    waitForNewEventsToBePublished.await(1, TimeUnit.MINUTES);
-                } catch (InterruptedException ex) {
-                    throw new RuntimeException(ex);
-                }
+                awaitLatch(waitForNewEventsToBePublished);
             }
             state.add(e);
         }).waitUntilStarted();