BeanAssert.java

package cn.org.rapid_framework.test.util;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Array;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import org.springframework.beans.BeanUtils;

/**
 * 
 * 用于批量assert Bean的属性
 * 
 */
public class BeanAssert {

	public static void assertPropertiesNotNull(Object bean,String... ignoreProperties) {
		List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
		for(PropertyValue p : values) {
			if(p.getValue() == null) {
				String beanClass = bean.getClass().getSimpleName();
	    		throw new AssertionError("["+beanClass+"."+p.getName()+"] must be not null ");
			}
		}
	}

	public static void assertPropertiesNull(Object bean,String... ignoreProperties) {
		List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
		for(PropertyValue p : values) {
			if(p.getValue() != null) {
				String beanClass = bean.getClass().getSimpleName();
	    		throw new AssertionError("["+beanClass+"."+p.getName()+"] must be null ");
			}
		}
	}

	public static void assertPropertiesNotEmpty(Object bean,String... ignoreProperties) {
		List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
		for(PropertyValue p : values) {
			if(Utils.isEmpty(p.getValue())) {
				String beanClass = bean.getClass().getSimpleName();
	    		throw new AssertionError("["+beanClass+"."+p.getName()+"] must be not empty");
			}
		}
	}

	public static void assertPropertiesEmpty(Object bean,String... ignoreProperties) {
		List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
		for(PropertyValue p : values) {
			if(!Utils.isEmpty(p.getValue())) {
				String beanClass = bean.getClass().getSimpleName();
	    		throw new AssertionError("["+beanClass+"."+p.getName()+"] must be not empty");
			}
		}
	}

//    public static void assertNumberPropertiesEquals(Object bean,String expectedValue,String[] ignoreProperties) {
//        List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
//        for(PropertyValue p : values) {
//        	String actualValue = p.getValue() == null ? null : p.getValue().toString();
//        	System.out.println("out "+p.getName()+":"+actualValue);
//            if(p.getReturnType().isAssignableFrom(Number.class) || p.getReturnType().isPrimitive()) {
//            	System.out.println("\t in "+p.getName()+":"+actualValue);
//				if(! (expectedValue == p.getValue() || expectedValue.equals(actualValue))) {
//                    String beanClass = bean.getClass().getSimpleName();
//                    throw new AssertionError("["+beanClass+"."+p.getName()+"] must be equals:"+expectedValue+",actual value:"+p.getValue());                
//                }
//            }
//        }
//    }
//
//    public static void assertPrimitivePropertiesEquals(Object bean,String expectedValue,String[] ignoreProperties) {
//        List<PropertyValue> values = Utils.getPropertyValues(bean, ignoreProperties);
//        for(PropertyValue p : values) {
//            if(p.getReturnType().isPrimitive() || p.getReturnType().isAssignableFrom(String.class)) {
//                String actualValue = p.getValue() == null ? null : p.getValue().toString();
//                if(!(expectedValue == p.getValue() || expectedValue.equals(actualValue))) {
//                    String beanClass = bean.getClass().getSimpleName();
//                    throw new AssertionError("["+beanClass+"."+p.getName()+"] must be equals:"+expectedValue+",actual value:"+p.getValue());                
//                }
//            }
//        }
//    }
    
	private static class Utils {
    	private static List<PropertyValue> getPropertyValues(Object bean,String... ignoreProperties) throws Error {
    		PropertyDescriptor[] descriptors = BeanUtils.getPropertyDescriptors(bean.getClass());
    		
    		List<String> ignoreList = (ignoreProperties != null) ? Arrays.asList(ignoreProperties) : new ArrayList(0);
    		
    		List<PropertyValue> propertyValues = new ArrayList();
    		for(int i = 0; i < descriptors.length; i++ ) {
                String name = descriptors[i].getName();
                Method readMethod = descriptors[i].getReadMethod();
                if("class".equals(name)) {
                	continue;
                }
                if (readMethod != null && !ignoreList.contains(name)) {
    				Object value = invokeMethod(bean, name, readMethod);
    				PropertyValue pv = new PropertyValue(name,value,readMethod.getReturnType()); 
    				propertyValues.add(pv);
                }
            }
    		return propertyValues;
    	}
    	
    	private static Object invokeMethod(Object bean, String name, Method readMethod)throws Error {
    		try {
    			return readMethod.invoke(bean);
    		} catch (IllegalArgumentException e) {
    			throw new Error("AssertionError, get property value:["+bean.getClass().getSimpleName()+"."+name+"] occer error.",e);
    		} catch (IllegalAccessException e) {
    			throw new Error("AssertionError, get property value:["+bean.getClass().getSimpleName()+"."+name+"] occer error.",e);
    		} catch (InvocationTargetException e) {
    			throw new Error("AssertionError, get property value:["+bean.getClass().getSimpleName()+"."+name+"] occer error.",e);
    		}
    	}
    	
    	public static boolean isEmpty(Object o) throws IllegalArgumentException {
    		if(o == null) return true;
    
    		if(o instanceof String) {
    			if(((String)o).length() == 0){
    				return true;
    			}
    		} else if(o instanceof Collection) {
    			if(((Collection)o).isEmpty()){
    				return true;
    			}
    		} else if(o.getClass().isArray()) {
    			if(Array.getLength(o) == 0){
    				return true;
    			}
    		} else if(o instanceof Map) {
    			if(((Map)o).isEmpty()){
    				return true;
    			}
    		}else {
    			return false;
    		}
    
    		return false;
    	}
	}
    private static class PropertyValue {
        String name;
        Object value;
        Class<?> returnType;
        
        public PropertyValue(String name, Object value, Class<?> returnType) {
            super();
            this.name = name;
            this.value = value;
            this.returnType = returnType;
        }
        public String getName() {
            return name;
        }
        public Object getValue() {
            return value;
        }
        public Class getReturnType() {
            return returnType;
        }
        
    }
}