diff options
| -rw-r--r-- | budget/tests.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/budget/tests.py b/budget/tests.py index d127b06..3fdc0fa 100644 --- a/budget/tests.py +++ b/budget/tests.py @@ -21,9 +21,11 @@ class TestCase(unittest.TestCase): run.app.config['SQLALCHEMY_DATABASE_URI'] = "sqlite:///memory" run.app.config['CSRF_ENABLED'] = False # simplify the tests self.app = run.app.test_client() - - models.db.init_app(run.app) - run.mail.init_app(run.app) + try: + models.db.init_app(run.app) + run.mail.init_app(run.app) + except: + pass models.db.app = run.app models.db.create_all() |