add mock metricsSystem before init sql session config

Author: teo

core/src/test/java/org/apache/gravitino/authorization/TestAccessControlManager.java

@@ -70,6 +70,7 @@
 import org.apache.gravitino.meta.BaseMetalake;
 import org.apache.gravitino.meta.CatalogEntity;
 import org.apache.gravitino.meta.SchemaVersion;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.storage.RandomIdGenerator;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.Assertions;
@@ -131,6 +132,7 @@ public static void setUp() throws Exception {
     Mockito.doReturn(1000L).when(config).get(TREE_LOCK_MIN_NODE_IN_MEMORY);
     Mockito.doReturn(36000L).when(config).get(TREE_LOCK_CLEAN_INTERVAL);
     FieldUtils.writeField(GravitinoEnv.getInstance(), "lockManager", new LockManager(config), true);
+    FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
 
     entityStore = EntityStoreFactory.createEntityStore(config);
     entityStore.initialize(config);

core/src/test/java/org/apache/gravitino/authorization/TestOwnerManager.java

@@ -59,6 +59,7 @@
 import org.apache.gravitino.meta.GroupEntity;
 import org.apache.gravitino.meta.SchemaVersion;
 import org.apache.gravitino.meta.UserEntity;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.storage.IdGenerator;
 import org.apache.gravitino.storage.RandomIdGenerator;
 import org.junit.jupiter.api.AfterAll;
@@ -106,6 +107,7 @@ public static void setUp() throws IOException, IllegalAccessException {
 
     FieldUtils.writeField(GravitinoEnv.getInstance(), "lockManager", new LockManager(config), true);
     FieldUtils.writeField(GravitinoEnv.getInstance(), "catalogManager", catalogManager, true);
+    FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
 
     entityStore = EntityStoreFactory.createEntityStore(config);
     entityStore.initialize(config);

core/src/test/java/org/apache/gravitino/storage/TestEntityStorage.java

@@ -57,6 +57,7 @@
 import org.apache.gravitino.EntityAlreadyExistsException;
 import org.apache.gravitino.EntityStore;
 import org.apache.gravitino.EntityStoreFactory;
+import org.apache.gravitino.GravitinoEnv;
 import org.apache.gravitino.NameIdentifier;
 import org.apache.gravitino.Namespace;
 import org.apache.gravitino.authorization.AuthorizationUtils;
@@ -82,6 +83,7 @@
 import org.apache.gravitino.meta.TableEntity;
 import org.apache.gravitino.meta.TopicEntity;
 import org.apache.gravitino.meta.UserEntity;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.rel.types.Type;
 import org.apache.gravitino.rel.types.Types;
 import org.apache.gravitino.storage.relational.RelationalBackend;
@@ -149,6 +151,8 @@ private void init(String type, Config config) {
 
         FieldUtils.writeStaticField(
             SQLExceptionConverterFactory.class, "converter", new H2ExceptionConverter(), true);
+        FieldUtils.writeField(
+            GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
 
       } else if (type.equalsIgnoreCase("mysql")) {
         String mysqlJdbcUrl = baseIT.startAndInitMySQLBackend();

core/src/test/java/org/apache/gravitino/storage/relational/TestJDBCBackend.java

@@ -52,11 +52,13 @@
 import java.util.Objects;
 import java.util.UUID;
 import org.apache.commons.io.IOUtils;
+import org.apache.commons.lang3.reflect.FieldUtils;
 import org.apache.gravitino.Catalog;
 import org.apache.gravitino.Config;
 import org.apache.gravitino.Configs;
 import org.apache.gravitino.Entity;
 import org.apache.gravitino.EntityAlreadyExistsException;
+import org.apache.gravitino.GravitinoEnv;
 import org.apache.gravitino.NameIdentifier;
 import org.apache.gravitino.Namespace;
 import org.apache.gravitino.authorization.AuthorizationUtils;
@@ -78,6 +80,7 @@
 import org.apache.gravitino.meta.TagEntity;
 import org.apache.gravitino.meta.TopicEntity;
 import org.apache.gravitino.meta.UserEntity;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.storage.RandomIdGenerator;
 import org.apache.gravitino.storage.relational.mapper.GroupMetaMapper;
 import org.apache.gravitino.storage.relational.mapper.UserMetaMapper;
@@ -127,6 +130,7 @@ public static void setup() {
 
     try {
       backend = (RelationalBackend) Class.forName(className).getDeclaredConstructor().newInstance();
+      FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
       backend.initialize(config);
     } catch (Exception e) {
       throw new RuntimeException(

core/src/test/java/org/apache/gravitino/storage/relational/session/TestSqlSession.java

@@ -40,7 +40,10 @@
 import java.util.UUID;
 import org.apache.commons.dbcp2.BasicDataSource;
 import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.reflect.FieldUtils;
 import org.apache.gravitino.Config;
+import org.apache.gravitino.GravitinoEnv;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.ibatis.session.SqlSession;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.AfterEach;
@@ -76,7 +79,12 @@ public static void setUp() {
 
   @BeforeEach
   public void init() {
-    SqlSessionFactoryHelper.getInstance().init(config);
+    try {
+      FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
+      SqlSessionFactoryHelper.getInstance().init(config);
+    } catch (Exception e) {
+      throw new RuntimeException("Failed to init sql session factory: ", e);
+    }
   }
 
   @AfterEach

core/src/test/java/org/apache/gravitino/tag/TestTagManager.java

@@ -73,6 +73,7 @@
 import org.apache.gravitino.meta.SchemaVersion;
 import org.apache.gravitino.meta.TableEntity;
 import org.apache.gravitino.metalake.MetalakeDispatcher;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.rel.types.Types;
 import org.apache.gravitino.storage.IdGenerator;
 import org.apache.gravitino.storage.RandomIdGenerator;
@@ -134,6 +135,7 @@ public static void setUp() throws IOException, IllegalAccessException {
     Mockito.doReturn(1000L).when(config).get(TREE_LOCK_MIN_NODE_IN_MEMORY);
     Mockito.doReturn(36000L).when(config).get(TREE_LOCK_CLEAN_INTERVAL);
     FieldUtils.writeField(GravitinoEnv.getInstance(), "lockManager", new LockManager(config), true);
+    FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
 
     entityStore = EntityStoreFactory.createEntityStore(config);
     entityStore.initialize(config);