add mock metricsSystem before init sql session config

Author: teo

catalogs/catalog-kafka/src/test/java/org/apache/gravitino/catalog/kafka/TestKafkaCatalogOperations.java

@@ -51,6 +51,7 @@
 import org.apache.gravitino.Config;
 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.Schema;
@@ -65,6 +66,7 @@
 import org.apache.gravitino.messaging.TopicChange;
 import org.apache.gravitino.meta.AuditInfo;
 import org.apache.gravitino.meta.CatalogEntity;
+import org.apache.gravitino.metrics.MetricsSystem;
 import org.apache.gravitino.storage.IdGenerator;
 import org.apache.gravitino.storage.RandomIdGenerator;
 import org.apache.gravitino.storage.relational.service.CatalogMetaService;
@@ -76,6 +78,7 @@
 import org.junit.jupiter.api.Test;
 import org.mockito.MockedStatic;
 import org.mockito.Mockito;
+import org.testcontainers.shaded.org.apache.commons.lang3.reflect.FieldUtils;
 
 public class TestKafkaCatalogOperations extends KafkaClusterEmbedded {
 
@@ -169,8 +172,13 @@ public static void setUp() {
         .when(CatalogMetaService::getInstance)
         .thenReturn(spyCatalogMetaService);
 
-    store = EntityStoreFactory.createEntityStore(config);
-    store.initialize(config);
+    try {
+      store = EntityStoreFactory.createEntityStore(config);
+      FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
+      store.initialize(config);
+    } catch (Exception e) {
+      throw new RuntimeException("Failed to create entity store: ", e);
+    }
     idGenerator = new RandomIdGenerator();
     kafkaCatalogEntity =
         CatalogEntity.builder()

catalogs/catalog-model/src/test/java/org/apache/gravtitino/catalog/model/TestModelCatalogOperations.java

@@ -42,10 +42,12 @@
 import java.util.UUID;
 import java.util.stream.Collectors;
 import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.reflect.FieldUtils;
 import org.apache.gravitino.Catalog;
 import org.apache.gravitino.Config;
 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.Schema;
@@ -64,6 +66,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.model.Model;
 import org.apache.gravitino.model.ModelVersion;
 import org.apache.gravitino.storage.IdGenerator;
@@ -111,8 +114,13 @@ public static void setUp() throws IOException {
     when(config.get(STORE_TRANSACTION_MAX_SKEW_TIME)).thenReturn(1000L);
     when(config.get(STORE_DELETE_AFTER_TIME)).thenReturn(20 * 60 * 1000L);
 
-    store = EntityStoreFactory.createEntityStore(config);
-    store.initialize(config);
+    try {
+      store = EntityStoreFactory.createEntityStore(config);
+      FieldUtils.writeField(GravitinoEnv.getInstance(), "metricsSystem", new MetricsSystem(), true);
+      store.initialize(config);
+    } catch (Exception e) {
+      throw new RuntimeException("Failed to create entity store: ", e);
+    }
     idGenerator = new RandomIdGenerator();
 
     // Create the metalake and catalog