From ceb2ab88aebe2de7db54af29c141a24ba3f6a569 Mon Sep 17 00:00:00 2001
From: Nicolas Pepin-Perreault <nicolas.pepin-perreault@camunda.com>
Date: Sat, 28 Dec 2024 19:05:01 +0100
Subject: [PATCH] test: use correct network

---
 .../io/zeebe/containers/ZeebeBrokerNodeTest.java   | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/core/src/test/java/io/zeebe/containers/ZeebeBrokerNodeTest.java b/core/src/test/java/io/zeebe/containers/ZeebeBrokerNodeTest.java
index d5145d8c..f160df59 100644
--- a/core/src/test/java/io/zeebe/containers/ZeebeBrokerNodeTest.java
+++ b/core/src/test/java/io/zeebe/containers/ZeebeBrokerNodeTest.java
@@ -71,16 +71,21 @@ private static Stream<Arguments> reuseDataTestCases() {
     return Stream.of(
         new ReuseDataTestCase(
             "broker with embedded gateway should reuse host data",
-            path -> provideBrokerWithHostData(new ZeebeContainer(), path)),
+            path -> provideBrokerWithHostData(new ZeebeContainer(), path).withNetwork(NETWORK)),
         new ReuseDataTestCase(
             "broker without embedded gateway should reuse host data",
-            path -> provideBrokerWithHostData(new ZeebeBrokerContainer(), path)),
+            path ->
+                provideBrokerWithHostData(new ZeebeBrokerContainer(), path).withNetwork(NETWORK)),
         new ReuseDataTestCase(
             "broker with embedded gateway should reuse volume",
-            path -> new ZeebeContainer().withZeebeData(ZeebeVolume.newVolume())),
+            path ->
+                new ZeebeContainer().withZeebeData(ZeebeVolume.newVolume()).withNetwork(NETWORK)),
         new ReuseDataTestCase(
             "broker without embedded gateway should reuse volume",
-            path -> new ZeebeBrokerContainer().withZeebeData(ZeebeVolume.newVolume())));
+            path ->
+                new ZeebeBrokerContainer()
+                    .withZeebeData(ZeebeVolume.newVolume())
+                    .withNetwork(NETWORK)));
   }
 
   @SuppressWarnings("resource")
@@ -153,6 +158,7 @@ void shouldReuseHostDataOnRestart(
     try (final ZeebeBrokerNode<?> broker = brokerNodeProvider.apply(dataDir);
         final ZeebeGatewayContainer gateway =
             new ZeebeGatewayContainer()
+                .withNetwork(NETWORK)
                 .withEnv(
                     "ZEEBE_GATEWAY_CLUSTER_CONTACTPOINT", broker.getInternalClusterAddress())) {