[Kimchi-devel] [PATCHv8 2/8] Add testcase for GET param

[lvroyce0210 at gmail.com]

From: Royce Lv <lvroyce at linux.vnet.ibm.com>

Add a testcase to test GET param passing and demo how GET param work
with current model implementation,which means:
1. change the API.json
2. wrap its model implementation to accept parameters

Signed-off-by: Royce Lv <lvroyce at linux.vnet.ibm.com>
---
 tests/test_rest.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/tests/test_rest.py b/tests/test_rest.py
index a8e5842..437baca 100644
--- a/tests/test_rest.py
+++ b/tests/test_rest.py
@@ -21,6 +21,7 @@
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 
 import base64
+import cherrypy
 import json
 import os
 import time
@@ -1240,6 +1241,41 @@ class RestTests(unittest.TestCase):
         self.assertIn('net_recv_rate', stats)
         self.assertIn('net_sent_rate', stats)
 
+    def test_get_param(self):
+        def hack_model(func):
+            def _get_param_func(*args, **kwargs):
+                res = func()
+                return res
+            return _get_param_func
+
+        global model
+        cherrypy.request.app.root.api_schema['properties']['vms_get_list'] = {}
+        old_handler = model.vms_get_list
+        model.vms_get_list = hack_model(model.vms_get_list)
+
+        req = json.dumps({'name': 'test', 'cdrom': '/nonexistent.iso'})
+        self.request('/templates', req, 'POST')
+
+        # Create a VM
+        req = json.dumps({'name': 'test-vm1', 'template': '/templates/test'})
+        resp = self.request('/vms', req, 'POST')
+        self.assertEquals(201, resp.status)
+        req = json.dumps({'name': 'test-vm2', 'template': '/templates/test'})
+        resp = self.request('/vms', req, 'POST')
+        self.assertEquals(201, resp.status)
+
+        resp = request(host, port, '/vms')
+        self.assertEquals(200, resp.status)
+        res = json.loads(resp.read())
+        self.assertEquals(2, len(res))
+
+        resp = request(host, port, '/vms?name=test-vm1')
+        self.assertEquals(200, resp.status)
+        res = json.loads(resp.read())
+        self.assertEquals(1, len(res))
+        self.assertEquals('test-vm1', res[0]['name'])
+
+        model.vms_get_list = old_handler
 
 class HttpsRestTests(RestTests):
     """
-- 
1.8.1.2