Skip to content

Commit 7885de2

Browse files
committed
Move duplicated code into KCatContainer
1 parent 6e104da commit 7885de2

File tree

3 files changed

+21
-31
lines changed

3 files changed

+21
-31
lines changed

modules/kafka/src/test/java/org/testcontainers/containers/KafkaContainerTest.java

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
import org.awaitility.Awaitility;
1212
import org.junit.Test;
1313
import org.testcontainers.AbstractKafka;
14+
import org.testcontainers.KCatContainer;
1415
import org.testcontainers.Testcontainers;
15-
import org.testcontainers.images.builder.Transferable;
1616
import org.testcontainers.utility.DockerImageName;
1717
import org.testcontainers.utility.MountableFile;
1818

@@ -195,13 +195,7 @@ public void testUsageWithListener() throws Exception {
195195
.withNetwork(network);
196196
// }
197197
// createKCatContainer {
198-
GenericContainer<?> kcat = new GenericContainer<>("confluentinc/cp-kcat:7.9.0")
199-
.withCreateContainerCmdModifier(cmd -> {
200-
cmd.withEntrypoint("sh");
201-
})
202-
.withCopyToContainer(Transferable.of("Message produced by kcat"), "/data/msgs.txt")
203-
.withNetwork(network)
204-
.withCommand("-c", "tail -f /dev/null")
198+
KCatContainer kcat = new KCatContainer().withNetwork(network)
205199
// }
206200
) {
207201
kafka.start();
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
package org.testcontainers.redpanda;
2+
3+
import org.testcontainers.containers.GenericContainer;
4+
import org.testcontainers.images.builder.Transferable;
5+
6+
public class KCatContainer extends GenericContainer<KCatContainer> {
7+
8+
public KCatContainer() {
9+
super("confluentinc/cp-kcat:7.9.0");
10+
withCreateContainerCmdModifier(cmd -> {
11+
cmd.withEntrypoint("sh");
12+
});
13+
withCopyToContainer(Transferable.of("Message produced by kcat"), "/data/msgs.txt");
14+
withCommand("-c", "tail -f /dev/null");
15+
}
16+
}

modules/redpanda/src/test/java/org/testcontainers/redpanda/RedpandaContainerTest.java

Lines changed: 3 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,8 @@
1313
import org.apache.kafka.common.errors.TopicAuthorizationException;
1414
import org.awaitility.Awaitility;
1515
import org.junit.Test;
16-
import org.testcontainers.containers.GenericContainer;
1716
import org.testcontainers.containers.Network;
1817
import org.testcontainers.containers.SocatContainer;
19-
import org.testcontainers.images.builder.Transferable;
2018
import org.testcontainers.utility.DockerImageName;
2119

2220
import java.util.Collection;
@@ -111,13 +109,7 @@ public void testUsageWithListener() throws Exception {
111109
RedpandaContainer redpanda = new RedpandaContainer("docker.redpanda.com/redpandadata/redpanda:v23.1.7")
112110
.withListener(() -> "redpanda:19092")
113111
.withNetwork(network);
114-
GenericContainer<?> kcat = new GenericContainer<>("confluentinc/cp-kcat:7.9.0")
115-
.withCreateContainerCmdModifier(cmd -> {
116-
cmd.withEntrypoint("sh");
117-
})
118-
.withCopyToContainer(Transferable.of("Message produced by kcat"), "/data/msgs.txt")
119-
.withNetwork(network)
120-
.withCommand("-c", "tail -f /dev/null")
112+
KCatContainer kcat = new KCatContainer().withNetwork(network)
121113
) {
122114
redpanda.start();
123115
kcat.start();
@@ -141,13 +133,7 @@ public void testUsageWithListenerInTheSameNetwork() throws Exception {
141133
.withNetwork(network);
142134
// }
143135
// createKCatContainer {
144-
GenericContainer<?> kcat = new GenericContainer<>("confluentinc/cp-kcat:7.9.0")
145-
.withCreateContainerCmdModifier(cmd -> {
146-
cmd.withEntrypoint("sh");
147-
})
148-
.withCopyToContainer(Transferable.of("Message produced by kcat"), "/data/msgs.txt")
149-
.withNetwork(network)
150-
.withCommand("-c", "tail -f /dev/null")
136+
KCatContainer kcat = new KCatContainer().withNetwork(network)
151137
// }
152138
) {
153139
kafka.start();
@@ -200,13 +186,7 @@ public void testUsageWithListenerAndSasl() throws Exception {
200186
.withSuperuser("panda")
201187
.withListener("my-panda:29092")
202188
.withNetwork(network);
203-
GenericContainer<?> kcat = new GenericContainer<>("confluentinc/cp-kcat:7.9.0")
204-
.withCreateContainerCmdModifier(cmd -> {
205-
cmd.withEntrypoint("sh");
206-
})
207-
.withCopyToContainer(Transferable.of("Message produced by kcat"), "/data/msgs.txt")
208-
.withNetwork(network)
209-
.withCommand("-c", "tail -f /dev/null")
189+
KCatContainer kcat = new KCatContainer().withNetwork(network)
210190
) {
211191
redpanda.start();
212192

0 commit comments

Comments
 (0)