test: add cfg to mock

camunda · Oct 25, 2021 · 600364b · 600364b
1 parent e5e6e4d
commit 600364b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/broker/src/test/java/io/camunda/zeebe/broker/system/partitions/ZeebePartitionTest.java b/broker/src/test/java/io/camunda/zeebe/broker/system/partitions/ZeebePartitionTest.java
@@ -21,6 +21,7 @@
 import io.atomix.raft.RaftServer.Role;
 import io.atomix.raft.partition.RaftPartition;
 import io.atomix.raft.partition.impl.RaftPartitionServer;
+import io.camunda.zeebe.broker.system.configuration.BrokerCfg;
 import io.camunda.zeebe.util.exception.UnrecoverableException;
 import io.camunda.zeebe.util.health.CriticalComponentsHealthMonitor;
 import io.camunda.zeebe.util.health.FailureListener;
@@ -65,6 +66,8 @@ public void setup() {
     when(ctx.getComponentHealthMonitor()).thenReturn(healthMonitor);
     when(ctx.createTransitionContext()).thenReturn(ctx);
 
+    when(ctx.getBrokerCfg()).thenReturn(new BrokerCfg());
+
     partition = new ZeebePartition(ctx, transition, List.of(new NoopStartupStep()));
   }
 

diff --git a/...va/io/camunda/zeebe/broker/system/partitions/ZeebePartitionTransitionIntegrationTest.java b/...va/io/camunda/zeebe/broker/system/partitions/ZeebePartitionTransitionIntegrationTest.java
@@ -15,6 +15,7 @@
 import io.atomix.raft.RaftServer.Role;
 import io.atomix.raft.partition.RaftPartition;
 import io.atomix.raft.partition.impl.RaftPartitionServer;
+import io.camunda.zeebe.broker.system.configuration.BrokerCfg;
 import io.camunda.zeebe.broker.system.partitions.impl.PartitionTransitionImpl;
 import io.camunda.zeebe.broker.system.partitions.impl.TestPartitionStep;
 import io.camunda.zeebe.util.health.CriticalComponentsHealthMonitor;
@@ -53,6 +54,7 @@ public void setup() {
     when(ctx.getPartitionContext()).thenReturn(ctx);
     when(ctx.getComponentHealthMonitor()).thenReturn(healthMonitor);
     when(ctx.createTransitionContext()).thenReturn(ctx);
+    when(ctx.getBrokerCfg()).thenReturn(new BrokerCfg());
   }
 
   @Test