YARN-5702. Refactor TestPBImplRecords for reuse in other YARN modules. Contributed by Subru Krishnan
This commit is contained in:
Chris Douglas
parent
8867762256
commit
d65b957776
|
|
@ -0,0 +1,264 @@
|
||||||
|
/**
|
||||||
|
* 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.yarn.api;
|
||||||
|
|
||||||
|
import com.google.common.collect.Lists;
|
||||||
|
import com.google.common.collect.Maps;
|
||||||
|
import com.google.common.collect.Sets;
|
||||||
|
import org.apache.commons.logging.Log;
|
||||||
|
import org.apache.commons.logging.LogFactory;
|
||||||
|
import org.junit.Assert;
|
||||||
|
|
||||||
|
import java.lang.reflect.*;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.util.*;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generic helper class to validate protocol records.
|
||||||
|
*/
|
||||||
|
public class BasePBImplRecordsTest {
|
||||||
|
static final Log LOG = LogFactory.getLog(BasePBImplRecordsTest.class);
|
||||||
|
|
||||||
|
@SuppressWarnings("checkstyle:visibilitymodifier")
|
||||||
|
protected static HashMap<Type, Object> typeValueCache =
|
||||||
|
new HashMap<Type, Object>();
|
||||||
|
private static Random rand = new Random();
|
||||||
|
private static byte [] bytes = new byte[] {'1', '2', '3', '4'};
|
||||||
|
|
||||||
|
@SuppressWarnings({"rawtypes", "unchecked"})
|
||||||
|
private static Object genTypeValue(Type type) {
|
||||||
|
Object ret = typeValueCache.get(type);
|
||||||
|
if (ret != null) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
// only use positive primitive values
|
||||||
|
if (type.equals(boolean.class)) {
|
||||||
|
return rand.nextBoolean();
|
||||||
|
} else if (type.equals(byte.class)) {
|
||||||
|
return bytes[rand.nextInt(4)];
|
||||||
|
} else if (type.equals(int.class) || type.equals(Integer.class)) {
|
||||||
|
return rand.nextInt(1000000);
|
||||||
|
} else if (type.equals(long.class) || type.equals(Long.class)) {
|
||||||
|
return Long.valueOf(rand.nextInt(1000000));
|
||||||
|
} else if (type.equals(float.class)) {
|
||||||
|
return rand.nextFloat();
|
||||||
|
} else if (type.equals(double.class)) {
|
||||||
|
return rand.nextDouble();
|
||||||
|
} else if (type.equals(String.class)) {
|
||||||
|
return String.format("%c%c%c",
|
||||||
|
'a' + rand.nextInt(26),
|
||||||
|
'a' + rand.nextInt(26),
|
||||||
|
'a' + rand.nextInt(26));
|
||||||
|
} else if (type instanceof Class) {
|
||||||
|
Class clazz = (Class)type;
|
||||||
|
if (clazz.isArray()) {
|
||||||
|
Class compClass = clazz.getComponentType();
|
||||||
|
if (compClass != null) {
|
||||||
|
ret = Array.newInstance(compClass, 2);
|
||||||
|
Array.set(ret, 0, genTypeValue(compClass));
|
||||||
|
Array.set(ret, 1, genTypeValue(compClass));
|
||||||
|
}
|
||||||
|
} else if (clazz.isEnum()) {
|
||||||
|
Object [] values = clazz.getEnumConstants();
|
||||||
|
ret = values[rand.nextInt(values.length)];
|
||||||
|
} else if (clazz.equals(ByteBuffer.class)) {
|
||||||
|
// return new ByteBuffer every time
|
||||||
|
// to prevent potential side effects
|
||||||
|
ByteBuffer buff = ByteBuffer.allocate(4);
|
||||||
|
rand.nextBytes(buff.array());
|
||||||
|
return buff;
|
||||||
|
}
|
||||||
|
} else if (type instanceof ParameterizedType) {
|
||||||
|
ParameterizedType pt = (ParameterizedType)type;
|
||||||
|
Type rawType = pt.getRawType();
|
||||||
|
Type [] params = pt.getActualTypeArguments();
|
||||||
|
// only support EnumSet<T>, List<T>, Set<T>, Map<K,V>
|
||||||
|
if (rawType.equals(EnumSet.class)) {
|
||||||
|
if (params[0] instanceof Class) {
|
||||||
|
Class c = (Class)(params[0]);
|
||||||
|
return EnumSet.allOf(c);
|
||||||
|
}
|
||||||
|
} if (rawType.equals(List.class)) {
|
||||||
|
ret = Lists.newArrayList(genTypeValue(params[0]));
|
||||||
|
} else if (rawType.equals(Set.class)) {
|
||||||
|
ret = Sets.newHashSet(genTypeValue(params[0]));
|
||||||
|
} else if (rawType.equals(Map.class)) {
|
||||||
|
Map<Object, Object> map = Maps.newHashMap();
|
||||||
|
map.put(genTypeValue(params[0]), genTypeValue(params[1]));
|
||||||
|
ret = map;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (ret == null) {
|
||||||
|
throw new IllegalArgumentException("type " + type + " is not supported");
|
||||||
|
}
|
||||||
|
typeValueCache.put(type, ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* this method generate record instance by calling newIntance
|
||||||
|
* using reflection, add register the generated value to typeValueCache
|
||||||
|
*/
|
||||||
|
@SuppressWarnings("rawtypes")
|
||||||
|
protected static Object generateByNewInstance(Class clazz) throws Exception {
|
||||||
|
Object ret = typeValueCache.get(clazz);
|
||||||
|
if (ret != null) {
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
Method newInstance = null;
|
||||||
|
Type [] paramTypes = new Type[0];
|
||||||
|
// get newInstance method with most parameters
|
||||||
|
for (Method m : clazz.getMethods()) {
|
||||||
|
int mod = m.getModifiers();
|
||||||
|
if (m.getDeclaringClass().equals(clazz) &&
|
||||||
|
Modifier.isPublic(mod) &&
|
||||||
|
Modifier.isStatic(mod) &&
|
||||||
|
m.getName().equals("newInstance")) {
|
||||||
|
Type [] pts = m.getGenericParameterTypes();
|
||||||
|
if (newInstance == null
|
||||||
|
|| (pts.length > paramTypes.length)) {
|
||||||
|
newInstance = m;
|
||||||
|
paramTypes = pts;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (newInstance == null) {
|
||||||
|
throw new IllegalArgumentException("type " + clazz.getName() +
|
||||||
|
" does not have newInstance method");
|
||||||
|
}
|
||||||
|
Object [] args = new Object[paramTypes.length];
|
||||||
|
for (int i=0;i<args.length;i++) {
|
||||||
|
args[i] = genTypeValue(paramTypes[i]);
|
||||||
|
}
|
||||||
|
ret = newInstance.invoke(null, args);
|
||||||
|
typeValueCache.put(clazz, ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
private class GetSetPair {
|
||||||
|
public String propertyName;
|
||||||
|
public Method getMethod;
|
||||||
|
public Method setMethod;
|
||||||
|
public Type type;
|
||||||
|
public Object testValue;
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String toString() {
|
||||||
|
return String.format("{ name=%s, class=%s, value=%s }", propertyName,
|
||||||
|
type, testValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private <R> Map<String, GetSetPair> getGetSetPairs(Class<R> recordClass)
|
||||||
|
throws Exception {
|
||||||
|
Map<String, GetSetPair> ret = new HashMap<String, GetSetPair>();
|
||||||
|
Method [] methods = recordClass.getDeclaredMethods();
|
||||||
|
// get all get methods
|
||||||
|
for (int i = 0; i < methods.length; i++) {
|
||||||
|
Method m = methods[i];
|
||||||
|
int mod = m.getModifiers();
|
||||||
|
if (m.getDeclaringClass().equals(recordClass) &&
|
||||||
|
Modifier.isPublic(mod) &&
|
||||||
|
(!Modifier.isStatic(mod))) {
|
||||||
|
String name = m.getName();
|
||||||
|
if (name.equals("getProto")) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if ((name.length() > 3) && name.startsWith("get") &&
|
||||||
|
(m.getParameterTypes().length == 0)) {
|
||||||
|
String propertyName = name.substring(3);
|
||||||
|
Type valueType = m.getGenericReturnType();
|
||||||
|
GetSetPair p = ret.get(propertyName);
|
||||||
|
if (p == null) {
|
||||||
|
p = new GetSetPair();
|
||||||
|
p.propertyName = propertyName;
|
||||||
|
p.type = valueType;
|
||||||
|
p.getMethod = m;
|
||||||
|
ret.put(propertyName, p);
|
||||||
|
} else {
|
||||||
|
Assert.fail("Multiple get method with same name: " + recordClass
|
||||||
|
+ p.propertyName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// match get methods with set methods
|
||||||
|
for (int i = 0; i < methods.length; i++) {
|
||||||
|
Method m = methods[i];
|
||||||
|
int mod = m.getModifiers();
|
||||||
|
if (m.getDeclaringClass().equals(recordClass) &&
|
||||||
|
Modifier.isPublic(mod) &&
|
||||||
|
(!Modifier.isStatic(mod))) {
|
||||||
|
String name = m.getName();
|
||||||
|
if (name.startsWith("set") && (m.getParameterTypes().length == 1)) {
|
||||||
|
String propertyName = name.substring(3);
|
||||||
|
Type valueType = m.getGenericParameterTypes()[0];
|
||||||
|
GetSetPair p = ret.get(propertyName);
|
||||||
|
if (p != null && p.type.equals(valueType)) {
|
||||||
|
p.setMethod = m;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// exclude incomplete get/set pair, and generate test value
|
||||||
|
Iterator<Map.Entry<String, GetSetPair>> itr = ret.entrySet().iterator();
|
||||||
|
while (itr.hasNext()) {
|
||||||
|
Map.Entry<String, GetSetPair> cur = itr.next();
|
||||||
|
GetSetPair gsp = cur.getValue();
|
||||||
|
if ((gsp.getMethod == null) ||
|
||||||
|
(gsp.setMethod == null)) {
|
||||||
|
LOG.info(String.format("Exclude protential property: %s\n", gsp.propertyName));
|
||||||
|
itr.remove();
|
||||||
|
} else {
|
||||||
|
LOG.info(String.format("New property: %s type: %s", gsp.toString(), gsp.type));
|
||||||
|
gsp.testValue = genTypeValue(gsp.type);
|
||||||
|
LOG.info(String.format(" testValue: %s\n", gsp.testValue));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
protected <R, P> void validatePBImplRecord(Class<R> recordClass,
|
||||||
|
Class<P> protoClass)
|
||||||
|
throws Exception {
|
||||||
|
LOG.info(String.format("Validate %s %s\n", recordClass.getName(),
|
||||||
|
protoClass.getName()));
|
||||||
|
Constructor<R> emptyConstructor = recordClass.getConstructor();
|
||||||
|
Constructor<R> pbConstructor = recordClass.getConstructor(protoClass);
|
||||||
|
Method getProto = recordClass.getDeclaredMethod("getProto");
|
||||||
|
Map<String, GetSetPair> getSetPairs = getGetSetPairs(recordClass);
|
||||||
|
R origRecord = emptyConstructor.newInstance();
|
||||||
|
for (GetSetPair gsp : getSetPairs.values()) {
|
||||||
|
gsp.setMethod.invoke(origRecord, gsp.testValue);
|
||||||
|
}
|
||||||
|
Object ret = getProto.invoke(origRecord);
|
||||||
|
Assert.assertNotNull(recordClass.getName() + "#getProto returns null", ret);
|
||||||
|
if (!(protoClass.isAssignableFrom(ret.getClass()))) {
|
||||||
|
Assert.fail("Illegal getProto method return type: " + ret.getClass());
|
||||||
|
}
|
||||||
|
R deserRecord = pbConstructor.newInstance(ret);
|
||||||
|
Assert.assertEquals("whole " + recordClass + " records should be equal",
|
||||||
|
origRecord, deserRecord);
|
||||||
|
for (GetSetPair gsp : getSetPairs.values()) {
|
||||||
|
Object origValue = gsp.getMethod.invoke(origRecord);
|
||||||
|
Object deserValue = gsp.getMethod.invoke(deserRecord);
|
||||||
|
Assert.assertEquals("property " + recordClass.getName() + "#"
|
||||||
|
+ gsp.propertyName + " should be equal", origValue, deserValue);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -16,26 +16,8 @@
|
||||||
* limitations under the License.
|
* limitations under the License.
|
||||||
*/
|
*/
|
||||||
package org.apache.hadoop.yarn.api;
|
package org.apache.hadoop.yarn.api;
|
||||||
import java.io.IOException;
|
import com.google.common.collect.ImmutableSet;
|
||||||
import java.lang.reflect.Array;
|
|
||||||
import java.lang.reflect.Constructor;
|
|
||||||
import java.lang.reflect.Method;
|
|
||||||
import java.lang.reflect.Modifier;
|
|
||||||
import java.lang.reflect.ParameterizedType;
|
|
||||||
import java.lang.reflect.Type;
|
|
||||||
import java.nio.ByteBuffer;
|
|
||||||
import java.util.EnumSet;
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.Iterator;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
import java.util.Map.Entry;
|
|
||||||
import java.util.Random;
|
|
||||||
import java.util.Set;
|
|
||||||
|
|
||||||
import org.apache.commons.lang.math.LongRange;
|
import org.apache.commons.lang.math.LongRange;
|
||||||
import org.apache.commons.logging.Log;
|
|
||||||
import org.apache.commons.logging.LogFactory;
|
|
||||||
import org.apache.hadoop.security.proto.SecurityProtos.CancelDelegationTokenRequestProto;
|
import org.apache.hadoop.security.proto.SecurityProtos.CancelDelegationTokenRequestProto;
|
||||||
import org.apache.hadoop.security.proto.SecurityProtos.CancelDelegationTokenResponseProto;
|
import org.apache.hadoop.security.proto.SecurityProtos.CancelDelegationTokenResponseProto;
|
||||||
import org.apache.hadoop.security.proto.SecurityProtos.GetDelegationTokenRequestProto;
|
import org.apache.hadoop.security.proto.SecurityProtos.GetDelegationTokenRequestProto;
|
||||||
|
|
@ -343,127 +325,12 @@ import org.junit.BeforeClass;
|
||||||
import org.junit.Ignore;
|
import org.junit.Ignore;
|
||||||
import org.junit.Test;
|
import org.junit.Test;
|
||||||
|
|
||||||
import com.google.common.collect.ImmutableSet;
|
import java.io.IOException;
|
||||||
import com.google.common.collect.Lists;
|
|
||||||
import com.google.common.collect.Maps;
|
|
||||||
import com.google.common.collect.Sets;
|
|
||||||
|
|
||||||
public class TestPBImplRecords {
|
/**
|
||||||
static final Log LOG = LogFactory.getLog(TestPBImplRecords.class);
|
* Test class for YARN API protocol records.
|
||||||
|
|
||||||
private static HashMap<Type, Object> typeValueCache = new HashMap<Type, Object>();
|
|
||||||
private static Random rand = new Random();
|
|
||||||
private static byte [] bytes = new byte[] {'1', '2', '3', '4'};
|
|
||||||
|
|
||||||
@SuppressWarnings({"rawtypes", "unchecked"})
|
|
||||||
private static Object genTypeValue(Type type) {
|
|
||||||
Object ret = typeValueCache.get(type);
|
|
||||||
if (ret != null) {
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
// only use positive primitive values
|
|
||||||
if (type.equals(boolean.class)) {
|
|
||||||
return rand.nextBoolean();
|
|
||||||
} else if (type.equals(byte.class)) {
|
|
||||||
return bytes[rand.nextInt(4)];
|
|
||||||
} else if (type.equals(int.class) || type.equals(Integer.class)) {
|
|
||||||
return rand.nextInt(1000000);
|
|
||||||
} else if (type.equals(long.class) || type.equals(Long.class)) {
|
|
||||||
return Long.valueOf(rand.nextInt(1000000));
|
|
||||||
} else if (type.equals(float.class)) {
|
|
||||||
return rand.nextFloat();
|
|
||||||
} else if (type.equals(double.class)) {
|
|
||||||
return rand.nextDouble();
|
|
||||||
} else if (type.equals(String.class)) {
|
|
||||||
return String.format("%c%c%c",
|
|
||||||
'a' + rand.nextInt(26),
|
|
||||||
'a' + rand.nextInt(26),
|
|
||||||
'a' + rand.nextInt(26));
|
|
||||||
} else if (type instanceof Class) {
|
|
||||||
Class clazz = (Class)type;
|
|
||||||
if (clazz.isArray()) {
|
|
||||||
Class compClass = clazz.getComponentType();
|
|
||||||
if (compClass != null) {
|
|
||||||
ret = Array.newInstance(compClass, 2);
|
|
||||||
Array.set(ret, 0, genTypeValue(compClass));
|
|
||||||
Array.set(ret, 1, genTypeValue(compClass));
|
|
||||||
}
|
|
||||||
} else if (clazz.isEnum()) {
|
|
||||||
Object [] values = clazz.getEnumConstants();
|
|
||||||
ret = values[rand.nextInt(values.length)];
|
|
||||||
} else if (clazz.equals(ByteBuffer.class)) {
|
|
||||||
// return new ByteBuffer every time
|
|
||||||
// to prevent potential side effects
|
|
||||||
ByteBuffer buff = ByteBuffer.allocate(4);
|
|
||||||
rand.nextBytes(buff.array());
|
|
||||||
return buff;
|
|
||||||
}
|
|
||||||
} else if (type instanceof ParameterizedType) {
|
|
||||||
ParameterizedType pt = (ParameterizedType)type;
|
|
||||||
Type rawType = pt.getRawType();
|
|
||||||
Type [] params = pt.getActualTypeArguments();
|
|
||||||
// only support EnumSet<T>, List<T>, Set<T>, Map<K,V>
|
|
||||||
if (rawType.equals(EnumSet.class)) {
|
|
||||||
if (params[0] instanceof Class) {
|
|
||||||
Class c = (Class)(params[0]);
|
|
||||||
return EnumSet.allOf(c);
|
|
||||||
}
|
|
||||||
} if (rawType.equals(List.class)) {
|
|
||||||
ret = Lists.newArrayList(genTypeValue(params[0]));
|
|
||||||
} else if (rawType.equals(Set.class)) {
|
|
||||||
ret = Sets.newHashSet(genTypeValue(params[0]));
|
|
||||||
} else if (rawType.equals(Map.class)) {
|
|
||||||
Map<Object, Object> map = Maps.newHashMap();
|
|
||||||
map.put(genTypeValue(params[0]), genTypeValue(params[1]));
|
|
||||||
ret = map;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (ret == null) {
|
|
||||||
throw new IllegalArgumentException("type " + type + " is not supported");
|
|
||||||
}
|
|
||||||
typeValueCache.put(type, ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* this method generate record instance by calling newIntance
|
|
||||||
* using reflection, add register the generated value to typeValueCache
|
|
||||||
*/
|
*/
|
||||||
@SuppressWarnings("rawtypes")
|
public class TestPBImplRecords extends BasePBImplRecordsTest {
|
||||||
private static Object generateByNewInstance(Class clazz) throws Exception {
|
|
||||||
Object ret = typeValueCache.get(clazz);
|
|
||||||
if (ret != null) {
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
Method newInstance = null;
|
|
||||||
Type [] paramTypes = new Type[0];
|
|
||||||
// get newInstance method with most parameters
|
|
||||||
for (Method m : clazz.getMethods()) {
|
|
||||||
int mod = m.getModifiers();
|
|
||||||
if (m.getDeclaringClass().equals(clazz) &&
|
|
||||||
Modifier.isPublic(mod) &&
|
|
||||||
Modifier.isStatic(mod) &&
|
|
||||||
m.getName().equals("newInstance")) {
|
|
||||||
Type [] pts = m.getGenericParameterTypes();
|
|
||||||
if (newInstance == null
|
|
||||||
|| (pts.length > paramTypes.length)) {
|
|
||||||
newInstance = m;
|
|
||||||
paramTypes = pts;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (newInstance == null) {
|
|
||||||
throw new IllegalArgumentException("type " + clazz.getName() +
|
|
||||||
" does not have newInstance method");
|
|
||||||
}
|
|
||||||
Object [] args = new Object[paramTypes.length];
|
|
||||||
for (int i=0;i<args.length;i++) {
|
|
||||||
args[i] = genTypeValue(paramTypes[i]);
|
|
||||||
}
|
|
||||||
ret = newInstance.invoke(null, args);
|
|
||||||
typeValueCache.put(clazz, ret);
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
@BeforeClass
|
@BeforeClass
|
||||||
public static void setup() throws Exception {
|
public static void setup() throws Exception {
|
||||||
|
|
@ -532,118 +399,6 @@ public class TestPBImplRecords {
|
||||||
generateByNewInstance(CommitResponse.class);
|
generateByNewInstance(CommitResponse.class);
|
||||||
}
|
}
|
||||||
|
|
||||||
private class GetSetPair {
|
|
||||||
public String propertyName;
|
|
||||||
public Method getMethod;
|
|
||||||
public Method setMethod;
|
|
||||||
public Type type;
|
|
||||||
public Object testValue;
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String toString() {
|
|
||||||
return String.format("{ name=%s, class=%s, value=%s }", propertyName,
|
|
||||||
type, testValue);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private <R> Map<String, GetSetPair> getGetSetPairs(Class<R> recordClass)
|
|
||||||
throws Exception {
|
|
||||||
Map<String, GetSetPair> ret = new HashMap<String, GetSetPair>();
|
|
||||||
Method [] methods = recordClass.getDeclaredMethods();
|
|
||||||
// get all get methods
|
|
||||||
for (int i = 0; i < methods.length; i++) {
|
|
||||||
Method m = methods[i];
|
|
||||||
int mod = m.getModifiers();
|
|
||||||
if (m.getDeclaringClass().equals(recordClass) &&
|
|
||||||
Modifier.isPublic(mod) &&
|
|
||||||
(!Modifier.isStatic(mod))) {
|
|
||||||
String name = m.getName();
|
|
||||||
if (name.equals("getProto")) {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
if ((name.length() > 3) && name.startsWith("get") &&
|
|
||||||
(m.getParameterTypes().length == 0)) {
|
|
||||||
String propertyName = name.substring(3);
|
|
||||||
Type valueType = m.getGenericReturnType();
|
|
||||||
GetSetPair p = ret.get(propertyName);
|
|
||||||
if (p == null) {
|
|
||||||
p = new GetSetPair();
|
|
||||||
p.propertyName = propertyName;
|
|
||||||
p.type = valueType;
|
|
||||||
p.getMethod = m;
|
|
||||||
ret.put(propertyName, p);
|
|
||||||
} else {
|
|
||||||
Assert.fail("Multiple get method with same name: " + recordClass
|
|
||||||
+ p.propertyName);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// match get methods with set methods
|
|
||||||
for (int i = 0; i < methods.length; i++) {
|
|
||||||
Method m = methods[i];
|
|
||||||
int mod = m.getModifiers();
|
|
||||||
if (m.getDeclaringClass().equals(recordClass) &&
|
|
||||||
Modifier.isPublic(mod) &&
|
|
||||||
(!Modifier.isStatic(mod))) {
|
|
||||||
String name = m.getName();
|
|
||||||
if (name.startsWith("set") && (m.getParameterTypes().length == 1)) {
|
|
||||||
String propertyName = name.substring(3);
|
|
||||||
Type valueType = m.getGenericParameterTypes()[0];
|
|
||||||
GetSetPair p = ret.get(propertyName);
|
|
||||||
if (p != null && p.type.equals(valueType)) {
|
|
||||||
p.setMethod = m;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// exclude incomplete get/set pair, and generate test value
|
|
||||||
Iterator<Entry<String, GetSetPair>> itr = ret.entrySet().iterator();
|
|
||||||
while (itr.hasNext()) {
|
|
||||||
Entry<String, GetSetPair> cur = itr.next();
|
|
||||||
GetSetPair gsp = cur.getValue();
|
|
||||||
if ((gsp.getMethod == null) ||
|
|
||||||
(gsp.setMethod == null)) {
|
|
||||||
LOG.info(String.format("Exclude protential property: %s\n", gsp.propertyName));
|
|
||||||
itr.remove();
|
|
||||||
} else {
|
|
||||||
LOG.info(String.format("New property: %s type: %s", gsp.toString(), gsp.type));
|
|
||||||
gsp.testValue = genTypeValue(gsp.type);
|
|
||||||
LOG.info(String.format(" testValue: %s\n", gsp.testValue));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return ret;
|
|
||||||
}
|
|
||||||
|
|
||||||
private <R, P> void validatePBImplRecord(Class<R> recordClass,
|
|
||||||
Class<P> protoClass)
|
|
||||||
throws Exception {
|
|
||||||
LOG.info(String.format("Validate %s %s\n", recordClass.getName(),
|
|
||||||
protoClass.getName()));
|
|
||||||
Constructor<R> emptyConstructor = recordClass.getConstructor();
|
|
||||||
Constructor<R> pbConstructor = recordClass.getConstructor(protoClass);
|
|
||||||
Method getProto = recordClass.getDeclaredMethod("getProto");
|
|
||||||
Map<String, GetSetPair> getSetPairs = getGetSetPairs(recordClass);
|
|
||||||
R origRecord = emptyConstructor.newInstance();
|
|
||||||
for (GetSetPair gsp : getSetPairs.values()) {
|
|
||||||
gsp.setMethod.invoke(origRecord, gsp.testValue);
|
|
||||||
}
|
|
||||||
Object ret = getProto.invoke(origRecord);
|
|
||||||
Assert.assertNotNull(recordClass.getName() + "#getProto returns null", ret);
|
|
||||||
if (!(protoClass.isAssignableFrom(ret.getClass()))) {
|
|
||||||
Assert.fail("Illegal getProto method return type: " + ret.getClass());
|
|
||||||
}
|
|
||||||
R deserRecord = pbConstructor.newInstance(ret);
|
|
||||||
Assert.assertEquals("whole " + recordClass + " records should be equal",
|
|
||||||
origRecord, deserRecord);
|
|
||||||
for (GetSetPair gsp : getSetPairs.values()) {
|
|
||||||
Object origValue = gsp.getMethod.invoke(origRecord);
|
|
||||||
Object deserValue = gsp.getMethod.invoke(deserRecord);
|
|
||||||
Assert.assertEquals("property " + recordClass.getName() + "#"
|
|
||||||
+ gsp.propertyName + " should be equal", origValue, deserValue);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
public void testAllocateRequestPBImpl() throws Exception {
|
public void testAllocateRequestPBImpl() throws Exception {
|
||||||
validatePBImplRecord(AllocateRequestPBImpl.class, AllocateRequestProto.class);
|
validatePBImplRecord(AllocateRequestPBImpl.class, AllocateRequestProto.class);
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue