From e9360fdb6617dc7df93ffb58b744bbc9d108e380 Mon Sep 17 00:00:00 2001
From: Simone Bordet <simone.bordet@gmail.com>
Date: Tue, 2 Feb 2021 13:32:57 +0100
Subject: [PATCH] Fixes #56 - "ready" event not emitted without warmup
 iterations.

Now emitting the event when warmupIterations=0.

Signed-off-by: Simone Bordet <simone.bordet@gmail.com>
---
 .../jetty/load/generator/LoadGenerator.java   | 10 ++++++---
 .../load/generator/LoadGeneratorTest.java     | 21 +++++++++++++++++++
 2 files changed, 28 insertions(+), 3 deletions(-)

diff --git a/jetty-load-generator-client/src/main/java/org/mortbay/jetty/load/generator/LoadGenerator.java b/jetty-load-generator-client/src/main/java/org/mortbay/jetty/load/generator/LoadGenerator.java
index 88557dd0..646a8171 100644
--- a/jetty-load-generator-client/src/main/java/org/mortbay/jetty/load/generator/LoadGenerator.java
+++ b/jetty-load-generator-client/src/main/java/org/mortbay/jetty/load/generator/LoadGenerator.java
@@ -1144,8 +1144,8 @@ private class WarmupCallback extends Callback.Nested {
         public WarmupCallback(Callback callback, int warmupIterations) {
             super(callback);
             latch = new CountDownLatch(warmupIterations == 0 ? 0 : 1);
-            counter = warmupIterations == 0 ? Callback.from(NOOP::succeeded, callback::failed) :
-                    new CountingCallback(Callback.from(this::success, this::failure), warmupIterations);
+            // If there are no warmup iterations, the callback will never be invoked.
+            counter = warmupIterations == 0 ? NOOP : new CountingCallback(Callback.from(this::success, this::failure), warmupIterations);
         }
 
         @Override
@@ -1160,6 +1160,9 @@ public void failed(Throwable x) {
 
         public void join() {
             try {
+                if (counter == NOOP) {
+                    fireReadyEvent();
+                }
                 latch.await();
             } catch (InterruptedException x) {
                 throw new RuntimeException(x);
@@ -1179,7 +1182,8 @@ private void success() {
                     LOGGER.debug("awaiting barrier for ready listener");
                 }
                 awaitBarrier();
-                // Do not succeed the nested callback, as these are just warmup iterations.
+                // Do not forward success the nested callback,
+                // as these are just warmup iterations.
             } finally {
                 latch.countDown();
             }
diff --git a/jetty-load-generator-client/src/test/java/org/mortbay/jetty/load/generator/LoadGeneratorTest.java b/jetty-load-generator-client/src/test/java/org/mortbay/jetty/load/generator/LoadGeneratorTest.java
index 073b534b..97c078cf 100644
--- a/jetty-load-generator-client/src/test/java/org/mortbay/jetty/load/generator/LoadGeneratorTest.java
+++ b/jetty-load-generator-client/src/test/java/org/mortbay/jetty/load/generator/LoadGeneratorTest.java
@@ -505,6 +505,27 @@ public void testReadyEvent() throws Exception {
         Assert.assertEquals(threads * count, resources.get());
     }
 
+    @Test
+    public void testReadyEventWithoutWarmup() throws Exception {
+        startServer(new TestHandler());
+
+        int threads = 2;
+        int count = 30;
+        CountDownLatch readyLatch = new CountDownLatch(1);
+        LoadGenerator loadGenerator = LoadGenerator.builder()
+                .port(connector.getLocalPort())
+                .httpClientTransportBuilder(clientTransportBuilder)
+                .threads(threads)
+                .iterationsPerThread(count)
+                .resourceRate(0)
+                .listener((LoadGenerator.ReadyListener)g -> readyLatch.countDown())
+                .build();
+
+        loadGenerator.begin().get();
+
+        Assert.assertTrue(readyLatch.await(5, TimeUnit.SECONDS));
+    }
+
     private enum TransportType {
         H1C, H2C
     }