zcash-grant-system/backend/tests/config.py

from flask_testing import TestCase

from grant.app import create_app
from grant.user.models import User, SocialMedia, db, Avatar
from .test_data import test_user, message


class BaseTestConfig(TestCase):

    def create_app(self):
        app = create_app()
        app.config.from_object('tests.settings')
        return app

    def setUp(self):
        db.drop_all()
        self.app = self.create_app().test_client()
        db.create_all()

    def tearDown(self):
        db.session.remove()
        db.drop_all()


class BaseUserConfig(BaseTestConfig):
    headers = {
        "MsgSignature": message["sig"],
        "RawTypedData": message["data"]
    }

    def setUp(self):
        super(BaseUserConfig, self).setUp()
        self.user = User.create(
            account_address=test_user["accountAddress"],
            email_address=test_user["emailAddress"],
            display_name=test_user["displayName"],
            title=test_user["title"],
            _send_email=False
        )
        sm = SocialMedia(social_media_link=test_user['socialMedias'][0]['link'], user_id=self.user.id)
        db.session.add(sm)
        avatar = Avatar(image_url=test_user["avatar"]["link"], user_id=self.user.id)
        db.session.add(avatar)
        db.session.commit()

    def remove_default_user(self):
        User.query.filter_by(id=self.user.id).delete()
        db.session.commit()