Skip to content

Commit 88c86cd

Browse files
committed
Move duplicated code into KCatContainer
1 parent 6e104da commit 88c86cd

File tree

3 files changed

+21
-24
lines changed

3 files changed

+21
-24
lines changed

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

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
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;
1516
import org.testcontainers.images.builder.Transferable;
1617
import org.testcontainers.utility.DockerImageName;
@@ -195,13 +196,8 @@ public void testUsageWithListener() throws Exception {
195196
.withNetwork(network);
196197
// }
197198
// 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")
199+
KCatContainer kcat = new KCatContainer()
203200
.withNetwork(network)
204-
.withCommand("-c", "tail -f /dev/null")
205201
// }
206202
) {
207203
kafka.start();
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
package org.testcontainers;
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 & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -111,13 +111,8 @@ public void testUsageWithListener() throws Exception {
111111
RedpandaContainer redpanda = new RedpandaContainer("docker.redpanda.com/redpandadata/redpanda:v23.1.7")
112112
.withListener(() -> "redpanda:19092")
113113
.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")
114+
KCatContainer kcat = new KCatContainer()
119115
.withNetwork(network)
120-
.withCommand("-c", "tail -f /dev/null")
121116
) {
122117
redpanda.start();
123118
kcat.start();
@@ -141,13 +136,8 @@ public void testUsageWithListenerInTheSameNetwork() throws Exception {
141136
.withNetwork(network);
142137
// }
143138
// 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")
139+
KCatContainer kcat = new KCatContainer()
149140
.withNetwork(network)
150-
.withCommand("-c", "tail -f /dev/null")
151141
// }
152142
) {
153143
kafka.start();
@@ -200,13 +190,8 @@ public void testUsageWithListenerAndSasl() throws Exception {
200190
.withSuperuser("panda")
201191
.withListener("my-panda:29092")
202192
.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")
193+
KCatContainer kcat = new KCatContainer()
208194
.withNetwork(network)
209-
.withCommand("-c", "tail -f /dev/null")
210195
) {
211196
redpanda.start();
212197

0 commit comments

Comments
 (0)