flaky test

LangStream · May 20, 2024 · 72edb23 · 72edb23
1 parent 88f2240
commit 72edb23
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/langstream-api-gateway/src/test/java/ai/langstream/apigateway/http/GatewayResourceTest.java b/langstream-api-gateway/src/test/java/ai/langstream/apigateway/http/GatewayResourceTest.java
@@ -570,8 +570,7 @@ void testService() throws Exception {
                         url,
                         "{\"key\": \"my-key2\", \"value\": \"my-value\", \"headers\": {\"header1\":\"value1\"}}"));
 
-        // sorry but kafka can't keep up
-        final int numParallel = getStreamingCluster().type().equals("kafka") ? 5 : 30;
+        final int numParallel = 10;
 
         List<CompletableFuture<Void>> futures1 = new ArrayList<>();
         for (int i = 0; i < numParallel; i++) {
@@ -589,7 +588,7 @@ void testService() throws Exception {
             futures1.add(future);
         }
         CompletableFuture.allOf(futures1.toArray(new CompletableFuture[] {}))
-                .get(2, TimeUnit.MINUTES);
+                .get(3, TimeUnit.MINUTES);
     }
 
     private void startTopicExchange(String logicalFromTopic, String logicalToTopic)