summaryrefslogtreecommitdiff
path: root/nova/tests/unit/fake_notifier.py
diff options
context:
space:
mode:
authorSean Dague <sean@dague.net>2014-11-07 14:27:03 +0100
committerSean Dague <sean@dague.net>2014-11-12 15:31:08 -0500
commit89cd6a0c493e26b5a9e017c99d731464292abbaf (patch)
treec2bf790d1684cd539b820247113492495123a163 /nova/tests/unit/fake_notifier.py
parent5c8bbaafef590e4d346a03051a0ba55c8be26c5c (diff)
downloadnova-89cd6a0c493e26b5a9e017c99d731464292abbaf.tar.gz
move all tests to nova/tests/unit
As part of the split of functional and unit tests we need to isolate the unit tests into a separate directory for having multiple test targets in a sane way. Part of bp:functional-tests-for-nova Change-Id: Id42ba373c1bda6a312b673ab2b489ca56da8c628
Diffstat (limited to 'nova/tests/unit/fake_notifier.py')
-rw-r--r--nova/tests/unit/fake_notifier.py69
1 files changed, 69 insertions, 0 deletions
diff --git a/nova/tests/unit/fake_notifier.py b/nova/tests/unit/fake_notifier.py
new file mode 100644
index 0000000000..110418215d
--- /dev/null
+++ b/nova/tests/unit/fake_notifier.py
@@ -0,0 +1,69 @@
+# Copyright 2013 Red Hat, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance with the License. You may obtain
+# a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations
+# under the License.
+
+import collections
+import functools
+
+import anyjson
+from oslo import messaging
+
+from nova import rpc
+
+NOTIFICATIONS = []
+
+
+def reset():
+ del NOTIFICATIONS[:]
+
+
+FakeMessage = collections.namedtuple('Message',
+ ['publisher_id', 'priority',
+ 'event_type', 'payload'])
+
+
+class FakeNotifier(object):
+
+ def __init__(self, transport, publisher_id, serializer=None):
+ self.transport = transport
+ self.publisher_id = publisher_id
+ self._serializer = serializer or messaging.serializer.NoOpSerializer()
+
+ for priority in ['debug', 'info', 'warn', 'error', 'critical']:
+ setattr(self, priority,
+ functools.partial(self._notify, priority.upper()))
+
+ def prepare(self, publisher_id=None):
+ if publisher_id is None:
+ publisher_id = self.publisher_id
+ return self.__class__(self.transport, publisher_id,
+ serializer=self._serializer)
+
+ def _notify(self, priority, ctxt, event_type, payload):
+ payload = self._serializer.serialize_entity(ctxt, payload)
+ # NOTE(sileht): simulate the kombu serializer
+ # this permit to raise an exception if something have not
+ # been serialized correctly
+ anyjson.serialize(payload)
+ msg = FakeMessage(self.publisher_id, priority, event_type, payload)
+ NOTIFICATIONS.append(msg)
+
+
+def stub_notifier(stubs):
+ stubs.Set(messaging, 'Notifier', FakeNotifier)
+ if rpc.NOTIFIER:
+ stubs.Set(rpc, 'NOTIFIER',
+ FakeNotifier(rpc.NOTIFIER.transport,
+ rpc.NOTIFIER.publisher_id,
+ serializer=getattr(rpc.NOTIFIER, '_serializer',
+ None)))