Improve feed throttler and tests

1 files changed, 30 insertions, 0 deletions

diff --git a/vespa-feed-client/src/test/java/ai/vespa/feed/client/impl/DynamicThrottlerTest.java b/vespa-feed-client/src/test/java/ai/vespa/feed/client/impl/DynamicThrottlerTest.java
new file mode 100644
index 00000000000..7e07fc6e116
--- /dev/null
+++ b/vespa-feed-client/src/test/java/ai/vespa/feed/client/impl/DynamicThrottlerTest.java
@@ -0,0 +1,30 @@
+package ai.vespa.feed.client.impl;
+
+import org.junit.jupiter.api.Test;
+
+import java.net.URI;
+import java.util.List;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+/**
+ * @author jonmv
+ */
+class DynamicThrottlerTest {
+
+    @Test
+    void testThrottler() {
+        DynamicThrottler throttler = new DynamicThrottler(new FeedClientBuilderImpl(List.of(URI.create("http://localhost:8080"))));
+        assertEquals(16, throttler.targetInflight());
+
+        for (int i = 0; i < 30; i++) {
+            throttler.sent(1, null);
+            throttler.success();
+        }
+        assertEquals(18, throttler.targetInflight());
+
+        throttler.throttled(34);
+        assertEquals(17, throttler.targetInflight());
+    }
+
+}