HBASE-24657 add unit test for JSONBean.java
Closes #2004 Signed-off-by: Viraj Jasani <vjasani@apache.org>
This commit is contained in:
David Manning
Viraj Jasani
parent
afe2eac7b6
commit
e614b89c33
|
|
@ -0,0 +1,118 @@
|
||||||
|
/**
|
||||||
|
*
|
||||||
|
* 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.hbase.util;
|
||||||
|
|
||||||
|
import static org.junit.Assert.assertEquals;
|
||||||
|
import static org.mockito.ArgumentMatchers.eq;
|
||||||
|
import static org.mockito.Mockito.any;
|
||||||
|
import static org.mockito.Mockito.mock;
|
||||||
|
import static org.mockito.Mockito.when;
|
||||||
|
import java.io.PrintWriter;
|
||||||
|
import java.io.StringWriter;
|
||||||
|
import java.lang.reflect.Type;
|
||||||
|
import java.util.HashSet;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Set;
|
||||||
|
import javax.management.MBeanAttributeInfo;
|
||||||
|
import javax.management.MBeanInfo;
|
||||||
|
import javax.management.MBeanServer;
|
||||||
|
import javax.management.ObjectName;
|
||||||
|
import org.apache.hadoop.hbase.HBaseClassTestRule;
|
||||||
|
import org.apache.hadoop.hbase.testclassification.MiscTests;
|
||||||
|
import org.apache.hadoop.hbase.testclassification.SmallTests;
|
||||||
|
import org.junit.ClassRule;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.experimental.categories.Category;
|
||||||
|
import org.apache.hbase.thirdparty.com.google.common.reflect.TypeToken;
|
||||||
|
import org.apache.hbase.thirdparty.com.google.gson.Gson;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Test {@link JSONBean}.
|
||||||
|
*/
|
||||||
|
@Category({MiscTests.class, SmallTests.class})
|
||||||
|
public class TestJSONBean {
|
||||||
|
@ClassRule
|
||||||
|
public static final HBaseClassTestRule CLASS_RULE =
|
||||||
|
HBaseClassTestRule.forClass(TestJSONBean.class);
|
||||||
|
|
||||||
|
private MBeanServer getMockMBeanServer() throws Exception {
|
||||||
|
MBeanServer mbeanServer = mock(MBeanServer.class);
|
||||||
|
Set<ObjectName> names = new HashSet<>();
|
||||||
|
names.add(new ObjectName("test1:type=test2"));
|
||||||
|
when(mbeanServer.queryNames(any(), any())).thenReturn(names);
|
||||||
|
MBeanInfo mbeanInfo = mock(MBeanInfo.class);
|
||||||
|
when(mbeanInfo.getClassName()).thenReturn("testClassName");
|
||||||
|
String[] attributeNames = new String[] {"intAttr", "nanAttr", "infinityAttr",
|
||||||
|
"strAttr", "boolAttr"};
|
||||||
|
MBeanAttributeInfo[] attributeInfos = new MBeanAttributeInfo[attributeNames.length];
|
||||||
|
for (int i = 0; i < attributeInfos.length; i++) {
|
||||||
|
attributeInfos[i] = new MBeanAttributeInfo(attributeNames[i],
|
||||||
|
null,
|
||||||
|
null,
|
||||||
|
true,
|
||||||
|
false,
|
||||||
|
false);
|
||||||
|
}
|
||||||
|
when(mbeanInfo.getAttributes()).thenReturn(attributeInfos);
|
||||||
|
when(mbeanServer.getMBeanInfo(any())).thenReturn(mbeanInfo);
|
||||||
|
when(mbeanServer.getAttribute(any(), eq("intAttr"))).thenReturn(3);
|
||||||
|
when(mbeanServer.getAttribute(any(), eq("nanAttr"))).thenReturn(Double.NaN);
|
||||||
|
when(mbeanServer.getAttribute(any(), eq("infinityAttr"))).
|
||||||
|
thenReturn(Double.POSITIVE_INFINITY);
|
||||||
|
when(mbeanServer.getAttribute(any(), eq("strAttr"))).thenReturn("aString");
|
||||||
|
when(mbeanServer.getAttribute(any(), eq("boolAttr"))).thenReturn(true);
|
||||||
|
return mbeanServer;
|
||||||
|
}
|
||||||
|
|
||||||
|
private String getExpectedJSON() {
|
||||||
|
StringWriter sw = new StringWriter();
|
||||||
|
PrintWriter pw = new PrintWriter(sw);
|
||||||
|
pw.println("{");
|
||||||
|
pw.println(" \"beans\": [");
|
||||||
|
pw.println(" {");
|
||||||
|
pw.println(" \"name\": \"test1:type=test2\",");
|
||||||
|
pw.println(" \"modelerType\": \"testClassName\",");
|
||||||
|
pw.println(" \"intAttr\": 3,");
|
||||||
|
pw.println(" \"nanAttr\": \"NaN\",");
|
||||||
|
pw.println(" \"infinityAttr\": \"Infinity\",");
|
||||||
|
pw.println(" \"strAttr\": \"aString\",");
|
||||||
|
pw.println(" \"boolAttr\": true");
|
||||||
|
pw.println(" }");
|
||||||
|
pw.println(" ]");
|
||||||
|
pw.print("}");
|
||||||
|
return sw.toString();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testJSONBeanValueTypes() throws Exception {
|
||||||
|
JSONBean bean = new JSONBean();
|
||||||
|
StringWriter stringWriter = new StringWriter();
|
||||||
|
try (
|
||||||
|
PrintWriter printWriter = new PrintWriter(stringWriter);
|
||||||
|
JSONBean.Writer jsonWriter = bean.open(printWriter)) {
|
||||||
|
jsonWriter.write(getMockMBeanServer(), null, null, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
final Gson gson = GsonUtil.createGson().create();
|
||||||
|
Type typeOfHashMap = new TypeToken<Map<String, Object>>() {}.getType();
|
||||||
|
Map<String, Object> expectedJson = gson.fromJson(getExpectedJSON(), typeOfHashMap);
|
||||||
|
Map<String, Object> actualJson = gson.fromJson(stringWriter.toString(), typeOfHashMap);
|
||||||
|
assertEquals(expectedJson, actualJson);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue