commit bd18b431ed5326076d1388ae296ae8861744eb76
author InvisibleProgrammer <zsolt.miskolczi@gmail.com> Fri Jan 12 22:18:43 2024 +0100
committer GitHub <noreply@github.com> Sat Jan 13 02:48:43 2024 +0530
tree 818a00d12e82031c4930a396b90a8c65e6b1390c
parent 3b064f45fb909b68e3ceb3ac259b8b1b37c8c0de

HIVE-27974: Fix flaky test - TestReplicationMetricCollector.testSuccessStageFailure (#4976). (Zsolt Miskolczi, reviewed by Ayush Saxena)

ql/src/test/org/apache/hadoop/hive/ql/parse/repl/metric/TestReplicationMetricCollector.java

diff --git a/ql/src/test/org/apache/hadoop/hive/ql/parse/repl/metric/TestReplicationMetricCollector.java b/ql/src/test/org/apache/hadoop/hive/ql/parse/repl/metric/TestReplicationMetricCollector.java
index 842eb1d..ce4c774 100644
--- a/ql/src/test/org/apache/hadoop/hive/ql/parse/repl/metric/TestReplicationMetricCollector.java
+++ b/ql/src/test/org/apache/hadoop/hive/ql/parse/repl/metric/TestReplicationMetricCollector.java
@@ -49,7 +49,6 @@
 import org.junit.runner.RunWith;
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
-import org.mockito.Mock;
 import org.mockito.junit.MockitoJUnitRunner;
 
 import java.util.Map;
@@ -70,16 +69,17 @@
 
   HiveConf conf;
 
-  @Mock
   private FailoverMetaData fmd;
 
-  @Mock
   private MetricSink metricSinkInstance;
 
-  static MockedStatic<MetricSink> metricSinkMockedStatic;
+  MockedStatic<MetricSink> metricSinkMockedStatic;
 
   @Before
   public void setup() throws Exception {
+    fmd = Mockito.mock(FailoverMetaData.class);
+    metricSinkInstance = Mockito.mock(MetricSink.class);
+
     conf = new HiveConf();
     conf.set(Constants.SCHEDULED_QUERY_SCHEDULENAME, "repl");
     conf.set(Constants.SCHEDULED_QUERY_EXECUTIONID, "1");