Force close pscProducer in testTransactionalProducerWithKafkaBackendC…

…annotBeNonTransactional after 1 second
pinterest · Dec 13, 2024 · e30306f · e30306f
1 parent 7406f35
commit e30306f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...ration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java b/...ration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java
@@ -412,7 +412,7 @@ public void testTransactionalProducerWithKafkaBackendCannotBeNonTransactional()
         PscProducerMessage<Integer, Integer> producerMessage = new PscProducerMessage<>(topicUriStr1, 0);
         Exception e = assertThrows(ProducerException.class, () -> pscProducer.send(producerMessage));
         assertEquals(IllegalStateException.class, e.getCause().getClass());
-        pscProducer.close();
+        pscProducer.close(Duration.ofSeconds(1));
     }
 
     /**

diff --git a/psc/src/main/java/com/pinterest/psc/producer/PscProducer.java b/psc/src/main/java/com/pinterest/psc/producer/PscProducer.java
@@ -772,7 +772,7 @@ public void flush() throws ProducerException {
      */
     @Override
     public void close() throws IOException {
-        close(Duration.ofMillis(0L));
+        close(Duration.ofMillis(Long.MAX_VALUE));
     }
 
     /**