73c9150afe
Implement a Counter class for use by notification plugins. Define a base class for Notification plugins. Define a dispatcher class for notification events to be passed to the plugins. Add a notification plugin for instance creation and "instance" counters. Add a reusable function for turning a Counter into a metering event dictionary. Change-Id: Iaa626b98e1a661ed31cc8b8e95263c111df20888
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import os
|
|
import cPickle as pickle
|
|
from StringIO import StringIO
|
|
import sys
|
|
import types
|
|
|
|
import mox
|
|
|
|
from nova.rpc import impl_kombu
|
|
|
|
# The module being tested is part of the tools directory,
|
|
# so make sure it is in our import path.
|
|
sys.path.insert(0, os.path.normpath(os.path.join(os.path.dirname(__file__),
|
|
'..', 'tools')))
|
|
import notificationclient
|
|
|
|
|
|
def test_send_messages():
|
|
message = {'timestamp': 'date goes here',
|
|
'event_type': 'compute.instance.exists',
|
|
# real messages have more fields...
|
|
}
|
|
input = StringIO(pickle.dumps(message))
|
|
conn = mox.MockObject(impl_kombu.Connection)
|
|
conn.topic_send('notifications.info', message)
|
|
mox.Replay(conn)
|
|
notificationclient.send_messages(conn, input)
|
|
mox.Verify(conn)
|
|
return
|
|
|
|
|
|
def test_record_messages():
|
|
conn = mox.MockObject(impl_kombu.Connection)
|
|
conn.declare_topic_consumer('notifications.info',
|
|
mox.IsA(types.FunctionType))
|
|
conn.consume()
|
|
mox.Replay(conn)
|
|
notificationclient.record_messages(conn, StringIO())
|
|
mox.Verify(conn)
|
|
return
|