add ITs

Signed-off-by: see-quick <[email protected]> # Conflicts: # src/test/java/io/strimzi/test/container/StrimziKafkaContainerIT.java # src/test/java/io/strimzi/test/container/StrimziKafkaKraftContainerIT.java
strimzi · Nov 4, 2024 · 467d26f · 467d26f
1 parent c13a0b1
commit 467d26f
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/src/test/java/io/strimzi/test/container/StrimziKafkaContainerIT.java b/src/test/java/io/strimzi/test/container/StrimziKafkaContainerIT.java
@@ -18,6 +18,7 @@
 import org.apache.kafka.common.serialization.StringDeserializer;
 import org.apache.kafka.common.serialization.StringSerializer;
 import org.junit.jupiter.api.AfterEach;
+import org.apache.logging.log4j.Level;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.MethodSource;
 import org.slf4j.Logger;
@@ -316,6 +317,25 @@ void testGetProxyWithNoContainer(final String imageName) {
         systemUnderTest.stop();
     }
 
+    @Test
+    void testWithKafkaLog() {
+        systemUnderTest = new StrimziKafkaContainer()
+            .waitForRunning()
+            .withKafkaLog(Level.DEBUG);
+        systemUnderTest.start();
+
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("INFO"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("DEBUG"));
+
+        systemUnderTest.stop();
+        systemUnderTest.withKafkaLog(Level.TRACE);
+        systemUnderTest.start();
+
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("INFO"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("DEBUG"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("TRACE"));
+    }
+
     @Test
     void testKafkaContainerFunctionality() {
         // using try-with-resources for AdminClient, KafkaProducer and KafkaConsumer (implicit closing connection)

diff --git a/src/test/java/io/strimzi/test/container/StrimziKafkaKraftContainerIT.java b/src/test/java/io/strimzi/test/container/StrimziKafkaKraftContainerIT.java
@@ -14,6 +14,8 @@
 import org.apache.kafka.common.TopicPartition;
 import org.apache.kafka.common.serialization.StringDeserializer;
 import org.apache.kafka.common.serialization.StringSerializer;
+import org.apache.logging.log4j.Level;
+import org.hamcrest.CoreMatchers;
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
@@ -143,6 +145,26 @@ void testUnsupportedKRaftUsingImageName() {
         }
     }
 
+    @Test
+    void testWithKafkaLog() {
+        systemUnderTest = new StrimziKafkaContainer()
+            .waitForRunning()
+            .withKraft()
+            .withKafkaLog(Level.DEBUG);
+        systemUnderTest.start();
+
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("INFO"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("DEBUG"));
+
+        systemUnderTest.stop();
+        systemUnderTest.withKafkaLog(Level.TRACE);
+        systemUnderTest.start();
+
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("INFO"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("DEBUG"));
+        assertThat(systemUnderTest.getLogs(), CoreMatchers.containsString("TRACE"));
+    }
+
     private void verify() throws InterruptedException, ExecutionException, TimeoutException {
         final String topicName = "topic";