[helix] 02/02: Add unit test for setting application name.

[jxue]

This is an automated email from the ASF dual-hosted git repository.

jxue pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/helix.git

commit cdbdd4fecd9b7ddbf62f99b112364b46097b4108
Author: Huizhi Lu <ihuizhi...@gmail.com>
AuthorDate: Mon Sep 30 10:15:09 2019 -0700

    Add unit test for setting application name.
---
 .../rest/server/TestDefaultMonitoringMbeans.java   | 32 ++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git 
a/helix-rest/src/test/java/org/apache/helix/rest/server/TestDefaultMonitoringMbeans.java
 
b/helix-rest/src/test/java/org/apache/helix/rest/server/TestDefaultMonitoringMbeans.java
index 90ea420..be44d57 100644
--- 
a/helix-rest/src/test/java/org/apache/helix/rest/server/TestDefaultMonitoringMbeans.java
+++ 
b/helix-rest/src/test/java/org/apache/helix/rest/server/TestDefaultMonitoringMbeans.java
@@ -20,19 +20,27 @@ package org.apache.helix.rest.server;
  */
 
 import java.lang.management.ManagementFactory;
+import java.util.Arrays;
+import java.util.HashSet;
 import java.util.Random;
+import java.util.Set;
+
 import javax.management.AttributeNotFoundException;
 import javax.management.InstanceNotFoundException;
 import javax.management.MBeanException;
 import javax.management.MBeanServer;
+import javax.management.MalformedObjectNameException;
 import javax.management.ObjectName;
 import javax.management.ReflectionException;
 import javax.ws.rs.core.Response;
+
 import org.apache.helix.TestHelper;
+import org.apache.helix.rest.common.HelixRestNamespace;
 import org.testng.Assert;
 import org.testng.annotations.Test;
 
 public class TestDefaultMonitoringMbeans extends AbstractTestClass {
+  private static final String DEFAULT_METRIC_DOMAIN = "org.glassfish.jersey";
 
   // For entire testing environment, we could have 2 - 4 rest server during 
the testing. So we dont
   // know which REST server got the request and report number. So we have to 
loop all of them to
@@ -74,4 +82,28 @@ public class TestDefaultMonitoringMbeans extends 
AbstractTestClass {
     Assert.assertTrue(correctReports);
     System.out.println("End test :" + TestHelper.getTestMethodName());
   }
+
+  @Test
+  public void testMBeanApplicationName()
+      throws MalformedObjectNameException {
+    Set<String> namespaces =
+        new HashSet<>(Arrays.asList(HelixRestNamespace.DEFAULT_NAMESPACE_NAME, 
TEST_NAMESPACE));
+    MBeanServer mBeanServer = ManagementFactory.getPlatformMBeanServer();
+    Set<ObjectName> objectNames =
+        mBeanServer.queryNames(new ObjectName(DEFAULT_METRIC_DOMAIN + ":*"), 
null);
+
+    Set<String> appNames = new HashSet<>();
+    for (ObjectName mBeanName : objectNames) {
+      appNames.add(mBeanName.getKeyProperty("type"));
+    }
+
+    Assert.assertEquals(appNames.size(), namespaces.size(), String
+        .format("appNames: %s does't have the same size as namespaces: %s.", 
appNames, namespaces));
+
+    for (String appName : appNames) {
+      Assert.assertTrue(namespaces.contains(appName), String
+          .format("Application name: %s is not one of the namespaces: %s.", 
appName, namespaces));
+      namespaces.remove(appName);
+    }
+  }
 }