de.tsl2.nano.autotest.TextComparison Maven / Gradle / Ivy

Go to download
/*
 * File: $HeadURL$
 * Id  : $Id$
 * 
 * created by: Tom
 * created on: 26.01.2018
 * 
 * Copyright: (c) Thomas Schneider 2018, all rights reserved
 */
package de.tsl2.nano.autotest;

import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.Set;

import org.junit.Assert;

import de.tsl2.nano.core.util.StringUtil;

/**
 * Utility to compare to texts, ignoring given regular expressions and printing the differences
 * @author Tom
 * @version $Revision$ 
 */
public class TextComparison {
    public static String REGEX_DATE_DE = "\\d\\d\\.\\d\\d\\.\\d\\d\\d\\d";
    public static String REGEX_DATE_US = "\\d\\d\\d\\d\\-\\d\\d\\-\\d\\d";
    public static String REGEX_TIME_DE = "\\d\\d\\:\\d\\d(\\:\\d\\d([.,]\\d(\\d(\\d)?)?)?)?";
    public static String XXX = "XXX";
    
    public static void assertEquals(String exptected, String result, boolean ignoreWhitespace, Map replacements) {
        if (ignoreWhitespace) {
            exptected = exptected.replaceAll("\\s+", " ");
            result = result.replaceAll("\\s+", " ");
        }
        StringBuilder exp = new StringBuilder(exptected);
        StringBuilder res = new StringBuilder(result);
        prepareForComparison(exp, res, replacements);
        
        exptected = exp.toString();
        result = res.toString();
        Map diffs = getDiffs(exptected, result);
        String formDiff = StringUtil.toFormattedString(diffs, -1, true);
        if (diffs.size() > 0) {
            formDiff = "\n====================================================\n"
				+ "!!! DIFFERENCE BETWEEN EXPECTED AND RESULT:\n" + StringUtil.toFormattedString(diffs, -1, true)
            	+ "\n====================================================\n";
        }
        Assert.assertEquals(formDiff, exptected, result);
    }
    
    public static void prepareForComparison(StringBuilder expected, StringBuilder result,
            Map replacements) {
        Set keys = replacements.keySet();
        int ignored = 0;
        for (String regex : keys) {
            StringUtil.replaceAll(expected, regex, replacements.get(regex));
            ignored += StringUtil.replaceAll_(result, regex, replacements.get(regex));
        }
        System.out.println("preparedForComparison with " + replacements.size() + " replacements: ignored " + ignored
                + " positions");
    }
    
    public static Map getDiffs(String text1, String text2) {
        Scanner sc1 = null;
        Scanner sc2 = null;
        Map diffs = new HashMap<>();
        try {
            sc1 = new Scanner(text1);
            sc2 = new Scanner(text2);
            String s1 = null, s2 = null;
            while(sc1.hasNext() || sc2.hasNext()) {
                if (sc1.hasNext() )
                    s1 = sc1.next();
                if (sc2.hasNext() )
                    s2 = sc2.next();
                if (s1 != null && !s1.equals(s2))
                    diffs.put(s1,  s2);
            }
        } finally {
            if (sc1 != null)
                sc1.close();
            if (sc2 != null)
                sc2.close();
        }
        return diffs;
    }
}