iframe-proxy | Sunbelt Computer Software

Repository navigation

from flask_testing import TestCase

from app.database.models.user import UserModel

from app.database.sqlalchemy_extension import db

from run import application

from tests.test_data import test_admin_user

class BaseTestCase(TestCase):

@classmethod

def create_app(cls):

application.config.from_object("config.TestingConfig")

# Setting up test environment variables

application.config["SECRET_KEY"] = "TEST_SECRET_KEY"

application.config["SECURITY_PASSWORD_SALT"] = "TEST_SECURITY_PWD_SALT"

return application

def setUp(self):

db.create_all()

self.admin_user = UserModel(

name=test_admin_user["name"],

email=test_admin_user["email"],

username=test_admin_user["username"],

password=test_admin_user["password"],

terms_and_conditions_checked=test_admin_user[

"terms_and_conditions_checked"

)

db.session.add(self.admin_user)

db.session.commit()

self.admin_user.is_email_verified = True

@classmethod

def tearDown(cls):

db.session.remove()

db.drop_all()