TestUtils.java

package com.github.mygreen.supercsv.tool;

import java.io.PrintWriter;
import java.lang.annotation.Annotation;
import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.security.SecureRandom;
import java.sql.Time;
import java.sql.Timestamp;
import java.text.DecimalFormat;
import java.text.SimpleDateFormat;
import java.time.format.DateTimeFormatter;
import java.time.temporal.TemporalAccessor;
import java.util.Arrays;
import java.util.Calendar;
import java.util.Comparator;
import java.util.Date;
import java.util.Locale;
import java.util.Optional;
import java.util.TimeZone;

import org.joda.time.ReadablePartial;
import org.joda.time.format.DateTimeFormat;
import org.supercsv.cellprocessor.CellProcessorAdaptor;
import org.supercsv.cellprocessor.ift.CellProcessor;
import org.supercsv.util.CsvContext;

import com.github.mygreen.supercsv.builder.ColumnMapping;
import com.github.mygreen.supercsv.builder.FieldAccessor;

/**
 * Utility methods and constants for tests.
 * 
 * @since 1.2
 * @author T.TSUCHIE
 *
 */
public class TestUtils {
    
    public static final CsvContext  ANONYMOUS_CSVCONTEXT = new CsvContext(1, 2, 3);
    
    /**
     * create Chracter from str.
     * @param value
     * @return
     */
    public static Character toCharacter(final String value) {
        return value.charAt(0);
    }
    
    /**
     * create Byte from str.
     * @param value
     * @return
     */
    public static Byte toByte(final String value) {
        return Byte.parseByte(value);
    }
    
    /**
     * create Intger from str.
     * @param value
     * @return
     */
    public static Short toShort(final String value) {
        return Short.parseShort(value);
    }
    
    /**
     * create Intger from str.
     * @param value
     * @return
     */
    public static Integer toInteger(final String value) {
        return Integer.parseInt(value);
    }
    
    /**
     * create Long from str.
     * @param value
     * @return
     */
    public static Long toLong(final String value) {
        return Long.parseLong(value);
    }
    
    /**
     * create Long from str.
     * @param value
     * @return
     */
    public static Float toFloat(final String value) {
        return Float.parseFloat(value);
    }
    
    /**
     * create Long from str.
     * @param value
     * @return
     */
    public static Double toDouble(final String value) {
        return Double.parseDouble(value);
    }
    
    public static BigDecimal toBigDecimal(final String value) {
        return new BigDecimal(value);
    }
    
    public static BigInteger toBigInteger(final String value) {
        return new BigInteger(value);
    }
    
    /**
     * create Date instance.
     * @param year
     * @param month Month(start with 1)
     * @param dayOfMonth
     * @return
     */
    public static Date toDate(final int year, final int month, final int dayOfMonth) {
        
        return toDate(year, month, dayOfMonth, 0, 0, 0);
    }
    
    /**
     * create Date instance.
     * @param year
     * @param month Month(start with 1)
     * @param dayOfMonth
     * @param hour
     * @param minute
     * @param second
     * @return
     */
    public static Date toDate(final int year, final int month, final int dayOfMonth,
            final int hour, final int minute, final int second) {
        
        Calendar cal = Calendar.getInstance();
        cal.set(year, month-1, dayOfMonth, hour, minute, second);
        cal.set(Calendar.MILLISECOND, 0);
        return cal.getTime();
    }
    
    public static Calendar toCalendar(final Date date) {
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        return cal;
    }
    
    public static Calendar toCalendar(final Date date, final Locale locale) {
        Calendar cal = Calendar.getInstance(locale);
        cal.setTime(date);
        return cal;
    }
    
    public static Calendar toCalendar(final Date date, final TimeZone timeZone) {
        Calendar cal = Calendar.getInstance(timeZone);
        cal.setTime(date);
        return cal;
    }
    
    public static Calendar toCalendar(final Date date, final TimeZone timeZone, final Locale locale) {
        Calendar cal = Calendar.getInstance(timeZone, locale);
        cal.setTime(date);
        return cal;
    }

