add UT for EventLogFileHandlerTests

opendistro-for-elasticsearch · Jan 4, 2021 · 8b8f994 · 8b8f994
1 parent 4e1dc69
commit 8b8f994
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/...riter_shared/EventLogFileHandlerTest.java → ...iter_shared/EventLogFileHandlerTests.java b/...riter_shared/EventLogFileHandlerTest.java → ...iter_shared/EventLogFileHandlerTests.java
@@ -1,22 +1,22 @@
 package com.amazon.opendistro.elasticsearch.performanceanalyzer.reader_writer_shared;
 
-import java.io.File;
-
-import org.junit.Before;
-import org.junit.Test;
-import org.mockito.Mock;
+import static org.mockito.Mockito.when;
+import static org.mockito.MockitoAnnotations.initMocks;
 
 import com.amazon.opendistro.elasticsearch.performanceanalyzer.collectors.HeapMetricsCollector;
 import com.amazon.opendistro.elasticsearch.performanceanalyzer.config.PerformanceAnalyzerController;
+import com.amazon.opendistro.elasticsearch.performanceanalyzer.http_action.config.PerformanceAnalyzerConfigAction;
 import com.amazon.opendistro.elasticsearch.performanceanalyzer.metrics.MetricsConfiguration;
 import com.amazon.opendistro.elasticsearch.performanceanalyzer.metrics.PerformanceAnalyzerMetrics;
 import com.amazon.opendistro.elasticsearch.performanceanalyzer.writer.EventLogQueueProcessor;
-import static org.mockito.Mockito.when;
-import static org.mockito.MockitoAnnotations.initMocks;
+import java.io.File;
+import org.junit.Before;
+import org.junit.Test;
+import org.mockito.Mock;
 
-public class EventLogFileHandlerTest {
-    @Mock
-    private PerformanceAnalyzerController mockController;
+public class EventLogFileHandlerTests {
+    @Mock private PerformanceAnalyzerController mockController;
+    @Mock private PerformanceAnalyzerConfigAction configAction;
 
     String pathToTestMetricsDir;
 
@@ -27,6 +27,7 @@ public void init() {
         deleteDirectory(new File(pathToTestMetricsDir));
         boolean newDir = new File(pathToTestMetricsDir).mkdir();
         when(mockController.isPerformanceAnalyzerEnabled()).thenReturn(true);
+        PerformanceAnalyzerConfigAction.setInstance(configAction);
     }
 
     private boolean deleteDirectory(File directoryToBeDeleted) {