Adding notifications for volumes

Added notifications for volumes have been added with tests.
This includes create/delete/exists events for volumes.

blueprint nova-notifications

Change-Id: I21b74974fac22c3621ccf7564dc5c0d339f8751a

2 files changed, 118 insertions, 1 deletions

diff --git a/nova/tests/test_volume.py b/nova/tests/test_volume.py
index d979c3d0e..88e8146b6 100644
--- a/nova/tests/test_volume.py
+++ b/nova/tests/test_volume.py
@@ -29,6 +29,7 @@ from nova import exception
 from nova import db
 from nova import flags
 from nova import log as logging
+from nova.notifier import test_notifier
 from nova.openstack.common import importutils
 import nova.policy
 from nova import rpc
@@ -46,18 +47,21 @@ class VolumeTestCase(test.TestCase):
         super(VolumeTestCase, self).setUp()
         self.compute = importutils.import_object(FLAGS.compute_manager)
         self.flags(connection_type='fake')
+        self.stubs.Set(nova.flags.FLAGS, 'notification_driver',
+                'nova.notifier.test_notifier')
         self.volume = importutils.import_object(FLAGS.volume_manager)
         self.context = context.get_admin_context()
         instance = db.instance_create(self.context, {})
         self.instance_id = instance['id']
         self.instance_uuid = instance['uuid']
+        test_notifier.NOTIFICATIONS = []
 
     def tearDown(self):
         db.instance_destroy(self.context, self.instance_id)
         super(VolumeTestCase, self).tearDown()
 
     @staticmethod
-    def _create_volume(size='0', snapshot_id=None):
+    def _create_volume(size=0, snapshot_id=None):
         """Create a volume object."""
         vol = {}
         vol['size'] = size
@@ -88,11 +92,14 @@ class VolumeTestCase(test.TestCase):
         """Test volume can be created and deleted."""
         volume = self._create_volume()
         volume_id = volume['id']
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 0)
         self.volume.create_volume(self.context, volume_id)
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 2)
         self.assertEqual(volume_id, db.volume_get(context.get_admin_context(),
                          volume_id).id)
 
         self.volume.delete_volume(self.context, volume_id)
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 4)
         self.assertRaises(exception.NotFound,
                           db.volume_get,
                           self.context,
@@ -363,6 +370,30 @@ class VolumeTestCase(test.TestCase):
         self.volume.delete_snapshot(self.context, snapshot_id)
         self.volume.delete_volume(self.context, volume_id)
 
+    def test_create_volume_usage_notification(self):
+        """Ensure create volume generates appropriate usage notification"""
+        volume = self._create_volume()
+        volume_id = volume['id']
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 0)
+        self.volume.create_volume(self.context, volume_id)
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 2)
+        msg = test_notifier.NOTIFICATIONS[0]
+        self.assertEquals(msg['event_type'], 'volume.create.start')
+        msg = test_notifier.NOTIFICATIONS[1]
+        self.assertEquals(msg['priority'], 'INFO')
+        self.assertEquals(msg['event_type'], 'volume.create.end')
+        payload = msg['payload']
+        self.assertEquals(payload['tenant_id'], volume['project_id'])
+        self.assertEquals(payload['user_id'], volume['user_id'])
+        self.assertEquals(payload['volume_id'], volume['id'])
+        self.assertEquals(payload['status'], 'creating')
+        self.assertEquals(payload['size'], volume['size'])
+        self.assertTrue('display_name' in payload)
+        self.assertTrue('snapshot_id' in payload)
+        self.assertTrue('launched_at' in payload)
+        self.assertTrue('created_at' in payload)
+        self.volume.delete_volume(self.context, volume_id)
+
 
 class DriverTestCase(test.TestCase):
     """Base Test class for Drivers."""
diff --git a/nova/tests/test_volume_utils.py b/nova/tests/test_volume_utils.py
new file mode 100644
index 000000000..453c0d925
--- /dev/null
+++ b/nova/tests/test_volume_utils.py
@@ -0,0 +1,86 @@
+# vim: tabstop=4 shiftwidth=4 softtabstop=4
+
+# Copyright 2011 OpenStack LLC.
+# All Rights Reserved.
+#
+#    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.
+
+"""Tests For miscellaneous util methods used with volume."""
+
+from nova import db
+from nova import flags
+from nova import context
+from nova import test
+from nova import log as logging
+import nova.image.fake
+from nova.volume import utils as volume_utils
+from nova.notifier import test_notifier
+from nova.openstack.common import importutils
+
+
+LOG = logging.getLogger(__name__)
+FLAGS = flags.FLAGS
+
+
+class UsageInfoTestCase(test.TestCase):
+
+    def setUp(self):
+        super(UsageInfoTestCase, self).setUp()
+        self.flags(connection_type='fake',
+                   stub_network=True,
+                   host='fake')
+        self.stubs.Set(nova.flags.FLAGS, 'notification_driver',
+                'nova.notifier.test_notifier')
+        self.volume = importutils.import_object(FLAGS.volume_manager)
+        self.user_id = 'fake'
+        self.project_id = 'fake'
+        self.snapshot_id = 'fake'
+        self.volume_size = 0
+        self.context = context.RequestContext(self.user_id, self.project_id)
+        test_notifier.NOTIFICATIONS = []
+
+    def _create_volume(self, params={}):
+        """Create a test volume"""
+        vol = {}
+        vol['snapshot_id'] = self.snapshot_id
+        vol['user_id'] = self.user_id
+        vol['project_id'] = self.project_id
+        vol['host'] = FLAGS.host
+        vol['availability_zone'] = FLAGS.storage_availability_zone
+        vol['status'] = "creating"
+        vol['attach_status'] = "detached"
+        vol['size'] = self.volume_size
+        vol.update(params)
+        return db.volume_create(self.context, vol)['id']
+
+    def test_notify_usage_exists(self):
+        """Ensure 'exists' notification generates appropriate usage data."""
+        volume_id = self._create_volume()
+        volume = db.volume_get(self.context, volume_id)
+        volume_utils.notify_usage_exists(self.context, volume)
+        self.assertEquals(len(test_notifier.NOTIFICATIONS), 1)
+        msg = test_notifier.NOTIFICATIONS[0]
+        self.assertEquals(msg['priority'], 'INFO')
+        self.assertEquals(msg['event_type'], 'volume.exists')
+        payload = msg['payload']
+        self.assertEquals(payload['tenant_id'], self.project_id)
+        self.assertEquals(payload['user_id'], self.user_id)
+        self.assertEquals(payload['snapshot_id'], self.snapshot_id)
+        self.assertEquals(payload['volume_id'], volume.id)
+        self.assertEquals(payload['size'], self.volume_size)
+        for attr in ('display_name', 'created_at', 'launched_at',
+                     'status', 'audit_period_beginning',
+                     'audit_period_ending'):
+            self.assertTrue(attr in payload,
+                            msg="Key %s not in payload" % attr)
+        db.volume_destroy(context.get_admin_context(), volume['id'])