diff options
Diffstat (limited to 'osprofiler/tests/unit/test_notifier.py')
-rw-r--r-- | osprofiler/tests/unit/test_notifier.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/osprofiler/tests/unit/test_notifier.py b/osprofiler/tests/unit/test_notifier.py index 7cccdad..8e658fe 100644 --- a/osprofiler/tests/unit/test_notifier.py +++ b/osprofiler/tests/unit/test_notifier.py @@ -23,6 +23,7 @@ class NotifierTestCase(test.TestCase): def tearDown(self): notifier.set(notifier._noop_notifier) # restore defaults + notifier.clear_notifier_cache() super(NotifierTestCase, self).tearDown() def test_set(self): @@ -49,3 +50,11 @@ class NotifierTestCase(test.TestCase): result = notifier.create("test", 10, b=20) mock_factory.assert_called_once_with("test", 10, b=20) self.assertEqual(mock_factory.return_value.notify, result) + + @mock.patch("osprofiler.notifier.base.get_driver") + def test_create_driver_init_failure(self, mock_get_driver): + mock_get_driver.side_effect = Exception() + + result = notifier.create("test", 10, b=20) + mock_get_driver.assert_called_once_with("test", 10, b=20) + self.assertEqual(notifier._noop_notifier, result) |