Adds the 'limit' parameter to sample list command in V2 API

Change-Id: I338590fcd75f39c3419e7e138023f6918f206ae2
Fixes: bug #1244172

2 files changed, 17 insertions, 1 deletions

diff --git a/ceilometerclient/tests/v2/test_samples.py b/ceilometerclient/tests/v2/test_samples.py
index a6eff4f..c279f35 100644
--- a/ceilometerclient/tests/v2/test_samples.py
+++ b/ceilometerclient/tests/v2/test_samples.py
@@ -36,6 +36,7 @@ del CREATE_SAMPLE['source']
 
 base_url = '/v2/meters/instance'
 args = 'q.field=resource_id&q.field=source&q.op=&q.op=&q.value=foo&q.value=bar'
+args_limit = 'limit=1'
 fixtures = {
     base_url:
     {
@@ -54,6 +55,13 @@ fixtures = {
             {},
             [],
         ),
+    },
+    '%s?%s' % (base_url, args_limit):
+    {
+        'GET': (
+            {},
+            [GET_SAMPLE]
+        ),
     }
 }
 
@@ -93,3 +101,9 @@ class SampleManagerTest(utils.BaseTestCase):
         ]
         self.assertEqual(self.api.calls, expect)
         self.assertTrue(sample)
+
+    def test_limit(self):
+        samples = list(self.mgr.list(meter_name='instance', limit=1))
+        expect = [('GET', '/v2/meters/instance?limit=1', {}, None)]
+        self.assertEqual(self.api.calls, expect)
+        self.assertEqual(len(samples), 1)
diff --git a/ceilometerclient/tests/v2/test_shell.py b/ceilometerclient/tests/v2/test_shell.py
index 1c5a0dc..0b471a3 100644
--- a/ceilometerclient/tests/v2/test_shell.py
+++ b/ceilometerclient/tests/v2/test_shell.py
@@ -173,6 +173,7 @@ class ShellSampleListCommandTest(utils.BaseTestCase):
         self.args = mock.Mock()
         self.args.meter = self.METER
         self.args.query = None
+        self.args.limit = None
 
     def test_sample_list(self):
 
@@ -186,7 +187,8 @@ class ShellSampleListCommandTest(utils.BaseTestCase):
             ceilometer_shell.do_sample_list(self.cc, self.args)
             self.cc.samples.list.assert_called_once_with(
                 meter_name=self.METER,
-                q=None)
+                q=None,
+                limit=None)
         finally:
             sys.stdout = org_stdout