[AIM] Fix teardown in NotificationTest

One local_api function patched in NotificationTest.setUp() wasn't
being undone in NotificationTest.tearDown(), causing intermittent
failures in other UTs.

Change-Id: I17db76eabc04b2cb2c4b9d62e98b62b3b00d6d2a
This commit is contained in:
Robert Kukura 2019-02-28 17:38:55 -05:00
parent 44e1913304
commit 5edbf6d1cf
1 changed files with 2 additions and 0 deletions

View File

@ -4584,6 +4584,8 @@ class NotificationTest(AIMBaseTestCase):
local_api._enqueue = self.orig_enqueue
local_api.send_or_queue_notification = (
self.orig_send_or_queue_notification)
local_api.send_or_queue_registry_notification = (
self.orig_send_or_queue_registry_notification)
local_api.post_notifications_from_queue = (
self.orig_post_notifications_from_queue)
local_api.discard_notifications_after_rollback = (