Advertisement
vsokoltsov

Untitled

Jan 4th, 2018
71
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.66 KB | None | 0 0
  1. import unittest
  2. from app import create_test_app, db
  3. import ipdb
  4.  
  5. class BaseTestCase(unittest.TestCase):
  6.     def __call__(self, result=None):
  7.         try:
  8.             self._pre_setup()
  9.             super(BaseTestCase, self).__call__(result)
  10.         finally:
  11.             self._post_teardown()
  12.  
  13.     def _pre_setup(self):
  14.         self.app = create_test_app()
  15.         self.client = self.app.test_client
  16.         self.app_context = self.app.app_context()
  17.         self.app_context.push()
  18.         ipdb.set_trace()
  19.         db.create_all(app=self.app)
  20.  
  21.     def _post_teardown(self):
  22.         db.session.remove()
  23.         db.drop_all()
  24.         self.app_context.pop()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement