Update tests for more stability

pelotoncycle · Oct 15, 2024 · 9287e58 · 9287e58
1 parent 32d77e1
commit 9287e58
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/test/kotlin/com/onepeloton/locust4k/examples/ExampleAppTest.kt b/src/test/kotlin/com/onepeloton/locust4k/examples/ExampleAppTest.kt
@@ -28,7 +28,7 @@ class ExampleAppTest : TestcontainersBase() {
 
             // when
             LogMessageWaitStrategy()
-                .withRegEx(".* Receive Message producer ZMQ socket closed.*\\s")
+                .withRegEx(".* Shutting down Locust Worker.*\\s")
                 .withStartupTimeout(Duration.ofSeconds(5))
                 .waitUntilReady(locustWorkerExampleContainer)
 
@@ -53,8 +53,6 @@ class ExampleAppTest : TestcontainersBase() {
                     "Stats consumer stopped",
                     "Stats reporter stopped",
                     "Shutting down Locust Worker",
-                    "Closing connection",
-                    "Receive Message producer ZMQ socket closed",
                 ),
             )
         } finally {
@@ -83,7 +81,7 @@ class ExampleAppTest : TestcontainersBase() {
 
             // when
             LogMessageWaitStrategy()
-                .withRegEx(".* Receive Message producer ZMQ socket closed.*\\s")
+                .withRegEx(".* Shutting down Locust Worker.*\\s")
                 .withStartupTimeout(Duration.ofSeconds(5))
                 .waitUntilReady(locustWorkerExampleContainer)
 
@@ -113,8 +111,6 @@ class ExampleAppTest : TestcontainersBase() {
                     "Stats consumer stopped",
                     "Stats reporter stopped",
                     "Shutting down Locust Worker",
-                    "Closing connection",
-                    "Receive Message producer ZMQ socket closed",
                 ),
             )
         } finally {