utils/test/result_collection_api/tests/unit/test_base.py

   1 import json
   2 from tornado.web import Application
   3 from tornado.testing import AsyncHTTPTestCase
   4
   5 from resources.handlers import VersionHandler, PodHandler, \
   6     TestProjectHandler, TestCasesHandler, TestResultsHandler, DashboardHandler
   7 import fake_pymongo
   8
   9
  10 class TestBase(AsyncHTTPTestCase):
  11     headers = {'Content-Type': 'application/json; charset=UTF-8'}
  12
  13     def setUp(self):
  14         self.addCleanup(self._clear)
  15         super(TestBase, self).setUp()
  16
  17     def get_app(self):
  18         return Application(
  19             [
  20                 (r"/version", VersionHandler),
  21                 (r"/pods", PodHandler),
  22                 (r"/pods/([^/]+)", PodHandler),
  23                 (r"/test_projects", TestProjectHandler),
  24                 (r"/test_projects/([^/]+)", TestProjectHandler),
  25                 (r"/test_projects/([^/]+)/cases", TestCasesHandler),
  26                 (r"/test_projects/([^/]+)/cases/([^/]+)", TestCasesHandler),
  27                 (r"/results", TestResultsHandler),
  28                 (r"/results([^/]*)", TestResultsHandler),
  29                 (r"/results/([^/]*)", TestResultsHandler),
  30                 (r"/dashboard", DashboardHandler),
  31                 (r"/dashboard([^/]*)", DashboardHandler),
  32                 (r"/dashboard/([^/]*)", DashboardHandler),
  33             ],
  34             db=fake_pymongo,
  35             debug=True,
  36         )
  37
  38     def create(self, uri, body=None):
  39         return self.fetch(uri,
  40                           method='POST',
  41                           body=json.dumps(body),
  42                           headers=self.headers)
  43
  44     def get(self, uri):
  45         return self.fetch(uri,
  46                           method='GET',
  47                           headers=self.headers)
  48
  49     @staticmethod
  50     def _clear():
  51         fake_pymongo.pod.clear()
  52         fake_pymongo.test_projects.clear()
  53         fake_pymongo.test_cases.clear()
  54         fake_pymongo.test_results.clear()