def test_notify_decorator(self): func_name = self._decorated_function.__name__ # Decorated with notify_decorator like monkey_patch self._decorated_function = notifications.notify_decorator( func_name, self._decorated_function) ctxt = o_context.RequestContext() self._decorated_function(1, ctxt) self.assertEqual(1, len(fake_notifier.NOTIFICATIONS)) n = fake_notifier.NOTIFICATIONS[0] self.assertEqual(n.priority, 'INFO') self.assertEqual(n.event_type, func_name) self.assertEqual(n.context, ctxt) self.assertTrue(self.decorated_function_called)
def test_notify_decorator_wrong_default_notification_level(self): func_name = self._decorated_function.__name__ self.flags(default_notification_level='fake_level') self._decorated_function = notifications.notify_decorator( func_name, self._decorated_function) mock_info = mock.Mock() class FakeNotifier(object): info = mock_info mock_notifier_p = mock.patch.object( notifications.rpc, 'get_notifier', mock.Mock(return_value=FakeNotifier) ) with mock_notifier_p: self._decorated_function(1, 2) self.assertTrue(self.decorated_function_called) self.assertTrue(mock_info.called)