#!/bin/bash

SHUNIT_FAILURES=0
SHUNIT_ERRORS=0
SHUNIT_TESTS=0
SHUNIT_TIMERSTART=$(date "+%s")

function junit_header()
{
    if [ "$1" == "--help" ]; then
	cat <<EOF
    junit_header()

    This function doesn't actually do anything, it's a stub. Because junit
    is XML, which has to be ordered a certain way, this function does nothing
    but initialize some counters. Your unit test output won't actually get
    printed until you call junit_footer, which calls junit_header_real.
EOF
	return 1
    fi
    SHUNIT_FAILURES=0
    SHUNIT_ERRORS=0
    SHUNIT_TESTS=0
    SHUNIT_TIMERSTART=$(date "+%s")
    > /tmp/$$.junit
}

function junit_header_real()
{
    if [ "$1" == "--help" ]; then
	cat <<EOF
    junit_header_real(tests, errors, failures, elapsed[, encoding])

    Generate a junit header, to display one or more junit_testcase()s.
    tests : total number of tests.
    errors : total number of errors
    failures : total number of failed test cases
    elapsed : The total number of seconds elapsed during testing
    encoding : Defaults to UTF-8, you should probably not change this.
EOF
	return 1
    fi

    local tests errors failures elapsed encoding

    tests=$1
    errors=$2
    failures=$3
    elapsed=$4
    encoding=$5

    if [ "$encoding" == "" ]; then
	encoding="UTF-8"
    fi

    echo '<?xml version="1.0" encoding="'$encoding'"?>'
    echo '<testsuites>'
    echo '<testsuite name="main" failures="'$failures'" time="'$elapsed'" timestamp="'$(date +"%Y-%m-%dT%H:%M:%S%:z")'" errors="'$failures'" tests="'$tests'">'
    echo '    <properties/>'
}

function junit_footer()
{
    if [ "$1" == "--help" ]; then
	cat <<EOF
    junit_footer()

    Print a junit footer. This will also call junit_header_real.
EOF
	return 1
    fi
    ELAPSED=$(expr $(date "+%s") - $SHUNIT_TIMERSTART)
    junit_header_real $SHUNIT_TESTS $SHUNIT_ERRORS $SHUNIT_FAILURES $ELAPSED "UTF-8"
    cat /tmp/$$.junit
    rm -f /tmp/$$.junit
    echo '</testsuite>'
    echo '</testsuites>'
    return 0
}

function junit_testcase()
{
    if [ "$1" == "--help" ]; then
	cat <<EOF
    junit_testcase(classname, testname, elapsed[, failtype, failmsg, cdata])

    Generate XML to describe a junit test case.

    classname : the name of the class containing the test
    testname : the name of the test in this class
    elapsed : Number of seconds elapsed during testing
    failtype : A one-word description of the failure type (generally an exception name or code)
    failmsg : A brief message (<32 chars) describing the failure
    cdata : More detailed information about the failure.
EOF
	return 1
    fi

    local classname testname elapsed failtype failmsg cdata

    classname="$1"
    testname="$2"
    elapsed=$3
    failtype="$4"
    failmsg="$5"
    cdata="$6"
    SHUNIT_TESTS=$(expr $SHUNIT_TESTS + 1)
    echo '    <testcase classname="'$classname'" time="'$elapsed'" name="'$testname'">' >> /tmp/$$.junit
    echo '        <system-out></system-out>' >> /tmp/$$.junit
    echo '        <system-err></system-err>' >> /tmp/$$.junit
    if [ "$failtype" != "" ]; then
	SHUNIT_ERRORS=$(expr $SHUNIT_ERRORS + 1)
	SHUNIT_FAILURES=$(expr $SHUNIT_FAILURES + 1)
        echo '        <failure type="'$failtype'" message="'$failmsg'">' >> /tmp/$$.junit
        echo '            <![CDATA[' >> /tmp/$$.junit
	echo "${cdata}" >> /tmp/$$.junit
        echo '            ]]>' >> /tmp/$$.junit
        echo '        </failure>' >> /tmp/$$.junit
    fi
    echo '    </testcase>' >> /tmp/$$.junit
}