def processCppTestingFunction_(self, doAdd, functionName, classNames=None, functionNames=None): mainFileName = os.path.join(TEST_FILES_FOLDER, "testAnnotateMainCpp.cpp") originalFileName = os.path.join(TEST_FILES_FOLDER, functionName + ".h") resultFileName = os.path.join(TEST_FILES_FOLDER, functionName + "_result.h") expectedFileName = os.path.join(TEST_FILES_FOLDER, functionName + "_expected.h") if not doAdd: aux = originalFileName originalFileName = expectedFileName expectedFileName = aux mySourceCodeParser = SourceCodeParser( ProgramExecution.Languages.C_PLUS_PLUS, mainFileName) headerToInclude = functionName + ".h" #Verify second (un)annotation yields the same file try: originalFileBkp = originalFileName + '.bkp' shutil.copy(originalFileName, originalFileBkp) for i in range(2): try: ok = False if doAdd: if classNames is None and functionNames is None: mySourceCodeParser.annotateCppFile( originalFileName, resultFileName, headerToInclude) else: if classNames is not None: if functionNames is not None: assert len(classNames) == 1 mySourceCodeParser.annotateCppClassFunctions( originalFileName, resultFileName, headerToInclude, classNames[0], functionNames) else: mySourceCodeParser.annotateCppClasses( originalFileName, resultFileName, headerToInclude, classNames) if functionNames is not None and classNames is None: mySourceCodeParser.annotateCppFunctions( originalFileName, resultFileName, headerToInclude, functionNames) else: if classNames is None and functionNames is None: mySourceCodeParser.unannotateCppFile( originalFileName, resultFileName) else: if classNames is not None: mySourceCodeParser.unannotateCppClasses( originalFileName, resultFileName, classNames) if functionNames is not None and classNames is None: mySourceCodeParser.unannotateCppFunctions( originalFileName, resultFileName, functionNames) self.compareFileContents_(resultFileName, expectedFileName) ok = True finally: if i == 0 and ok: shutil.copy(resultFileName, originalFileName) if os.path.exists(resultFileName): os.remove(resultFileName) finally: if os.path.exists(originalFileBkp): shutil.copy(originalFileBkp, originalFileName) os.remove(originalFileBkp)