ogn-python/tests/base.py

23 wiersze
508 B
Python

import unittest
from app import create_app, db
class TestBaseDB(unittest.TestCase):
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.session.execute("CREATE EXTENSION IF NOT EXISTS postgis;")
db.session.commit()
db.create_all()
def tearDown(self):
db.session.remove()
db.drop_all()
self.app_context.pop()
if __name__ == "__main__":
unittest.main()