scripts/convert_case_list_to_xml.py

#-------------------------------------------------------------------------
# drawElements Quality Program utilities
# --------------------------------------
#
# Copyright 2015 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 re
import sys
from xml.dom.minidom import Document

class TestCase:
	def __init__(self, casePath, description, caseType):
		self.casePath		= casePath
		self.description	= description
		self.caseType		= caseType
		self.children		= []

def findAllMatches(haystack, needle):
	matches = []
	ndx = -1
	while True:
		ndx = haystack.find(needle, ndx+1)
		if (ndx == -1):
			break
		matches.append(ndx)
	return matches

def createAncestors(casePath):
	parentCase = None
	for dotNdx in findAllMatches(casePath, "."):
		ancestorPath = casePath[:dotNdx]
		if ancestorPath not in caseNameHash:
			case = TestCase(ancestorPath, "Test Group", "TestGroup")
			parentCase.children.append(case)
			caseNameHash[ancestorPath] = case
			parentCase = case
		parentCase = caseNameHash[ancestorPath]
	return parentCase

def exportCase (doc, parent, testCase):
	#print(testCase.name, testCase.caseType)
	element = doc.createElement("TestCase")
	element.setAttribute("Name", testCase.casePath.rsplit(".", 2)[-1])
	element.setAttribute("Description", testCase.description)
	element.setAttribute("CaseType", testCase.caseType)
	parent.appendChild(element)
	for child in testCase.children:
		exportCase(doc, element, child)

# Main.

packageName = sys.argv[1]

rootCase = TestCase(packageName, packageName, "TestPackage" )
caseNameHash = { packageName:rootCase }
caseRE = re.compile(r"^\s*([a-zA-Z0-9_\.\-]+) '([^']*)' (\w+)\s*$".replace(" ", r"\s+"))

lines = open(packageName + ".cases").readlines()
numMatches = 0
for line in lines:
	line = line[:-1]
	if line.startswith(packageName + "."):
		m = caseRE.match(line)
		if m:
			casePath	= m.group(1)
			description	= m.group(2)
			caseType	= m.group(3)
			parent = createAncestors(casePath)
			parent.children.append(TestCase(casePath, description, caseType))
			numMatches += 1

# Create XML document.
doc = Document()
element = doc.createElement("TestCaseList")
doc.appendChild(element)
for testCase in rootCase.children:
	exportCase(doc, element, testCase)

# Dump XML document.
xml = doc.toprettyxml(indent="  ")
open(packageName + "-cases.xml", "wt").write(xml)

print("%d cases converted." % numMatches)