fix ut coverage

Signed-off-by: Peng Huo <[email protected]>
opensearch-project · Oct 28, 2023 · cd01cf8 · cd01cf8
1 parent c20762a
commit cd01cf8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/spark/src/main/java/org/opensearch/sql/spark/leasemanager/DefaultLeaseManager.java b/spark/src/main/java/org/opensearch/sql/spark/leasemanager/DefaultLeaseManager.java
@@ -12,7 +12,6 @@
 import java.util.Arrays;
 import java.util.List;
 import java.util.function.Predicate;
-import lombok.RequiredArgsConstructor;
 import org.opensearch.sql.common.setting.Settings;
 import org.opensearch.sql.spark.dispatcher.model.JobType;
 import org.opensearch.sql.spark.execution.statestore.StateStore;
@@ -49,7 +48,6 @@ interface Rule<T> extends Predicate<T> {
     String description();
   }
 
-  @RequiredArgsConstructor
   public class ConcurrentSessionRule implements Rule<LeaseRequest> {
     @Override
     public String description() {

diff --git a/spark/src/test/java/org/opensearch/sql/spark/leasemanager/DefaultLeaseManagerTest.java b/spark/src/test/java/org/opensearch/sql/spark/leasemanager/DefaultLeaseManagerTest.java
@@ -5,15 +5,23 @@
 
 package org.opensearch.sql.spark.leasemanager;
 
-import static org.junit.jupiter.api.Assertions.*;
-
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
+import org.opensearch.sql.common.setting.Settings;
+import org.opensearch.sql.spark.dispatcher.model.JobType;
+import org.opensearch.sql.spark.execution.statestore.StateStore;
+import org.opensearch.sql.spark.leasemanager.model.LeaseRequest;
 
 @ExtendWith(MockitoExtension.class)
 class DefaultLeaseManagerTest {
+  @Mock private Settings settings;
+
+  @Mock private StateStore stateStore;
 
   @Test
-  public void concurrentSessionRuleOnlyApplyToInteractiveQuery() {}
+  public void concurrentSessionRuleOnlyApplyToInteractiveQuery() {
+    new DefaultLeaseManager(settings, stateStore).borrow(new LeaseRequest(JobType.BATCH, "mys3"));
+  }
 }