From e30306f6f5a483ef46545c5e2582a835aae56f0d Mon Sep 17 00:00:00 2001 From: Jeff Xiang Date: Thu, 12 Dec 2024 21:10:24 -0500 Subject: [PATCH] Force close pscProducer in testTransactionalProducerWithKafkaBackendCannotBeNonTransactional after 1 second --- .../pinterest/psc/integration/producer/TestOneKafkaBackend.java | 2 +- psc/src/main/java/com/pinterest/psc/producer/PscProducer.java | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/psc-integration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java b/psc-integration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java index b7bedf7..397cdc7 100644 --- a/psc-integration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java +++ b/psc-integration-test/src/test/java/com/pinterest/psc/integration/producer/TestOneKafkaBackend.java @@ -412,7 +412,7 @@ public void testTransactionalProducerWithKafkaBackendCannotBeNonTransactional() PscProducerMessage 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 index c38d44c..4c07443 100644 --- 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)); } /**