def testLoadData(self): """ _testLoadData_ Create a job and save it to the database. Load it back from the database using the name and the id. Verify that all job information is correct including input files and the job mask. """ testJobA = self.createTestJob() testJobA["mask"]["FirstEvent"] = 1 testJobA["mask"]["LastEvent"] = 2 testJobA["mask"]["FirstLumi"] = 3 testJobA["mask"]["LastLumi"] = 4 testJobA["mask"]["FirstRun"] = 5 testJobA["mask"]["LastRun"] = 6 testJobA.save() testJobB = Job(id=testJobA["id"]) testJobC = Job(name=testJobA["name"]) testJobB.loadData() testJobC.loadData() assert ( (testJobA["id"] == testJobB["id"]) and (testJobA["name"] == testJobB["name"]) and (testJobA["jobgroup"] == testJobB["jobgroup"]) and (testJobA["couch_record"] == testJobB["couch_record"]) and (testJobA["location"] == testJobB["location"]) ), "ERROR: Load from ID didn't load everything correctly" assert ( (testJobA["id"] == testJobC["id"]) and (testJobA["name"] == testJobC["name"]) and (testJobA["jobgroup"] == testJobC["jobgroup"]) and (testJobA["couch_record"] == testJobC["couch_record"]) and (testJobA["location"] == testJobC["location"]) ), "ERROR: Load from name didn't load everything correctly" assert testJobA["mask"] == testJobB["mask"], "ERROR: Job mask did not load properly" assert testJobA["mask"] == testJobC["mask"], "ERROR: Job mask did not load properly" goldenFiles = testJobA.getFiles() for testFile in testJobB.getFiles(): assert testFile in goldenFiles, "ERROR: Job loaded an unknown file" goldenFiles.remove(testFile) assert len(goldenFiles) == 0, "ERROR: Job didn't load all files" goldenFiles = testJobA.getFiles() for testFile in testJobC.getFiles(): assert testFile in goldenFiles, "ERROR: Job loaded an unknown file" goldenFiles.remove(testFile) assert len(goldenFiles) == 0, "ERROR: Job didn't load all files" return
def testGetFiles(self): """ _testGetFiles_ Test the Job's getFiles() method. This should load the files from the database if they haven't been loaded already. """ testJobA = self.createTestJob() testJobB = Job(id=testJobA["id"]) testJobB.loadData() goldenFiles = testJobA.getFiles() for testFile in testJobB.getFiles(): assert testFile in goldenFiles, "ERROR: Job loaded an unknown file: %s" % testFile goldenFiles.remove(testFile) assert len(goldenFiles) == 0, "ERROR: Job didn't load all files" return
def testGetFiles(self): """ _testGetFiles_ Test the Job's getFiles() method. This should load the files from the database if they haven't been loaded already. """ testJobA = self.createTestJob() testJobB = Job(id=testJobA["id"]) testJobB.loadData() goldenFiles = testJobA.getFiles() for testFile in testJobB.getFiles(): assert testFile in goldenFiles, \ "ERROR: Job loaded an unknown file: %s" % testFile goldenFiles.remove(testFile) assert not goldenFiles, "ERROR: Job didn't load all files" return
def testLoadData(self): """ _testLoadData_ Create a job and save it to the database. Load it back from the database using the name and the id. Verify that all job information is correct including input files and the job mask. """ testJobA = self.createTestJob() testJobA["mask"]["FirstEvent"] = 1 testJobA["mask"]["LastEvent"] = 2 testJobA["mask"]["FirstLumi"] = 3 testJobA["mask"]["LastLumi"] = 4 testJobA["mask"]["FirstRun"] = 5 testJobA["mask"]["LastRun"] = 6 testJobA.save() testJobB = Job(id=testJobA["id"]) testJobC = Job(name=testJobA["name"]) testJobB.loadData() testJobC.loadData() assert (testJobA["id"] == testJobB["id"]) and \ (testJobA["name"] == testJobB["name"]) and \ (testJobA["jobgroup"] == testJobB["jobgroup"]) and \ (testJobA["couch_record"] == testJobB["couch_record"]) and \ (testJobA["location"] == testJobB["location"]), \ "ERROR: Load from ID didn't load everything correctly" assert (testJobA["id"] == testJobC["id"]) and \ (testJobA["name"] == testJobC["name"]) and \ (testJobA["jobgroup"] == testJobC["jobgroup"]) and \ (testJobA["couch_record"] == testJobC["couch_record"]) and \ (testJobA["location"] == testJobC["location"]), \ "ERROR: Load from name didn't load everything correctly" assert testJobA["mask"] == testJobB["mask"], \ "ERROR: Job mask did not load properly" assert testJobA["mask"] == testJobC["mask"], \ "ERROR: Job mask did not load properly" goldenFiles = testJobA.getFiles() for testFile in testJobB.getFiles(): assert testFile in goldenFiles, \ "ERROR: Job loaded an unknown file" goldenFiles.remove(testFile) assert len(goldenFiles) == 0, \ "ERROR: Job didn't load all files" goldenFiles = testJobA.getFiles() for testFile in testJobC.getFiles(): assert testFile in goldenFiles, \ "ERROR: Job loaded an unknown file" goldenFiles.remove(testFile) assert len(goldenFiles) == 0, \ "ERROR: Job didn't load all files" return