You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.6 KiB
49 lines
1.6 KiB
# Copyright 2016 IBM Corp. |
|
# |
|
# 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 |
|
|
|
|
|
NOTIFICATIONS = [] |
|
|
|
|
|
def reset(): |
|
del NOTIFICATIONS[:] |
|
|
|
|
|
FakeMessage = collections.namedtuple('Message', [ |
|
'publisher_id', 'priority', 'event_type', 'payload', 'context']) |
|
|
|
|
|
class FakeNotifier(object): |
|
|
|
def __init__(self, transport, publisher_id=None, driver=None, |
|
topic=None, serializer=None, retry=None): |
|
self.transport = transport |
|
self.publisher_id = publisher_id or 'fake.id' |
|
for priority in ('debug', 'info', 'warn', 'error', 'critical'): |
|
setattr( |
|
self, priority, |
|
functools.partial(self._notify, priority=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=publisher_id) |
|
|
|
def _notify(self, ctxt, event_type, payload, priority): |
|
msg = FakeMessage(self.publisher_id, priority, event_type, |
|
payload, ctxt) |
|
NOTIFICATIONS.append(msg)
|
|
|