HBASE-30039 Upgrade hbase-server to use junit5 Part3 (#8090)

Signed-off-by: Duo Zhang <zhangduo@apache.org>

(cherry picked from commit 39f5903)
(cherry picked from commit b369b38)

Author: liuxiaocs7

hbase-server/src/test/java/org/apache/hadoop/hbase/quotas/SpaceQuotaHelperForTests.java

@@ -17,9 +17,9 @@
  */
 package org.apache.hadoop.hbase.quotas;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
-import static org.junit.Assert.fail;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -64,7 +64,6 @@
 import org.apache.hadoop.hbase.util.Pair;
 import org.apache.hadoop.util.StringUtils;
 import org.apache.yetus.audience.InterfaceAudience;
-import org.junit.rules.TestName;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -83,11 +82,11 @@ public class SpaceQuotaHelperForTests {
   public static final long ONE_GIGABYTE = ONE_MEGABYTE * ONE_KILOBYTE;
 
   private final HBaseTestingUtility testUtil;
-  private final TestName testName;
+  private final String testName;
   private final AtomicLong counter;
   private static final int NUM_RETRIES = 10;
 
-  public SpaceQuotaHelperForTests(HBaseTestingUtility testUtil, TestName testName,
+  public SpaceQuotaHelperForTests(HBaseTestingUtility testUtil, String testName,
     AtomicLong counter) {
     this.testUtil = Objects.requireNonNull(testUtil);
     this.testName = Objects.requireNonNull(testName);
@@ -231,12 +230,13 @@ void verifyViolation(SpaceViolationPolicy policyToViolate, TableName tn, Mutatio
         assertTrue(
           msg.contains("TableNotEnabledException") || msg.contains(policyToViolate.name()));
       } else {
-        assertTrue("Expected exception message to contain the word '" + policyToViolate.name()
-          + "', but was " + msg, msg.contains(policyToViolate.name()));
+        assertTrue(msg.contains(policyToViolate.name()),
+          "Expected exception message to contain the word '" + policyToViolate.name()
+            + "', but was " + msg);
       }
     }
-    assertTrue("Expected to see an exception writing data to a table exceeding its quota",
-      sawError);
+    assertTrue(sawError,
+      "Expected to see an exception writing data to a table exceeding its quota");
   }
 
   /**
@@ -276,7 +276,7 @@ void verifyNoViolation(TableName tn, Mutation m) throws Exception {
         scanner.close();
       }
     }
-    assertTrue("Expected to succeed in writing data to a table not having quota ", sawSuccess);
+    assertTrue(sawSuccess, "Expected to succeed in writing data to a table not having quota ");
   }
 
   /**
@@ -289,8 +289,8 @@ void verifyTableUsageSnapshotForSpaceQuotaExist(TableName tn) throws Exception {
       ResultScanner rs = quotaTable.getScanner(s);
       sawUsageSnapshot = (rs.next() != null);
     }
-    assertTrue("Expected to succeed in getting table usage snapshots for space quota",
-      sawUsageSnapshot);
+    assertTrue(sawUsageSnapshot,
+      "Expected to succeed in getting table usage snapshots for space quota");
   }
 
   /**
@@ -335,7 +335,7 @@ ClientServiceCallable<Void> generateFileToLoad(TableName tn, int numFiles, int n
     Connection conn = testUtil.getConnection();
     FileSystem fs = testUtil.getTestFileSystem();
     Configuration conf = testUtil.getConfiguration();
-    Path baseDir = new Path(fs.getHomeDirectory(), testName.getMethodName() + "_files");
+    Path baseDir = new Path(fs.getHomeDirectory(), testName + "_files");
     fs.mkdirs(baseDir);
     final List<Pair<byte[], String>> famPaths = new ArrayList<Pair<byte[], String>>();
     for (int i = 1; i <= numFiles; i++) {
@@ -363,44 +363,6 @@ public Void rpcCall() throws Exception {
     };
   }
 
-  /**
-   * Bulk-loads a number of files with a number of rows to the given table.
-   */
-  // ClientServiceCallable<Boolean> generateFileToLoad(
-  // TableName tn, int numFiles, int numRowsPerFile) throws Exception {
-  // Connection conn = testUtil.getConnection();
-  // FileSystem fs = testUtil.getTestFileSystem();
-  // Configuration conf = testUtil.getConfiguration();
-  // Path baseDir = new Path(fs.getHomeDirectory(), testName.getMethodName() + "_files");
-  // fs.mkdirs(baseDir);
-  // final List<Pair<byte[], String>> famPaths = new ArrayList<>();
-  // for (int i = 1; i <= numFiles; i++) {
-  // Path hfile = new Path(baseDir, "file" + i);
-  // TestHRegionServerBulkLoad.createHFile(
-  // fs, hfile, Bytes.toBytes(SpaceQuotaHelperForTests.F1), Bytes.toBytes("my"),
-  // Bytes.toBytes("file"), numRowsPerFile);
-  // famPaths.add(new Pair<>(Bytes.toBytes(SpaceQuotaHelperForTests.F1), hfile.toString()));
-  // }
-  //
-  // // bulk load HFiles
-  // Table table = conn.getTable(tn);
-  // final String bulkToken = new SecureBulkLoadClient(conf, table).prepareBulkLoad(conn);
-  // return new ClientServiceCallable<Boolean>(
-  // conn, tn, Bytes.toBytes("row"), new RpcControllerFactory(conf).newController(),
-  // HConstants.PRIORITY_UNSET) {
-  // @Override
-  // public Boolean rpcCall() throws Exception {
-  // SecureBulkLoadClient secureClient = null;
-  // byte[] regionName = getLocation().getRegion().getRegionName();
-  // try (Table table = conn.getTable(getTableName())) {
-  // secureClient = new SecureBulkLoadClient(conf, table);
-  // return secureClient.secureBulkLoadHFiles(getStub(), famPaths, regionName,
-  // true, null, bulkToken);
-  // }
-  // }
-  // };
-  // }
-
   /**
    * Removes the space quota from the given namespace
    */
@@ -596,7 +558,7 @@ TableName getNextTableName() {
   }
 
   TableName getNextTableName(String namespace) {
-    return TableName.valueOf(namespace, testName.getMethodName() + counter.getAndIncrement());
+    return TableName.valueOf(namespace, testName + counter.getAndIncrement());
   }
 
   TableName createTable() throws Exception {
@@ -645,8 +607,7 @@ TableName createTableWithRegions(Admin admin, String namespace, int numRegions,
 
   TableName createTableInNamespace(NamespaceDescriptor nd) throws Exception {
     final Admin admin = testUtil.getAdmin();
-    final TableName tn =
-      TableName.valueOf(nd.getName(), testName.getMethodName() + counter.getAndIncrement());
+    final TableName tn = TableName.valueOf(nd.getName(), testName + counter.getAndIncrement());
 
     // Delete the old table
     if (admin.tableExists(tn)) {

hbase-server/src/test/java/org/apache/hadoop/hbase/quotas/TestActivePolicyEnforcement.java

@@ -17,42 +17,36 @@
  */
 package org.apache.hadoop.hbase.quotas;
 
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertFalse;
-import static org.junit.Assert.assertNotNull;
-import static org.junit.Assert.assertTrue;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.Mockito.mock;
 
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Map.Entry;
-import org.apache.hadoop.hbase.HBaseClassTestRule;
 import org.apache.hadoop.hbase.TableName;
 import org.apache.hadoop.hbase.quotas.SpaceQuotaSnapshot.SpaceQuotaStatus;
 import org.apache.hadoop.hbase.quotas.policies.DefaultViolationPolicyEnforcement;
 import org.apache.hadoop.hbase.quotas.policies.MissingSnapshotViolationPolicyEnforcement;
 import org.apache.hadoop.hbase.quotas.policies.NoWritesViolationPolicyEnforcement;
 import org.apache.hadoop.hbase.regionserver.RegionServerServices;
 import org.apache.hadoop.hbase.testclassification.SmallTests;
-import org.junit.Before;
-import org.junit.ClassRule;
-import org.junit.Test;
-import org.junit.experimental.categories.Category;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
 
 /**
  * Test class for {@link ActivePolicyEnforcement}.
  */
-@Category(SmallTests.class)
+@Tag(SmallTests.TAG)
 public class TestActivePolicyEnforcement {
 
-  @ClassRule
-  public static final HBaseClassTestRule CLASS_RULE =
-    HBaseClassTestRule.forClass(TestActivePolicyEnforcement.class);
-
   private RegionServerServices rss;
 
-  @Before
+  @BeforeEach
   public void setup() {
     rss = mock(RegionServerServices.class);
   }
@@ -73,8 +67,9 @@ public void testNoPolicyReturnsNoopEnforcement() {
     SpaceViolationPolicyEnforcement enforcement =
       ape.getPolicyEnforcement(TableName.valueOf("nonexistent"));
     assertNotNull(enforcement);
-    assertTrue("Expected an instance of MissingSnapshotViolationPolicyEnforcement, but got "
-      + enforcement.getClass(), enforcement instanceof MissingSnapshotViolationPolicyEnforcement);
+    assertTrue(enforcement instanceof MissingSnapshotViolationPolicyEnforcement,
+      "Expected an instance of MissingSnapshotViolationPolicyEnforcement, but got "
+        + enforcement.getClass());
   }
 
   @Test
@@ -84,7 +79,7 @@ public void testNoBulkLoadChecksOnNoSnapshot() {
         Collections.<TableName, SpaceQuotaSnapshot> emptyMap(), mock(RegionServerServices.class));
     SpaceViolationPolicyEnforcement enforcement =
       ape.getPolicyEnforcement(TableName.valueOf("nonexistent"));
-    assertFalse("Should not check bulkloads", enforcement.shouldCheckBulkLoads());
+    assertFalse(enforcement.shouldCheckBulkLoads(), "Should not check bulkloads");
   }
 
   @Test
@@ -109,10 +104,10 @@ public void testNonViolatingQuotaCachesPolicyEnforcment() {
     final ActivePolicyEnforcement ape =
       new ActivePolicyEnforcement(Collections.emptyMap(), snapshots, rss);
     SpaceViolationPolicyEnforcement policyEnforcement = ape.getPolicyEnforcement(tableName);
-    assertTrue("Found the wrong class: " + policyEnforcement.getClass(),
-      policyEnforcement instanceof DefaultViolationPolicyEnforcement);
+    assertTrue(policyEnforcement instanceof DefaultViolationPolicyEnforcement,
+      "Found the wrong class: " + policyEnforcement.getClass());
     SpaceViolationPolicyEnforcement copy = ape.getPolicyEnforcement(tableName);
-    assertTrue("Expected the instance to be cached", policyEnforcement == copy);
+    assertTrue(policyEnforcement == copy, "Expected the instance to be cached");
     Entry<TableName, SpaceViolationPolicyEnforcement> entry =
       ape.getLocallyCachedPolicies().entrySet().iterator().next();
     assertTrue(policyEnforcement == entry.getValue());

hbase-server/src/test/java/org/apache/hadoop/hbase/quotas/TestClusterScopeQuotaThrottle.java

@@ -22,11 +22,10 @@
 import static org.apache.hadoop.hbase.quotas.ThrottleQuotaTestUtil.triggerNamespaceCacheRefresh;
 import static org.apache.hadoop.hbase.quotas.ThrottleQuotaTestUtil.triggerTableCacheRefresh;
 import static org.apache.hadoop.hbase.quotas.ThrottleQuotaTestUtil.triggerUserCacheRefresh;
-import static org.junit.Assert.assertEquals;
-import static org.junit.Assert.assertTrue;
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import java.util.concurrent.TimeUnit;
-import org.apache.hadoop.hbase.HBaseClassTestRule;
 import org.apache.hadoop.hbase.HBaseTestingUtility;
 import org.apache.hadoop.hbase.NamespaceDescriptor;
 import org.apache.hadoop.hbase.TableName;
@@ -38,20 +37,16 @@
 import org.apache.hadoop.hbase.util.Bytes;
 import org.apache.hadoop.hbase.util.EnvironmentEdgeManager;
 import org.apache.hadoop.hbase.util.JVMClusterUtil.RegionServerThread;
-import org.junit.After;
-import org.junit.AfterClass;
-import org.junit.BeforeClass;
-import org.junit.ClassRule;
-import org.junit.Test;
-import org.junit.experimental.categories.Category;
-
-@Category({ RegionServerTests.class, LargeTests.class })
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.AfterEach;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+
+@Tag(RegionServerTests.TAG)
+@Tag(LargeTests.TAG)
 public class TestClusterScopeQuotaThrottle {
 
-  @ClassRule
-  public static final HBaseClassTestRule CLASS_RULE =
-    HBaseClassTestRule.forClass(TestClusterScopeQuotaThrottle.class);
-
   private final static int REFRESH_TIME = 30 * 60000;
   private final static HBaseTestingUtility TEST_UTIL = new HBaseTestingUtility();
 
@@ -67,7 +62,7 @@ public class TestClusterScopeQuotaThrottle {
   private final static TableName TABLE_NAME = TableName.valueOf(NAMESPACE, "TestTable");
   private static Table table;
 
-  @BeforeClass
+  @BeforeAll
   public static void setUpBeforeClass() throws Exception {
     TEST_UTIL.getConfiguration().setBoolean(QuotaUtil.QUOTA_CONF_KEY, true);
     TEST_UTIL.getConfiguration().setInt(QuotaCache.REFRESH_CONF_KEY, REFRESH_TIME);
@@ -93,7 +88,7 @@ public static void setUpBeforeClass() throws Exception {
       .build();
   }
 
-  @AfterClass
+  @AfterAll
   public static void tearDownAfterClass() throws Exception {
     EnvironmentEdgeManager.reset();
     for (int i = 0; i < tables.length; ++i) {
@@ -107,7 +102,7 @@ public static void tearDownAfterClass() throws Exception {
     TEST_UTIL.shutdownMiniCluster();
   }
 
-  @After
+  @AfterEach
   public void tearDown() throws Exception {
     ThrottleQuotaTestUtil.clearQuotaCache(TEST_UTIL);
   }