Mock out app NotificationSender during testing

This commit is contained in:
richard-dds
2019-05-15 10:45:42 -04:00
parent 97b9d84c38
commit 9e97df79f5
2 changed files with 17 additions and 1 deletions

View File

@@ -1,5 +1,8 @@
from flask import template_rendered
from contextlib import contextmanager
from unittest.mock import Mock
from atst.utils.notification_sender import NotificationSender
@contextmanager
@@ -37,3 +40,6 @@ class FakeLogger:
self.messages.append(msg)
if "extra" in kwargs:
self.extras.append(kwargs["extra"])
FakeNotificationSender = Mock(spec=NotificationSender)