Add tests for samples

Change-Id: I8725fedc6f728adaef91ffe9bb80df1c04debf85
2013-01-18 17:48:05 +11:00
parent a8605b57e5
commit 1a76566511
1 changed files with 146 additions and 0 deletions
--- a/tests/v1/test_samples.py
+++ b/tests/v1/test_samples.py
@@ -0,0 +1,146 @@
+# Copyright 2012 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.
+
+import unittest
+
+import ceilometerclient.v1.meters
+from tests import utils
+
+
+fixtures = {
+    '/v1/users/freddy/meters/balls': {
+        'GET': (
+            {},
+            {'events': [
+                {
+                    'resource_id': 'inst-0045',
+                    'project_id': 'melbourne_open',
+                    'user_id': 'freddy',
+                    'name': 'tennis',
+                    'type': 'counter',
+                    'unit': 'balls',
+                    'volume': 3,
+                    'timestamp': None,
+                    'resource_metadata': None,
+                },
+            ]},
+        ),
+    },
+    '/v1/sources/openstack/meters/this': {
+        'GET': (
+            {},
+            {'events': [
+                {
+                    'resource_id': 'b',
+                    'project_id': 'dig_the_ditch',
+                    'user_id': 'joey',
+                    'name': 'this',
+                    'type': 'counter',
+                    'unit': 'b',
+                    'volume': 45,
+                    'timestamp': None,
+                    'resource_metadata': None,
+                },
+            ]},
+        ),
+    },
+    '/v1/projects/dig_the_ditch/meters/meters': {
+        'GET': (
+            {},
+            {'events': [
+                {
+                    'resource_id': 'b',
+                    'project_id': 'dig_the_ditch',
+                    'user_id': 'joey',
+                    'name': 'meters',
+                    'type': 'counter',
+                    'unit': 'meters',
+                    'volume': 345,
+                    'timestamp': None,
+                    'resource_metadata': None,
+                },
+            ]},
+        ),
+    },
+    '/v1/meters?metadata.zxc_id=foo': {
+        'GET': (
+            {},
+            {'events': [
+                {
+                    'resource_id': 'b',
+                    'project_id': 'dig_the_ditch',
+                    'user_id': 'joey',
+                    'name': 'this',
+                    'type': 'counter',
+                    'unit': 'meters',
+                    'volume': 98,
+                    'timestamp': None,
+                    'resource_metadata': {'zxc_id': 'foo'},
+                },
+            ]},
+        ),
+    },
+}
+
+
+class SampleManagerTest(unittest.TestCase):
+
+    def setUp(self):
+        self.api = utils.FakeAPI(fixtures)
+        self.mgr = ceilometerclient.v1.meters.SampleManager(self.api)
+
+    def test_list_by_source(self):
+        samples = list(self.mgr.list(source='openstack',
+                                     counter_name='this'))
+        expect = [
+            ('GET', '/v1/sources/openstack/meters/this', {}, None),
+        ]
+        self.assertEqual(self.api.calls, expect)
+        self.assertEqual(len(samples), 1)
+        self.assertEqual(samples[0].resource_id, 'b')
+
+    def test_list_by_user(self):
+        samples = list(self.mgr.list(user_id='freddy',
+                                     counter_name='balls'))
+        expect = [
+            ('GET', '/v1/users/freddy/meters/balls', {}, None),
+        ]
+        self.assertEqual(self.api.calls, expect)
+        self.assertEqual(len(samples), 1)
+        self.assertEqual(samples[0].project_id, 'melbourne_open')
+        self.assertEqual(samples[0].user_id, 'freddy')
+        self.assertEqual(samples[0].volume, 3)
+
+    def test_list_by_project(self):
+        samples = list(self.mgr.list(project_id='dig_the_ditch',
+                                     counter_name='meters'))
+        expect = [
+            ('GET', '/v1/projects/dig_the_ditch/meters/meters', {}, None),
+        ]
+        self.assertEqual(self.api.calls, expect)
+        self.assertEqual(len(samples), 1)
+        self.assertEqual(samples[0].project_id, 'dig_the_ditch')
+        self.assertEqual(samples[0].volume, 345)
+        self.assertEqual(samples[0].unit, 'meters')
+
+    def test_list_by_metaquery(self):
+        samples = list(self.mgr.list(metaquery='metadata.zxc_id=foo',
+                                     counter_name='this'))
+        expect = [
+            ('GET', '/v1/meters?metadata.zxc_id=foo', {}, None),
+        ]
+        self.assertEqual(self.api.calls, expect)
+        self.assertEqual(len(samples), 1)
+        self.assertEqual(samples[0].resource_metadata['zxc_id'], 'foo')