com.scit.sling.test.MockValueMap.java Source code

Introduction

Here is the source code for com.scit.sling.test.MockValueMap.java

Source

/**
 * Copyright (c) 2007-2011 Southern Cross IT. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 3 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */
package com.scit.sling.test;

import java.lang.reflect.Array;
import java.math.BigDecimal;
import java.util.Calendar;
import java.util.Date;
import java.util.LinkedHashMap;
import java.util.Locale;

import org.apache.commons.lang.NotImplementedException;
import org.apache.sling.api.resource.ValueMap;
import org.joda.time.DateTime;
import org.joda.time.format.DateTimeFormat;
import org.joda.time.format.DateTimeFormatter;

/**
 * A Mock {@link ValueMap} that has some basic type conversion.
 * Dates should be stored as {@link DateTime} objects.
 */
public class MockValueMap extends LinkedHashMap<String, Object> implements ValueMap {
    private static final long serialVersionUID = 1L;

    public <T> T get(String name, Class<T> type) {
        return convertType(get(name), type);
    }

    public <T> T get(String name, T defaultValue) {
        @SuppressWarnings("unchecked")
        T val = (T) get(name, defaultValue.getClass());
        if (val == null) {
            return defaultValue;
        }
        return val;
    }

    @SuppressWarnings("unchecked")
    private <T> T convertType(Object o, Class<T> type) {
        if (o == null) {
            return null;
        }
        if (o.getClass().isArray() && type.isArray()) {
            if (type.getComponentType().isAssignableFrom(o.getClass().getComponentType())) {
                return (T) o;
            } else {
                // we need to convert the elements in the array
                Object array = Array.newInstance(type.getComponentType(), Array.getLength(o));
                for (int i = 0; i < Array.getLength(o); i++) {
                    Array.set(array, i, convertType(Array.get(o, i), type.getComponentType()));
                }
                return (T) array;
            }
        }
        if (o.getClass().isAssignableFrom(type)) {
            return (T) o;
        }
        if (String.class.isAssignableFrom(type)) {
            // Format dates
            if (o instanceof Calendar) {
                return (T) formatDate((Calendar) o);
            } else if (o instanceof Date) {
                return (T) formatDate((Date) o);
            } else if (o instanceof DateTime) {
                return (T) formatDate((DateTime) o);
            }
            return (T) String.valueOf(o);
        } else if (o instanceof DateTime) {
            DateTime dt = (DateTime) o;
            if (Calendar.class.isAssignableFrom(type)) {
                return (T) dt.toCalendar(Locale.getDefault());
            } else if (Date.class.isAssignableFrom(type)) {
                return (T) dt.toDate();
            } else if (Number.class.isAssignableFrom(type)) {
                return convertType(dt.getMillis(), type);
            }
        } else if (o instanceof Number && Number.class.isAssignableFrom(type)) {
            if (Byte.class.isAssignableFrom(type)) {
                return (T) (Byte) ((Number) o).byteValue();
            } else if (Double.class.isAssignableFrom(type)) {
                return (T) (Double) ((Number) o).doubleValue();
            } else if (Float.class.isAssignableFrom(type)) {
                return (T) (Float) ((Number) o).floatValue();
            } else if (Integer.class.isAssignableFrom(type)) {
                return (T) (Integer) ((Number) o).intValue();
            } else if (Long.class.isAssignableFrom(type)) {
                return (T) (Long) ((Number) o).longValue();
            } else if (Short.class.isAssignableFrom(type)) {
                return (T) (Short) ((Number) o).shortValue();
            } else if (BigDecimal.class.isAssignableFrom(type)) {
                return (T) new BigDecimal(o.toString());
            }
        } else if (o instanceof Number && type.isPrimitive()) {
            final Number num = (Number) o;
            if (type == byte.class) {
                return (T) new Byte(num.byteValue());
            } else if (type == double.class) {
                return (T) new Double(num.doubleValue());
            } else if (type == float.class) {
                return (T) new Float(num.floatValue());
            } else if (type == int.class) {
                return (T) new Integer(num.intValue());
            } else if (type == long.class) {
                return (T) new Long(num.longValue());
            } else if (type == short.class) {
                return (T) new Short(num.shortValue());
            }
        } else if (o instanceof String && Number.class.isAssignableFrom(type)) {
            if (Byte.class.isAssignableFrom(type)) {
                return (T) new Byte((String) o);
            } else if (Double.class.isAssignableFrom(type)) {
                return (T) new Double((String) o);
            } else if (Float.class.isAssignableFrom(type)) {
                return (T) new Float((String) o);
            } else if (Integer.class.isAssignableFrom(type)) {
                return (T) new Integer((String) o);
            } else if (Long.class.isAssignableFrom(type)) {
                return (T) new Long((String) o);
            } else if (Short.class.isAssignableFrom(type)) {
                return (T) new Short((String) o);
            } else if (BigDecimal.class.isAssignableFrom(type)) {
                return (T) new BigDecimal((String) o);
            }
        }
        throw new NotImplementedException(
                "Can't handle conversion from " + o.getClass().getName() + " to " + type.getName());
    }

    private String formatDate(DateTime o) {
        return DATE_TIME_FORMAT.print(o);
    }

    private String formatDate(Date o) {
        return DATE_TIME_FORMAT.print(new DateTime(o));
    }

    private String formatDate(Calendar o) {
        return DATE_TIME_FORMAT.print(new DateTime(o));
    }

    public static final DateTimeFormatter DATE_TIME_FORMAT = DateTimeFormat
            .forPattern("EEE MMM dd YYYY HH:mm:ss zZ");
}