source: trunk/QA/parabix_wf.py @ 3028

import libxml2
from subprocess import *
import os
import shutil
import fnmatch                  
import xml.parsers.expat
from string import *

parabix_home = "/home/linda/workspace/parabix/proto/parabix2/src"
xmlconf_home = "/home/linda/workspace/parabix/trunk/QA/xmlconf"
output_home = "/home/linda/workspace/parabix/trunk/QA/testoutput"
output_files = output_home+"/Files"
output_messages = output_home+"/Messages"
program_to_test = "xmlwf"
edition5_program_to_test = 'markup_stats_5e'
testcases = []

subdir_prefix = ''

def load(filename):
    xml_conf = libxml2.createFileParserCtxt(filename)
    xml_conf.replaceEntities(1)
    xml_conf.parseDocument()
    doc = xml_conf.doc()
    return doc

def start_element(name, attrs):
        global subdir_prefix
        if name == 'TESTCASES':
                for a in attrs:
                        if a == 'xml:base':
                                subdir_prefix = '/'+attrs[a]            
        elif name == 'TEST':
                program = program_to_test
                #for a in attrs:
                        #if a == 'EDITION':
                                #if attrs[a] == '5':
                                        #program = edition5_program_to_test
                for a in attrs:
                        if a == 'RECOMMENDATION':
                          if attrs[a] == 'XML1.1':
                            return
                for a in attrs:                     
                        if a == 'URI':
                                addr = attrs[a].split('/')
                                filename = addr[len(addr)-1]
                                subdir = attrs[a][:attrs[a].find(filename)]
                                do_file(subdir, filename, program_to_test)

def parser(xml_file):
        p = xml.parsers.expat.ParserCreate()
        p.StartElementHandler = start_element
        p.Parse(xml_file, 1)
                
def do_file(subdir,filename,program):
        global testcases
        fulldir = subdir_prefix + subdir
        exe_f = open(xmlconf_home + fulldir + filename, 'r')
        filedata = exe_f.read()
        if filedata.find('DOCTYPE')==-1:
          encoding = filedata[:2].encode("hex")
          if encoding != '0000' and encoding != 'feff' and encoding != 'fffe' and encoding != '003c' and encoding != '3c00' and encoding != '4c6f':
            testcases.append((fulldir,filename,program))
        
          
def run_test():
    for testcase in testcases:  
        print testcase[0] + testcase[1]
        messagedir = output_messages + testcase[0]
        outputdir = output_files  + testcase[0]
        if os.path.exists(outputdir)==False:
                os.makedirs(outputdir)
        if os.path.exists(messagedir)==False:
                os.makedirs(messagedir)
        out_f = open(output_files + testcase[0] + testcase[1], 'w')
        mes_f = open(output_messages + testcase[0] + testcase[1], 'w')
        call([parabix_home + '/' + testcase[2], xmlconf_home + testcase[0] + testcase[1]], stderr=mes_f, stdout=out_f, cwd = xmlconf_home + testcase[0])
        out_f.close()
        mes_f.close()
        
                
if __name__ == "__main__":
        if os.path.exists(output_home + ".bak" ):
                shutil.rmtree(output_home + ".bak")
        if os.path.exists( output_home ):
                shutil.move(output_home, output_home + ".bak")
        os.mkdir(output_home)
        os.mkdir(output_files)
        os.mkdir(output_messages)
        
        doc = load(xmlconf_home+'/xmlconf.xml')
        parser(str(doc))
        run_test()