def test_remove_expired(self):
# Create an 'expired' entry
threshold=5
task_id = "test_remove_expored_task_id"
created = datetime.now(tzutc()) - timedelta(seconds=threshold+1)
entry = SingleTaskInfo(task_id=task_id, owner_pid=os.getpid(), created=created)
entry.save()
self.assertEquals(1, SingleTaskInfo.objects(task_id=task_id).count())
retval = EnforceSingleTask.remove_expired(task_id, seconds_expire=threshold)
# Verify it is removed
self.assertIsNotNone(retval)
self.assertEquals(0, SingleTaskInfo.objects(task_id=task_id).count())
##
# Create an entry that is not expired
##
created = datetime.now(tzutc())
entry = SingleTaskInfo(task_id=task_id, owner_pid=os.getpid(), created=created)
entry.save()
self.assertEquals(1, SingleTaskInfo.objects(task_id=task_id).count())
retval = EnforceSingleTask.remove_expired(task_id, seconds_expire=threshold)
# Verify it is _not_ removed
self.assertIsNone(retval)
self.assertEquals(1, SingleTaskInfo.objects(task_id=task_id).count())
