From b8a6e4f89fe0408f3ffa98acb5d79fbbf75c5fb3 Mon Sep 17 00:00:00 2001
From: Constantin <constantin.ursache.dev@gmail.com>
Date: Thu, 21 Dec 2023 16:57:31 +0200
Subject: [PATCH] BAEL-7199: Include feedback

---
 .../spring/kafka/dlt/KafkaDltRetryOnErrorIntegrationTest.java    | 1 +
 .../com/baeldung/spring/kafka/dlt/KafkaNoDltIntegrationTest.java | 1 +
 2 files changed, 2 insertions(+)

diff --git a/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaDltRetryOnErrorIntegrationTest.java b/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaDltRetryOnErrorIntegrationTest.java
index 97e42dac9f..393eb2ae46 100644
--- a/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaDltRetryOnErrorIntegrationTest.java
+++ b/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaDltRetryOnErrorIntegrationTest.java
@@ -83,5 +83,6 @@ public class KafkaDltRetryOnErrorIntegrationTest {
 
         assertThat(mainTopicCountDownLatch.await(5, TimeUnit.SECONDS)).isTrue();
         assertThat(dlTTopicCountDownLatch.await(5, TimeUnit.SECONDS)).isTrue();
+        assertThat(dlTTopicCountDownLatch.getCount()).isEqualTo(0);
     }
 }
\ No newline at end of file
diff --git a/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaNoDltIntegrationTest.java b/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaNoDltIntegrationTest.java
index 568908d276..81cca9fec3 100644
--- a/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaNoDltIntegrationTest.java
+++ b/spring-kafka-2/src/test/java/com/baeldung/spring/kafka/dlt/KafkaNoDltIntegrationTest.java
@@ -82,5 +82,6 @@ public class KafkaNoDltIntegrationTest {
 
         assertThat(mainTopicCountDownLatch.await(5, TimeUnit.SECONDS)).isTrue();
         assertThat(dlTTopicCountDownLatch.await(5, TimeUnit.SECONDS)).isFalse();
+        assertThat(dlTTopicCountDownLatch.getCount()).isEqualTo(1);
     }
 }
\ No newline at end of file