Add QuotaEnforceCommand constructor for use in testing to manually bu…

…ild command config per cluster (#56) Co-authored-by: Daniel Fritsch <[email protected]>
teslamotors · Dec 20, 2021 · a57223b · a57223b
1 parent 9afa847
commit a57223b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/kafka_quota_enforcer/src/main/java/com/tesla/data/quota/enforcer/QuotaEnforceCommand.java b/kafka_quota_enforcer/src/main/java/com/tesla/data/quota/enforcer/QuotaEnforceCommand.java
@@ -25,6 +25,12 @@ public QuotaEnforceCommand(){
     // DO NOT REMOVE, this is needed by jcommander
   }
 
+  // for testing
+  public QuotaEnforceCommand(Map<String, Object> cmdConfig, String cluster) {
+    this.cmdConfig = cmdConfig;
+    this.cluster = cluster;
+  }
+
   @Override
   protected Enforcer<ConfiguredQuota> initEnforcer() {
     Map<String, Object> zkConfig = zookeeperConfig();