    public static java.sql.Date toSqlDate(final int year, final int month, final int dayOfMonth) {
        return new java.sql.Date(toDate(year, month, dayOfMonth).getTime());
    }
    
    public static java.sql.Date toSqlDate(final Date date) {
        return new java.sql.Date(date.getTime());
    }
    
    public static Timestamp toTimestamp(final int year, final int month, final int dayOfMonth,
            final int hour, final int minute, final int second, final int millsecond) {
        Calendar cal = Calendar.getInstance();
        cal.set(year, month-1, dayOfMonth, hour, minute, second);
        cal.set(Calendar.MILLISECOND, millsecond);
        
        return new Timestamp(cal.getTimeInMillis());
    }
    
    public static Timestamp toTimestamp(final Date date) {
        return new Timestamp(date.getTime());
    }
    
    public static Time toTime(final int hour, final int minute, final int second) {
        
        Calendar cal = Calendar.getInstance();
        cal.set(1970, 0, 1, hour, minute, second);
        cal.set(Calendar.MILLISECOND, 0);
        return new Time(cal.getTime().getTime());
    }
    
    public static Time toTime(final Date date) {
        return new Time(date.getTime());
    }

    
    public static Date plusDays(final Date date, final int daysToAdd) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.DAY_OF_MONTH, daysToAdd);
        
        return cal.getTime();
        
    }
    
    public static Date minusDays(final Date date, final int daysToSubstract) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.DAY_OF_MONTH, -daysToSubstract);
        
        return cal.getTime();
        
    }
    
    public static Date plusHours(final Date date, final int hoursToAdd) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.HOUR_OF_DAY, hoursToAdd);
        
        return cal.getTime();
        
    }
    
    public static Date minusHours(final Date date, final int hoursToSubstract) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.HOUR_OF_DAY, -hoursToSubstract);
        
        return cal.getTime();
        
    }
    
    public static Date plusSeconds(final Date date, final int secondsToAdd) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.SECOND, secondsToAdd);
        
        return cal.getTime();
        
    }
    
    public static Date minusSeconds(final Date date, final int secondsToSubstract) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.SECOND, -secondsToSubstract);
        
        return cal.getTime();
        
    }
    
    public static Date plusMillseconds(final Date date, final int millsecondsToAdd) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.MILLISECOND, millsecondsToAdd);
        
        return cal.getTime();
        
    }
    
    public static Date minusMillseconds(final Date date, final int millsecondsToSubstract) {
        
        Calendar cal = Calendar.getInstance();
        cal.setTime(date);
        
        cal.add(Calendar.MILLISECOND, -millsecondsToSubstract);
        
        return cal.getTime();
        
    }
    
    public static Annotation[] getAnnotations(final Class<?> clazz, final String fieldName) {
        try {
            Field field = clazz.getDeclaredField(fieldName);
            field.setAccessible(true);
            return field.getAnnotations();
        } catch(ReflectiveOperationException e) {
            throw new RuntimeException(e);
        }
    }
    
    public static void printCellProcessorChain(final Optional<CellProcessor> cellProcessor, final String message) {
        
        if(cellProcessor == null || !cellProcessor.isPresent()) {
            return;
        }
        
        printCellProcessorChain(cellProcessor.get(), message);
        
    }
    
    public static void printCellProcessorChain(final CellProcessor cellProcessor, final String message) {
        
        if(cellProcessor == null) {
            return;
        }
        
        System.out.printf("======= print CellProcessor chain structures. :: %s ========\n", message);
        printCellProcessorChain(cellProcessor, new PrintWriter(System.out));
        System.out.println();
    }
    
    public static void printCellProcessorChain(final Optional<CellProcessor> cellProcessor, final PrintWriter writer) {
        if(cellProcessor == null || !cellProcessor.isPresent()) {
            return;
        }
        
        printCellProcessorChain(cellProcessor.get(), writer);
    }
    
    public static void printCellProcessorChain(final CellProcessor cellProcessor, final PrintWriter writer) {
        
        String index = "";
        CellProcessor cp = cellProcessor;
        do {
            if(index.length() == 0) {
                writer.printf("%s%s\n", index, cp.getClass().getName());
                writer.flush();
            } else {
                writer.printf("%s└%s\n", index, cp.getClass().getName());
                writer.flush();
            }
            
            // next processor
            try {
                if(cp instanceof CellProcessorAdaptor) {
                    Field field = CellProcessorAdaptor.class.getDeclaredField("next");
                    field.setAccessible(true);
                    cp = (CellProcessor) field.get(cp);
                } else {
                    break;
                }
                
            } catch(ReflectiveOperationException e) {
                return;
            }
            
            index += "    ";
            
        } while(cp != null);
        
    }
    
    public static final String format(final Date value, final String pattern) {
        
        SimpleDateFormat formatter = new SimpleDateFormat(pattern);
        return formatter.format(value);
        
    }
    
    public static final String format(final Number value, final String pattern) {
        
        DecimalFormat formatter = new DecimalFormat(pattern);
        return formatter.format(value);
        
    }
    
    public static final String format(final TemporalAccessor value, final String pattern) {
        
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern(pattern);
        return formatter.format(value);
        
    }
    
    public static final String format(final ReadablePartial value, final String pattern) {
        
        org.joda.time.format.DateTimeFormatter formatter = DateTimeFormat.forPattern(pattern);
        return formatter.print(value);
        
    }
    
    /**
     * ランダムに、大文字、小文字に変換する。
     * @param str
     * @return
     */
    public static final String toRandomCase(final String str) {
        
        StringBuilder sb = new StringBuilder();
        try {
            SecureRandom random = new SecureRandom();
            final int len = str.length();
            for(int i=0; i < len; i++) {
                char c = str.charAt(i);
                if(random.nextBoolean()) {
                    sb.append(String.valueOf(c).toLowerCase());
                } else {
                    sb.append(String.valueOf(c).toUpperCase());
                }
            }
        } catch(Exception e) {
            throw new RuntimeException(e);
        }
        return sb.toString();
        
    }
    
    /**
     * {@link FieldAccessor}のインスタンスを取得する。
     * @param clazz
     * @param fieldName
     * @param comparator
     * @return
     */
    public static FieldAccessor getFieldAccessor(final Class<?> clazz, final String fieldName, final Comparator<Annotation> comparator) {
        try {
            return new FieldAccessor(clazz.getDeclaredField(fieldName), comparator);
        } catch (NoSuchFieldException | SecurityException e) {
            throw new RuntimeException(e);
        }
    }
    
    /**
     * カラム情報を元にテスト用のCsvContextを作成する。
     * @param columnMapping カラム情報
     * @param input 入力値
     * @return lineNumber=1, rowNumber=2の値で固定。columnNumberはカラムから取得する。
     */
    public static CsvContext testCsvContext(final ColumnMapping columnMapping, final Object input) {
        
        return testCsvContext(columnMapping, input, 1, 2);
    }
    
    /**
     * カラム情報を元にテスト用のCsvContextを作成する。
     * @param columnMapping カラム情報
     * @param input 入力値
     * @param lineNumber 実際の行番号
     * @param rowNumber CSV上の行番号
     * @return
     */
    public static CsvContext testCsvContext(final ColumnMapping columnMapping, final Object input, final int lineNumber, final int rowNumber) {
        
        CsvContext context = new CsvContext(lineNumber, rowNumber, columnMapping.getNumber());
        
        Object[] rowSource = new Object[columnMapping.getNumber()];
        rowSource[columnMapping.getNumber() -1] = input;
        
        context.setRowSource(Arrays.asList(rowSource));
        
        return context;
        
    }
    
    /**
     * 実行している環境がWindowsかどうか
     * @return trueの場合、windows。
     */
    public static boolean isWindowsPlatform() {
        return System.getProperty("os.name").toLowerCase().startsWith("windows");
    }
    
    /**
     * 実行している環境がLinuxかどうか
     * @return trueの場合、Linux。
     */
    public static boolean isLinuxPlatform() {
        return System.getProperty("os.name").toLowerCase().startsWith("linux");
    }
}