Fix: Use QuotaManager mock in cli unit tests (#582)

kafka-ops · Apr 3, 2024 · 40d70a6 · 40d70a6
1 parent f1fe064
commit 40d70a6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/test/java/com/purbon/kafka/topology/JulieOpsTest.java b/src/test/java/com/purbon/kafka/topology/JulieOpsTest.java
@@ -151,6 +151,7 @@ public void builderRunTestAsFromCLI() throws Exception {
     builder.setAccessControlManager(accessControlManager);
     builder.setConnectorManager(connectorManager);
     builder.setKSqlArtefactManager(ksqlArtefactManager);
+    builder.setQuotasManager(quotasManager);
 
     doNothing().when(topicManager).updatePlan(any(ExecutionPlan.class), any(Map.class));
 
@@ -162,6 +163,7 @@ public void builderRunTestAsFromCLI() throws Exception {
     verify(topicManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
     verify(accessControlManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
     verify(connectorManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
+    verify(quotasManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
   }
 
   @Test
@@ -217,6 +219,7 @@ public void builderRunTestAsFromCLIWithARedisBackend() throws Exception {
     builder.setAccessControlManager(accessControlManager);
     builder.setConnectorManager(connectorManager);
     builder.setKSqlArtefactManager(ksqlArtefactManager);
+    builder.setQuotasManager(quotasManager);
 
     doNothing().when(topicManager).updatePlan(any(ExecutionPlan.class), any(Map.class));
 
@@ -228,6 +231,7 @@ public void builderRunTestAsFromCLIWithARedisBackend() throws Exception {
     verify(stateProcessor, times(1)).createOrOpen();
     verify(topicManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
     verify(accessControlManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
+    verify(quotasManager, times(1)).updatePlan(any(ExecutionPlan.class), any(Map.class));
   }
 
   @Test