diff --git a/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/OMMXBeanTests.java b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/OMMXBeanTests.java new file mode 100644 index 000000000000..b821ac741937 --- /dev/null +++ b/hadoop-ozone/integration-test/src/test/java/org/apache/hadoop/ozone/om/OMMXBeanTests.java @@ -0,0 +1,88 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.hadoop.ozone.om; + +import static org.junit.jupiter.api.Assertions.assertEquals; +import static org.junit.jupiter.api.Assertions.assertInstanceOf; +import static org.junit.jupiter.api.Assertions.assertNotNull; + +import java.lang.management.ManagementFactory; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; +import javax.management.MBeanServer; +import javax.management.ObjectName; +import org.apache.ozone.test.NonHATests; +import org.junit.jupiter.api.BeforeAll; +import org.junit.jupiter.api.Test; +import org.junit.jupiter.api.TestInstance; + +/** + * Test cases for the JMX management interface for OM information. + */ +@TestInstance(TestInstance.Lifecycle.PER_CLASS) +public abstract class OMMXBeanTests implements NonHATests.TestCase { + + private OzoneManager om; + private MBeanServer mbs; + + @BeforeAll + void init() { + om = cluster().getOzoneManager(); + mbs = ManagementFactory.getPlatformMBeanServer(); + } + + @Test + public void testOMMXBean() throws Exception { + ObjectName bean = new ObjectName( + "Hadoop:service=OzoneManager," + + "name=OzoneManagerInfo," + + "component=ServerRuntime"); + + String namespace = (String) mbs.getAttribute(bean, "Namespace"); + assertEquals(om.getNamespace(), namespace); + + String rpcPort = (String) mbs.getAttribute(bean, "RpcPort"); + assertEquals(om.getRpcPort(), rpcPort); + + String hostname = (String) mbs.getAttribute(bean, "Hostname"); + assertEquals(om.getHostname(), hostname); + + String ratisLogDirectory = (String) mbs.getAttribute(bean, "RatisLogDirectory"); + assertEquals(om.getRatisLogDirectory(), ratisLogDirectory); + + String rocksDbDirectory = (String) mbs.getAttribute(bean, "RocksDbDirectory"); + assertEquals(om.getRocksDbDirectory(), rocksDbDirectory); + + Object ratisRoles = mbs.getAttribute(bean, "RatisRoles"); + assertNotNull(ratisRoles); + assertEquals(om.getRatisRoles(), toRatisRolesList(ratisRoles)); + + String ratisEvents = (String) mbs.getAttribute(bean, "RatisEvents"); + assertEquals(om.getRatisEvents(), ratisEvents); + } + + private static List> toRatisRolesList(Object ratisRoles) { + String[][] ratisRolesArray = assertInstanceOf(String[][].class, ratisRoles); + List> ratisRolesList = new ArrayList<>(); + for (String[] row : ratisRolesArray) { + ratisRolesList.add(Arrays.asList(row)); + } + return ratisRolesList; + } +} diff --git a/hadoop-ozone/integration-test/src/test/java/org/apache/ozone/test/NonHATests.java b/hadoop-ozone/integration-test/src/test/java/org/apache/ozone/test/NonHATests.java index b034bb21de83..12095ad756c6 100644 --- a/hadoop-ozone/integration-test/src/test/java/org/apache/ozone/test/NonHATests.java +++ b/hadoop-ozone/integration-test/src/test/java/org/apache/ozone/test/NonHATests.java @@ -42,6 +42,7 @@ import org.apache.hadoop.ozone.client.rpc.TestDiscardPreallocatedBlocks; import org.apache.hadoop.ozone.client.rpc.TestOzoneClientMultipartUploadWithFSO; import org.apache.hadoop.ozone.client.rpc.TestOzoneRpcClientWithKeyLatestVersion; +import org.apache.hadoop.ozone.om.OMMXBeanTests; import org.apache.hadoop.ozone.om.TestBucketLayoutWithOlderClient; import org.apache.hadoop.ozone.om.TestListKeys; import org.apache.hadoop.ozone.om.TestListKeysWithFSO; @@ -173,6 +174,14 @@ public MiniOzoneCluster cluster() { } } + @Nested + class OMMXBean extends OMMXBeanTests { + @Override + public MiniOzoneCluster cluster() { + return getCluster(); + } + } + @Nested class SCMNodeManagerMXBean extends TestSCMNodeManagerMXBean { @Override