import json
from flask_testing import TestCase

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


class BaseTestConfig(TestCase):

    def create_app(self):
        app = create_app(['grant.settings', 'tests.settings'])
        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()

    def assertStatus(self, response, status_code, message=None):
        """
        Overrides TestCase's default to print out response JSON.
        """

        message = message or 'HTTP Status %s expected but got %s. Response json: %s' \
                             % (status_code, response.status_code, response.json or response.data)
        self.assertEqual(response.status_code, status_code, message)

    assert_status = assertStatus


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

    def setUp(self):
        super(BaseUserConfig, self).setUp()
        self.user = User.create(
            email_address=test_user["emailAddress"],
            password=test_user["password"],
            display_name=test_user["displayName"],
            title=test_user["title"],
        )
        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)

        self.user_password = test_user["password"]

        self.other_user = User.create(
            email_address=test_other_user["emailAddress"],
            password=test_other_user["password"],
            display_name=test_other_user["displayName"],
            title=test_other_user["title"]
        )

        db.session.commit()

    def login_default_user(self):
        self.app.post(
            "/api/v1/users/auth",
            data=json.dumps({
                "email": self.user.email_address,
                "password": self.user_password
            }),
            content_type="application/json"
        )

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


class BaseProposalCreatorConfig(BaseUserConfig):
    def setUp(self):
        super().setUp()
        self.proposal = Proposal.create(
            status="DRAFT",
            title=test_proposal["title"],
            content=test_proposal["content"],
            brief=test_proposal["brief"],
            category=test_proposal["category"],
            target=test_proposal["target"],
            payout_address=test_proposal["payoutAddress"],
            trustees=test_proposal["trustees"][0],
            deadline_duration=test_proposal["deadlineDuration"],
            vote_duration=test_proposal["voteDuration"]
        )
        self.proposal.team.append(self.user)
        db.session.add(self.proposal)

        self.other_proposal = Proposal.create(status="DRAFT")
        self.other_proposal.team.append(self.other_user)
        db.session.add(self.other_proposal)
        db.session.commit()