From fe29ee1115261ca63bc55ab1c76af87152056977 Mon Sep 17 00:00:00 2001
From: Vaibhav Maheshwari <vmaheshw@vmaheshw-mn1.linkedin.biz>
Date: Tue, 30 Nov 2021 18:04:16 -0800
Subject: [PATCH] Fix compilation errors

---
 .../datastream/server/TestLoadBasedTaskCountEstimator.java    | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datastream-server/src/test/java/com/linkedin/datastream/server/TestLoadBasedTaskCountEstimator.java b/datastream-server/src/test/java/com/linkedin/datastream/server/TestLoadBasedTaskCountEstimator.java
index dbe55da9f..d68fa835a 100644
--- a/datastream-server/src/test/java/com/linkedin/datastream/server/TestLoadBasedTaskCountEstimator.java
+++ b/datastream-server/src/test/java/com/linkedin/datastream/server/TestLoadBasedTaskCountEstimator.java
@@ -106,8 +106,8 @@ public void throughputTaskEstimatorWithTopicLevelInformation() {
     List<String> assignedPartitions = Collections.emptyList();
     List<String> unassignedPartitions = Arrays.asList("apple-0", "apple-1", "apple-2", "banana-0");
     LoadBasedTaskCountEstimator estimator = new LoadBasedTaskCountEstimator(TASK_CAPACITY_MBPS,
-        TASK_CAPACITY_UTILIZATION_PCT);
-    int taskCount = estimator.getTaskCount(throughputInfo, assignedPartitions, unassignedPartitions);
+        TASK_CAPACITY_UTILIZATION_PCT, DEFAULT_BYTES_IN_KB_RATE, DEFAULT_MSGS_IN_RATE);
+    int taskCount = estimator.getTaskCount(throughputInfo, assignedPartitions, unassignedPartitions, "test");
     Assert.assertEquals(taskCount, 4);
   }
 }