def test_purge(self):
now = date_utils.get_datetime_utc_now()
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={
'hola': 'hi',
'kuraci': 'chicken'
},
occurrence_time=now -
timedelta(days=20),
status=TRIGGER_INSTANCE_PROCESSED)
TriggerInstance.add_or_update(instance_db)
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={
'hola': 'hi',
'kuraci': 'chicken'
},
occurrence_time=now -
timedelta(days=5),
status=TRIGGER_INSTANCE_PROCESSED)
TriggerInstance.add_or_update(instance_db)
self.assertEqual(len(TriggerInstance.get_all()), 2)
purge_trigger_instances(logger=LOG, timestamp=now - timedelta(days=10))
self.assertEqual(len(TriggerInstance.get_all()), 1)
def test_no_timestamp_doesnt_delete(self):
now = date_utils.get_datetime_utc_now()
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={'hola': 'hi', 'kuraci': 'chicken'},
occurrence_time=now - timedelta(days=20))
TriggerInstance.add_or_update(instance_db)
self.assertEqual(len(TriggerInstance.get_all()), 1)
purge_trigger_instances()
self.assertEqual(len(TriggerInstance.get_all()), 1)
def test_no_timestamp_doesnt_delete(self):
now = date_utils.get_datetime_utc_now()
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={'hola': 'hi', 'kuraci': 'chicken'},
occurrence_time=now - timedelta(days=20))
TriggerInstance.add_or_update(instance_db)
self.assertEqual(len(TriggerInstance.get_all()), 1)
expected_msg = 'Specify a valid timestamp'
self.assertRaisesRegexp(ValueError, expected_msg,
purge_trigger_instances,
logger=LOG, timestamp=None)
self.assertEqual(len(TriggerInstance.get_all()), 1)
def test_no_timestamp_doesnt_delete(self):
now = date_utils.get_datetime_utc_now()
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={'hola': 'hi', 'kuraci': 'chicken'},
occurrence_time=now - timedelta(days=20),
status=TRIGGER_INSTANCE_PROCESSED)
TriggerInstance.add_or_update(instance_db)
self.assertEqual(len(TriggerInstance.get_all()), 1)
expected_msg = 'Specify a valid timestamp'
self.assertRaisesRegexp(ValueError, expected_msg,
purge_trigger_instances,
logger=LOG, timestamp=None)
self.assertEqual(len(TriggerInstance.get_all()), 1)
def test_purge(self):
now = date_utils.get_datetime_utc_now()
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={'hola': 'hi', 'kuraci': 'chicken'},
occurrence_time=now - timedelta(days=20),
status=TRIGGER_INSTANCE_PROCESSED)
TriggerInstance.add_or_update(instance_db)
instance_db = TriggerInstanceDB(trigger='purge_tool.dummy.trigger',
payload={'hola': 'hi', 'kuraci': 'chicken'},
occurrence_time=now - timedelta(days=5),
status=TRIGGER_INSTANCE_PROCESSED)
TriggerInstance.add_or_update(instance_db)
self.assertEqual(len(TriggerInstance.get_all()), 2)
purge_trigger_instances(logger=LOG, timestamp=now - timedelta(days=10))
self.assertEqual(len(TriggerInstance.get_all()), 1)
def get_all(self, **kw):
"""
List all triggerinstances.
Handles requests:
GET /triggerinstances/
"""
trigger_instance_apis = [TriggerInstanceAPI.from_model(trigger_instance_db)
for trigger_instance_db in TriggerInstance.get_all(**kw)]
return trigger_instance_apis
