JavaScriptSharedUtilTest.java

/*
 * Copyright (c) Mirth Corporation. All rights reserved.
 * 
 * http://www.mirthcorp.com
 * 
 * The software in this package is published under the terms of the MPL license a copy of which has
 * been included with this distribution in the LICENSE.txt file.
 */

package com.mirth.connect.util;

import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;

import java.util.ArrayList;
import java.util.List;

import org.junit.Test;

import com.mirth.connect.util.JavaScriptSharedUtil.ExprPart;

public class JavaScriptSharedUtilTest {

    @Test
    public void testPrettyPrint() {
        String script = "for (var i = 0; i < getArrayOrXmlLength(msg['OBR']); i++) {for (var j = 0; j < getArrayOrXmlLength(msg['OBR'][i]['OBR.3']); j++) {if (typeof(tmp) == 'xml') {if (typeof(tmp['OBR'][i]) == 'undefined') {createSegment('OBR', tmp, i);}if (typeof(tmp['OBR'][i]['OBR.3'][j]) == 'undefined') {createSegment('OBR.3', tmp['OBR'][i], j);}} else {if (typeof(tmp) == 'undefined') {tmp = {};}if (typeof(tmp['OBR']) == 'undefined') {tmp['OBR'] = [];}if (typeof(tmp['OBR'][i]) == 'undefined') {tmp['OBR'][i] = {};}if (typeof(tmp['OBR'][i]['OBR.3']) == 'undefined') {tmp['OBR'][i]['OBR.3'] = [];}if (typeof(tmp['OBR'][i]['OBR.3'][j]) == 'undefined') {tmp['OBR'][i]['OBR.3'][j] = {};}if (typeof(tmp['OBR'][i]['OBR.3'][j]['OBR.3.1']) == 'undefined') {tmp['OBR'][i]['OBR.3'][j]['OBR.3.1'] = {};}}tmp['OBR'][i]['OBR.3'][j]['OBR.3.1']['OBR.3.1.1'] = validate(msg['OBR'][i]['OBR.3'][j]['OBR.3.1']['OBR.3.1.1'].toString(), '', new Array());}}";

        // @formatter:off
        String expected =
            "for (var i = 0; i < getArrayOrXmlLength(msg['OBR']); i++) {\n"+
            "\tfor (var j = 0; j < getArrayOrXmlLength(msg['OBR'][i]['OBR.3']); j++) {\n"+
            "\t\tif (typeof(tmp) == 'xml') {\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]) == 'undefined') {\n"+
            "\t\t\t\tcreateSegment('OBR', tmp, i);\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]['OBR.3'][j]) == 'undefined') {\n"+
            "\t\t\t\tcreateSegment('OBR.3', tmp['OBR'][i], j);\n"+
            "\t\t\t}\n"+
            "\t\t} else {\n"+
            "\t\t\tif (typeof(tmp) == 'undefined') {\n"+
            "\t\t\t\ttmp = {};\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR']) == 'undefined') {\n"+
            "\t\t\t\ttmp['OBR'] = [];\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]) == 'undefined') {\n"+
            "\t\t\t\ttmp['OBR'][i] = {};\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]['OBR.3']) == 'undefined') {\n"+
            "\t\t\t\ttmp['OBR'][i]['OBR.3'] = [];\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]['OBR.3'][j]) == 'undefined') {\n"+
            "\t\t\t\ttmp['OBR'][i]['OBR.3'][j] = {};\n"+
            "\t\t\t}\n"+
            "\t\t\tif (typeof(tmp['OBR'][i]['OBR.3'][j]['OBR.3.1']) == 'undefined') {\n"+
            "\t\t\t\ttmp['OBR'][i]['OBR.3'][j]['OBR.3.1'] = {};\n"+
            "\t\t\t}\n"+
            "\t\t}\n"+
            "\t\ttmp['OBR'][i]['OBR.3'][j]['OBR.3.1']['OBR.3.1.1'] = validate(msg['OBR'][i]['OBR.3'][j]['OBR.3.1']['OBR.3.1.1'].toString(), '', new Array());\n"+
            "\t}\n"+
            "}\n";
        // @formatter:on

        assertEquals(expected.trim(), JavaScriptSharedUtil.prettyPrint(script).trim());
    }

