diff --git a/src/test/java/com/purbon/kafka/topology/JulieOpsTest.java b/src/test/java/com/purbon/kafka/topology/JulieOpsTest.java index d3a2a1115..9688ed828 100644 --- 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