import unittest
import flaskbase
import contextlib

from flaskbase.extensions import db


class TestCase(unittest.TestCase):
    """
    Base testcase class
    """

    def setUp(self):
        self.flask_app = flaskbase.create_app(settings={
            'SQLALCHEMY_DATABASE_URI': 'sqlite://',
            })

        self.app = self.flask_app.test_client()
        flaskbase.init_db(self.flask_app)

    def tearDown(self):
        db.session.remove()
        with contextlib.closing(db.engine.connect()) as con:
            trans = con.begin()
            for table in reversed(db.Model.metadata.sorted_tables):
                con.execute(table.delete())
            trans.commit()