    @Test
    public void testPrettyPrintWithE4X() {
        String script = "var _results = Lists.list();\nfor (var i = 0; i < getArrayOrXmlLength(msg['OBX']); i++) {\n\ntFactory = Packages.javax.xml.transform.TransformerFactory.newInstance();\nxsltTemplate = new Packages.java.io.StringReader(<xsl:stylesheet version=\"1.0\" xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\">    <xsl:template match=\"/\">        <values>            <xsl:for-each select=\"OBX/OBX.5\">                <value><xsl:value-of select=\"OBX.5.1\"/></value>            </xsl:for-each>        </values>    </xsl:template></xsl:stylesheet>);\ntransformer = tFactory.newTransformer(new Packages.javax.xml.transform.stream.StreamSource(xsltTemplate));\nsourceVar = new Packages.java.io.StringReader(msg['OBX'][i]);\nresultVar = new Packages.java.io.StringWriter();\ntransformer.transform(new Packages.javax.xml.transform.stream.StreamSource(sourceVar), new Packages.javax.xml.transform.stream.StreamResult(resultVar));\n_results.add(resultVar.toString());\n\n\n}\nchannelMap.put('results', _results.toArray());";

        // @formatter:off
        String expected = 
            "var _results = Lists.list();\n"+
            "for (var i = 0; i < getArrayOrXmlLength(msg['OBX']); i++) {\n"+
            "\n"+
            "\ttFactory = Packages.javax.xml.transform.TransformerFactory.newInstance();\n"+
            "\txsltTemplate = new Packages.java.io.StringReader(<xsl:stylesheet version=\"1.0\" xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\">    <xsl:template match=\"/\">        <values>            <xsl:for-each select=\"OBX/OBX.5\">                <value><xsl:value-of select=\"OBX.5.1\"/></value>            </xsl:for-each>        </values>    </xsl:template></xsl:stylesheet>);\n"+
            "\ttransformer = tFactory.newTransformer(new Packages.javax.xml.transform.stream.StreamSource(xsltTemplate));\n"+
            "\tsourceVar = new Packages.java.io.StringReader(msg['OBX'][i]);\n"+
            "\tresultVar = new Packages.java.io.StringWriter();\n"+
            "\ttransformer.transform(new Packages.javax.xml.transform.stream.StreamSource(sourceVar), new Packages.javax.xml.transform.stream.StreamResult(resultVar));\n"+
            "\t_results.add(resultVar.toString());\n"+
            "\n"+
            "\n"+
            "}\n"+
            "channelMap.put('results', _results.toArray());\n";
        // @formatter:on

        assertEquals(expected.trim(), JavaScriptSharedUtil.prettyPrint(script).trim());
    }

    @Test
    public void testPrettyPrintWithE4XAndProlog() {
        String script = "var _results = Lists.list();\nfor (var i = 0; i < getArrayOrXmlLength(msg['OBX']); i++) {\n\n    tFactory = Packages.javax.xml.transform.TransformerFactory.newInstance();\n    xsltTemplate = new Packages.java.io.StringReader(<?xml version=\"1.0\" encoding=\"UTF-8\"?> <xsl:stylesheet version=\"1.0\" xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\">    <xsl:template match=\"/\">        <values>            <xsl:for-each select=\"OBX/OBX.5\">                <value><xsl:value-of select=\"OBX.5.1\"/></value>            </xsl:for-each>        </values>    </xsl:template></xsl:stylesheet>);\n    transformer = tFactory.newTransformer(new Packages.javax.xml.transform.stream.StreamSource(xsltTemplate));\n    sourceVar = new Packages.java.io.StringReader(msg['OBX'][i]);\n    resultVar = new Packages.java.io.StringWriter();\n    transformer.transform(new Packages.javax.xml.transform.stream.StreamSource(sourceVar), new Packages.javax.xml.transform.stream.StreamResult(resultVar));\n    _results.add(resultVar.toString());\n\n\n}\nchannelMap.put('results', _results.toArray());";

        // @formatter:off
        String expected = 
            "var _results = Lists.list();\n"+
            "for (var i = 0; i < getArrayOrXmlLength(msg['OBX']); i++) {\n"+
            "\n"+
            "\ttFactory = Packages.javax.xml.transform.TransformerFactory.newInstance();\n"+
            "\txsltTemplate = new Packages.java.io.StringReader(<?xml version=\"1.0\" encoding=\"UTF-8\"?> <xsl:stylesheet version=\"1.0\" xmlns:xsl=\"http://www.w3.org/1999/XSL/Transform\">    <xsl:template match=\"/\">        <values>            <xsl:for-each select=\"OBX/OBX.5\">                <value><xsl:value-of select=\"OBX.5.1\"/></value>            </xsl:for-each>        </values>    </xsl:template></xsl:stylesheet>);\n"+
            "\ttransformer = tFactory.newTransformer(new Packages.javax.xml.transform.stream.StreamSource(xsltTemplate));\n"+
            "\tsourceVar = new Packages.java.io.StringReader(msg['OBX'][i]);\n"+
            "\tresultVar = new Packages.java.io.StringWriter();\n"+
            "\ttransformer.transform(new Packages.javax.xml.transform.stream.StreamSource(sourceVar), new Packages.javax.xml.transform.stream.StreamResult(resultVar));\n"+
            "\t_results.add(resultVar.toString());\n"+
            "\n"+
            "\n"+
            "}\n"+
            "channelMap.put('results', _results.toArray());\n";
        // @formatter:on

        assertEquals(expected.trim(), JavaScriptSharedUtil.prettyPrint(script).trim());
    }

    @Test
    public void testRemoveNumberLiterals1() {
        String expression = "msg['OBR'][0]['OBR.3'][1]['OBR.3.1'].toString()";
        assertEquals("msg['OBR']['OBR.3']['OBR.3.1'].toString()", JavaScriptSharedUtil.removeNumberLiterals(expression));
    }

    @Test
    public void testRemoveNumberLiterals2() {
        String expression = "msg.OBR[0].*::['OBR.3'][1].ns::['OBR.3.1'][test].toString()";
        assertEquals("msg.OBR.*::['OBR.3'].ns::['OBR.3.1'][test].toString()", JavaScriptSharedUtil.removeNumberLiterals(expression));
    }

    @Test
    public void testGetExpressionParts1() {
        String expression = "msg['OBR'][0]['OBR.3'][1]['OBR.3.1']";

        List<ExprPart> expected = new ArrayList<ExprPart>();
        expected.add(new ExprPart("msg", "msg"));
        expected.add(new ExprPart("['OBR']", "'OBR'"));
        expected.add(new ExprPart("[0]", "0", true));
        expected.add(new ExprPart("['OBR.3']", "'OBR.3'"));
        expected.add(new ExprPart("[1]", "1", true));
        expected.add(new ExprPart("['OBR.3.1']", "'OBR.3.1'"));

        assertArrayEquals(expected.toArray(), JavaScriptSharedUtil.getExpressionParts(expression).toArray());
    }

    @Test
    public void testGetExpressionParts2() {
        String expression = "msg.OBR[0].*::['OBR.3'][1].ns::['OBR.3.1'][test]";

        List<ExprPart> expected = new ArrayList<ExprPart>();
        expected.add(new ExprPart("msg", "msg"));
        expected.add(new ExprPart(".OBR", "OBR"));
        expected.add(new ExprPart("[0]", "0", true));
        expected.add(new ExprPart(".*::['OBR.3']", "'OBR.3'"));
        expected.add(new ExprPart("[1]", "1", true));
        expected.add(new ExprPart(".ns::['OBR.3.1']", "'OBR.3.1'"));
        expected.add(new ExprPart("[test]", "test"));

        assertArrayEquals(expected.toArray(), JavaScriptSharedUtil.getExpressionParts(expression).toArray());
    }

    @Test
    public void testGetExpressionParts3() {
        assertArrayEquals(new Object[0], JavaScriptSharedUtil.getExpressionParts("  ").toArray());
    }

    @Test
    public void testGetExpressionParts4() {
        String expression = "#@$%.^&*(";

        List<ExprPart> expected = new ArrayList<ExprPart>();
        expected.add(new ExprPart("#@$%.^&*(", "#@$%.^&*("));

        assertArrayEquals(expected.toArray(), JavaScriptSharedUtil.getExpressionParts(expression).toArray());
    }
}