def run(self):
##########################################
#Setup a file tree.
##########################################
tempFileTree = TempFileTree(os.path.join(self.getGlobalTempDir(), getRandomAlphaNumericString()))
fileTreeRootFile = tempFileTree.getTempFile()
makeFileTree(fileTreeRootFile, \
self.depth, tempFileTree)
treePointer = tempFileTree.getTempFile()
makeTreePointer(fileTreeRootFile, treePointer)
logger.info("We've set up the file tree")
##########################################
#Issue the child and follow on jobs
##########################################
self.addChildTarget(ChildTarget(treePointer))
self.setFollowOnTarget(DestructFileTree(tempFileTree))
logger.info("We've added the child target and finished SetupFileTree.run()")
def setupJobTree(tempFileTree, jobTreeDir, batchSystem, depth=2):
"""Sets up a job tree using the jobTreeSetup.py command.
"""
#Setup a job
retryCount = random.choice(xrange(1,10))
logger.info("Setup the basic files for the test")
fileTreeRootFile = tempFileTree.getTempFile()
makeFileTree(fileTreeRootFile, depth, tempFileTree)
treePointerFile = makeTreePointer(fileTreeRootFile, tempFileTree.getTempFile())
#Setup the job
command = "jobTreeTest_CommandFirst.py --treePointer %s --job JOB_FILE" % \
(treePointerFile)
jobTreeCommand = "jobTree --jobTree %s --retryCount %i\
--command '%s' --logLevel=INFO --maxJobDuration 100 --batchSystem %s" % \
(jobTreeDir, retryCount, command, batchSystem)
logger.info("Setup the job okay")
return (jobTreeCommand, fileTreeRootFile)