Move security notification work into its own method to allow for return values

Fixes #1302
Fixes #1304
This commit is contained in:
Joseph Schorr 2016-03-28 16:41:37 -04:00
parent 685dd1a925
commit d62ec22fc9
3 changed files with 15 additions and 3 deletions

View file

@ -506,7 +506,7 @@ class TestSecurityScanner(unittest.TestCase):
# Test with an unknown notification.
with HTTMock(get_notification, unknown_notification):
worker = SecurityNotificationWorker(None)
self.assertFalse(worker.process_queue_item({
self.assertFalse(worker.perform_notification_work({
'Name': 'unknownnotification'
}))
@ -517,7 +517,7 @@ class TestSecurityScanner(unittest.TestCase):
with HTTMock(get_notification, delete_notification, unknown_notification):
worker = SecurityNotificationWorker(None)
self.assertTrue(worker.process_queue_item(data))
self.assertTrue(worker.perform_notification_work(data))
self.assertEquals(['GET-1', 'GET-2', 'DELETE'], pages_called)