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

   1 from tornado.web import Application
   2 from tornado.testing import AsyncHTTPTestCase
   3
   4 from resources.handlers import VersionHandler, PodHandler, \
   5     TestProjectHandler, TestCasesHandler, TestResultsHandler, DashboardHandler
   6 import fake_pymongo
   7
   8
   9 class TestBase(AsyncHTTPTestCase):
  10     def get_app(self):
  11         return Application(
  12             [
  13                 (r"/version", VersionHandler),
  14                 (r"/pods", PodHandler),
  15                 (r"/pods/([^/]+)", PodHandler),
  16                 (r"/test_projects", TestProjectHandler),
  17                 (r"/test_projects/([^/]+)", TestProjectHandler),
  18                 (r"/test_projects/([^/]+)/cases", TestCasesHandler),
  19                 (r"/test_projects/([^/]+)/cases/([^/]+)", TestCasesHandler),
  20                 (r"/results", TestResultsHandler),
  21                 (r"/results([^/]*)", TestResultsHandler),
  22                 (r"/results/([^/]*)", TestResultsHandler),
  23                 (r"/dashboard", DashboardHandler),
  24                 (r"/dashboard([^/]*)", DashboardHandler),
  25                 (r"/dashboard/([^/]*)", DashboardHandler),
  26             ],
  27             db=fake_pymongo,
  28             debug=True,
  29         )
  30
  31     def tearDown(self):
  32         yield fake_pymongo.pod.remove()
  33         yield fake_pymongo.test_projects.remove()
  34         yield fake_pymongo.test_cases.remove()
  35         yield fake_pymongo.test_results.remove()
  36         super(TestBase, self).tearDown()