#!/usr/bin/env python
#
# Copyright (C) 2012 The Android Open Source Project
#
# Licensed 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 os
import sys
from xml.dom import Node
from xml.dom import minidom

def getChildrenWithTag(parent, tagName):
    children = []
    for child in  parent.childNodes:
        if (child.nodeType == Node.ELEMENT_NODE) and (child.tagName == tagName):
            #print "parent " + parent.getAttribute("name") + " " + tagName +\
            #    " " + child.getAttribute("name")
            children.append(child)
    return children

def parseSuite(suite, parentName):
    if parentName != "":
        parentName += '.'
    failedCases = []
    childSuites = getChildrenWithTag(suite, "TestSuite")
    for child in childSuites:
        for failure in parseSuite(child, parentName + child.getAttribute("name")):
            failedCases.append(failure)
    childTestCases = getChildrenWithTag(suite, "TestCase")
    for child in childTestCases:
        className = parentName + child.getAttribute("name")
        for test in getChildrenWithTag(child, "Test"):
            if test.getAttribute("result") != "pass":
                failureName = className + "#" + test.getAttribute("name")
                failedCases.append(failureName)
    #if len(failedCases) > 0:
    #    print failedCases
    return failedCases

def getFailedCases(resultXml):
    failedCases = []
    doc = minidom.parse(resultXml)
    testResult = doc.getElementsByTagName("TestResult")[0]
    packages = getChildrenWithTag(testResult, "TestPackage")
    for package in packages:
        casesFromChild = parseSuite(package, "")
        for case in casesFromChild:
            if case not in failedCases:
                failedCases.append(case)

    return failedCases

def main(argv):
    if len(argv) < 3:
        print "rerun.py cts_path result_xml [-s serial]"
        print " cts_path should end with android-cts"
        sys.exit(1)
    ctsPath = os.path.abspath(argv[1])
    resultXml = os.path.abspath(argv[2])
    deviceSerial = ""
    if len(argv) > 3:
        if argv[3] == "-s":
            deviceSerial = argv[4]

    failedCases = getFailedCases(resultXml)
    print "Re-run follwong cases:"
    for failure in failedCases:
        print " " + failure
    for failure in failedCases:
        [className, methodName] = failure.split('#')
        command = ctsPath + "/tools/cts-tradefed run singleCommand cts"
        if deviceSerial != "":
            command += " --serial " + deviceSerial
        command += " --class " + className + " --method " + methodName
        print command
        os.system(command)

if __name__ == '__main__':
    main(sys.argv)