#!/usr/env python

# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

import re
import sys
import subprocess
from collections import namedtuple

""" Convert install-check results into a standardized JUnit XML format

Example of JUnit output:

<?xml version="1.0" encoding="UTF-8"?>
<testsuite tests="3">
    <testcase classname="foo1" name="ASuccessfulTest"/>
    <testcase classname="foo2" name="AnotherSuccessfulTest"/>
    <testcase classname="foo3" name="AFailingTest">
        <failure type="NotEnoughFoo"> details about failure </failure>
    </testcase>
</testsuite>
"""


TestResult = namedtuple("TestResult", 'name suite status duration message')


def _test_result_factory(workdir, install_check_log):
    """
    Args:
        @param install_check_log: File name containing results from install-check

    Returns:
        Next result of type test_result
    """
    with open(install_check_log, 'r') as ic_log:
        line = ic_log.readline()
        while line:
            m = re.match(r"^TEST CASE RESULT\|Module: (.*)\|(.*)\|(.*)\|Time: ([0-9]+)(.*)", line)
            if m:
                suite, name, status, duration = [m.group(i) for i in range(1, 5)]
                message = ""
                if status == 'FAIL':
                    # get the tmp file and log file containing error
                    # these two lines are output after each failure
                    tmp_file_line = ic_log.readline()
                    log_file_line = ic_log.readline()
                    failure_m = re.match(r".* Check the log at (.*)", log_file_line)
                    if failure_m:
                        log_file = failure_m.group(1)
                        try:
                            message = subprocess.check_output(['tail', '-n 100', workdir + log_file],
                                                              stderr=subprocess.STDOUT)
                        except subprocess.CalledProcessError as e:
                            message = e.output
                yield TestResult(name=name, suite=suite,
                                 status=status, duration=duration,
                                 message=message)
            line = ic_log.readline()
# ----------------------------------------------------------------------


def _add_header(out_log, n_tests):
    header = ['<?xml version="1.0" encoding="UTF-8"?>',
              '<testsuite tests="{0}">'.format(n_tests), '']
    out_log.write('\n'.join(header))


def _add_footer(out_log):
    header = ['', '</testsuite>']
    out_log.write('\n'.join(header))


def _add_test_case(out_log, test_results):
    for t in test_results:
        try:
            # convert duration from milliseconds to seconds
            duration = float(t.duration)/1000
        except TypeError:
            duration = 0.0
        output = ['<testcase classname="{t.suite}" name="{t.name}" '
                  'status="{t.status}" time="{d}">'.
                  format(t=t, d=duration)]
        if t.status == "FAIL":
            output.append("""<failure><![CDATA[
                          {0}
                          ]]></failure>""".format(t.message))
        output.append('</testcase>')
        out_log.write('\n'.join(output))


def main(workdir, install_check_log, test_output_log):

    # need number of test results - so have to create the iterable
    all_test_results = [i for i in _test_result_factory(workdir, install_check_log)]

    with open(test_output_log, 'w') as out_log:
        _add_header(out_log, len(all_test_results))
        _add_test_case(out_log, all_test_results)
        _add_footer(out_log)


if __name__ == "__main__":
    # argv[1] = working director
    # argv[2] = install check log
    # argv[3] = output file to store xml
    main(sys.argv[1], sys.argv[2], sys.argv[3])