add unittests for testcase and refactor testcase related implementation

[releng.git] / utils / test / result_collection_api / tests / unit / test_base.py

diff --git a/utils/test/result_collection_api/tests/unit/test_base.py b/utils/test/result_collection_api/tests/unit/test_base.py
index 04caa39..57d863c 100644 (file)
--- a/utils/test/result_collection_api/tests/unit/test_base.py
+++ b/utils/test/result_collection_api/tests/unit/test_base.py
@@ -3,7 +3,7 @@ from tornado.web import Application
 from tornado.testing import AsyncHTTPTestCase
 
 from resources.handlers import VersionHandler, PodHandler, \
-    TestProjectHandler, TestCasesHandler, TestResultsHandler, DashboardHandler
+    ProjectHandler, TestcaseHandler, TestResultsHandler, DashboardHandler
 from resources.models import CreateResponse
 import fake_pymongo
 
@@ -28,10 +28,10 @@ class TestBase(AsyncHTTPTestCase):
                 (r"/version", VersionHandler),
                 (r"/pods", PodHandler),
                 (r"/pods/([^/]+)", PodHandler),
-                (r"/projects", TestProjectHandler),
-                (r"/projects/([^/]+)", TestProjectHandler),
-                (r"/projects/([^/]+)/cases", TestCasesHandler),
-                (r"/projects/([^/]+)/cases/([^/]+)", TestCasesHandler),
+                (r"/projects", ProjectHandler),
+                (r"/projects/([^/]+)", ProjectHandler),
+                (r"/projects/([^/]+)/cases", TestcaseHandler),
+                (r"/projects/([^/]+)/cases/([^/]+)", TestcaseHandler),
                 (r"/results", TestResultsHandler),
                 (r"/results([^/]*)", TestResultsHandler),
                 (r"/results/([^/]*)", TestResultsHandler),
@@ -43,52 +43,71 @@ class TestBase(AsyncHTTPTestCase):
             debug=True,
         )
 
-    def create_d(self):
-        return self.create(self.req_d)
+    def create_d(self, *args):
+        return self.create(self.req_d, *args)
 
-    def create_e(self):
-        return self.create(self.req_e)
+    def create_e(self, *args):
+        return self.create(self.req_e, *args)
 
-    def create(self, req=None):
+    def create(self, req=None, *args):
         if req:
             req = req.format()
 
-        res = self.fetch(self.basePath,
+        res = self.fetch(self._get_uri(*args),
                          method='POST',
                          body=json.dumps(req),
                          headers=self.headers)
 
         return self._get_return(res, self.create_res)
 
-    def get(self, item=None):
-        res = self.fetch(self._get_uri(item),
+    def create_help(self, uri, req, cls):
+        res = self.fetch(uri,
+                         method='POST',
+                         body=json.dumps(req.format()),
+                         headers=self.headers)
+
+        return self._get_return(res, cls)
+
+    def get(self, *args):
+        res = self.fetch(self._get_uri(*args),
                          method='GET',
                          headers=self.headers)
 
         def inner():
-            return self.get_res if item else self.list_res
+            new_args, num = self._get_valid_args(*args)
+            return self.get_res if num != self._need_arg_num() else self.list_res
         return self._get_return(res, inner())
 
-    def update(self, item, new=None):
+    def update(self, new=None, *args):
         if new:
             new = new.format()
-        res = self.fetch(self._get_uri(item),
+        res = self.fetch(self._get_uri(*args),
                          method='PUT',
                          body=json.dumps(new),
                          headers=self.headers)
         return self._get_return(res, self.update_res)
 
-    def delete(self, item):
-        res = self.fetch(self._get_uri(item),
+    def delete(self, *args):
+        res = self.fetch(self._get_uri(*args),
                          method='DELETE',
                          headers=self.headers)
-        return res.code
+        return res.code, res.body
 
-    def _get_uri(self, item=None):
+    @staticmethod
+    def _get_valid_args(*args):
+        new_args = tuple(['%s' % arg for arg in args if arg is not None])
+        return new_args, len(new_args)
+
+    def _need_arg_num(self):
+        return self.basePath.count('%s')
+
+    def _get_uri(self, *args):
+        new_args, num = self._get_valid_args(*args)
         uri = self.basePath
-        if item:
-            uri += '/{}'.format(item)
-        return uri
+        if num != self._need_arg_num():
+            uri += '/%s'
+
+        return uri % tuple(['%s' % arg for arg in new_args])
 
     def _get_return(self, res, cls):
         code = res.code
@@ -99,15 +118,15 @@ class TestBase(AsyncHTTPTestCase):
     def _get_return_body(code, body, cls):
         return cls.from_dict(json.loads(body)) if code < 300 else body
 
-    def assert_create_body(self, body, req=None):
-        print(body.href)
+    def assert_create_body(self, body, req=None, *args):
         if not req:
             req = self.req_d
-        self.assertIn(self._get_uri(req.name), body.href)
+        new_args = args + tuple([req.name])
+        self.assertIn(self._get_uri(*new_args), body.href)
 
     @staticmethod
     def _clear():
         fake_pymongo.pods.clear()
         fake_pymongo.projects.clear()
-        fake_pymongo.test_cases.clear()
+        fake_pymongo.testcases.clear()
         fake_pymongo.test_results.clear()