summaryrefslogtreecommitdiff
path: root/ceilometer/tests/event/test_endpoint.py
diff options
context:
space:
mode:
Diffstat (limited to 'ceilometer/tests/event/test_endpoint.py')
-rw-r--r--ceilometer/tests/event/test_endpoint.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/ceilometer/tests/event/test_endpoint.py b/ceilometer/tests/event/test_endpoint.py
index 3ffe0721..b872d5a2 100644
--- a/ceilometer/tests/event/test_endpoint.py
+++ b/ceilometer/tests/event/test_endpoint.py
@@ -22,6 +22,7 @@ from oslo.config import fixture as fixture_config
import oslo.messaging
from stevedore import extension
+import ceilometer
from ceilometer.event import endpoint as event_endpoint
from ceilometer.storage import models
from ceilometer.tests import base as tests_base
@@ -108,7 +109,8 @@ class TestEventEndpoint(tests_base.BaseTestCase):
@mock.patch('ceilometer.event.endpoint.LOG')
def test_event_not_implemented(self, log):
- self.mock_dispatcher.record_events.side_effect = NotImplementedError
+ re = self.mock_dispatcher.record_events
+ re.side_effect = ceilometer.NotImplementedError
message = {'event_type': "foo", 'message_id': "abc"}
ret = self.endpoint.process_notification(message)
log.warn.assert_called_once_with(