def testInvalidIDs(self):
logger = HarvesterLog(stateDir=self.stateDir, logDir=self.logDir, name='name')
logger.startRepository()
logger.notifyHarvestedRecord('id:1')
logger.logInvalidData('id:1', 'exception message')
logger.notifyHarvestedRecord('id:2')
logger.logInvalidData('id:2', 'exception message')
self.assertEquals(['id:1', 'id:2'], logger.invalidIds())
def testClearInvalidData(self):
logger = HarvesterLog(stateDir=self.stateDir, logDir=self.logDir, name='name')
logger.startRepository()
logger.notifyHarvestedRecord('repoid:oai:bla/bla')
logger.logInvalidData('repoid:oai:bla/bla', "Error")
self.assertTrue(isfile(self.logDir + '/invalid/repoid/oai:bla%2Fbla'))
logger.notifyHarvestedRecord('repoid:recordid')
logger.logInvalidData('repoid:recordid', "Error")
self.assertTrue(isfile(self.logDir + '/invalid/repoid/recordid'))
logger.notifyHarvestedRecord('repo2:1')
logger.logInvalidData('repo2:1', "Error")
self.assertTrue(isfile(self.logDir + '/invalid/repo2/1'))
self.assertEquals(['repoid:oai:bla/bla', 'repoid:recordid', 'repo2:1'], logger.invalidIds())
logger.clearInvalidData('repoid')
self.assertEquals(['repo2:1'], logger.invalidIds())
self.assertFalse(isfile(self.logDir + '/invalid/repoid/oai:bla%2Fbla'))
self.assertFalse(isfile(self.logDir + '/invalid/repoid/recordid'))
self.assertTrue(isfile(self.logDir + '/invalid/repo2/1'))
