Move security notification work into its own method to allow for return values
Fixes #1302 Fixes #1304
This commit is contained in:
parent
685dd1a925
commit
d62ec22fc9
3 changed files with 15 additions and 3 deletions
|
@ -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)
|
||||
|
||||
|
|
Reference in a new issue