207 lines
6.7 KiB
Python
207 lines
6.7 KiB
Python
import json
|
|
from datetime import datetime
|
|
from datetime import timedelta
|
|
|
|
from flask_testing import TestCase
|
|
from mock import patch
|
|
|
|
from grant.app import create_app
|
|
from grant.ccr.models import CCR
|
|
from grant.extensions import limiter
|
|
from grant.milestone.models import Milestone
|
|
from grant.proposal.models import Proposal
|
|
from grant.settings import PROPOSAL_STAKING_AMOUNT
|
|
from grant.task.jobs import ProposalReminder
|
|
from grant.user.models import User, SocialMedia, db, Avatar
|
|
from grant.utils.enums import ProposalStatus
|
|
from .test_data import test_user, test_other_user, test_proposal, mock_blockchain_api_requests, test_ccr
|
|
|
|
|
|
class BaseTestConfig(TestCase):
|
|
|
|
def create_app(self):
|
|
app = create_app(['grant.settings', 'tests.settings'])
|
|
app.config.from_object('tests.settings')
|
|
limiter.enabled = False
|
|
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):
|
|
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"],
|
|
)
|
|
self._user.email_verification.has_verified = True
|
|
db.session.add(self._user)
|
|
sm = SocialMedia(
|
|
service=test_user['socialMedias'][0]['service'],
|
|
username=test_user['socialMedias'][0]['username'],
|
|
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"]
|
|
)
|
|
self.other_user_password = test_other_user["password"]
|
|
db.session.commit()
|
|
self._user_id = self._user.id
|
|
self._other_user_id = self._other_user.id
|
|
|
|
# always return fresh (avoid detached instance issues)
|
|
@property
|
|
def user(self):
|
|
return User.query.filter_by(id=self._user_id).first()
|
|
|
|
@property
|
|
def other_user(self):
|
|
return User.query.filter_by(id=self._other_user_id).first()
|
|
|
|
def mark_user_not_verified(self, user=None):
|
|
if not user:
|
|
user = self.user
|
|
user.email_verification.has_verified = False
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
|
|
def login_default_user(self, cust_pass=None):
|
|
return self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": cust_pass or self.user_password
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
|
|
def login_other_user(self):
|
|
return self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": self.other_user.email_address,
|
|
"password": self.other_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=ProposalStatus.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"],
|
|
deadline_duration=test_proposal["deadlineDuration"]
|
|
)
|
|
self._proposal.team.append(self.user)
|
|
db.session.add(self._proposal)
|
|
db.session.flush()
|
|
|
|
milestones = [
|
|
{
|
|
"title": "Milestone 1",
|
|
"content": "Content 1",
|
|
"days_estimated": "30",
|
|
"payout_percent": 50,
|
|
"immediate_payout": True
|
|
},
|
|
{
|
|
"title": "Milestone 2",
|
|
"content": "Content 2",
|
|
"days_estimated": "20",
|
|
"payout_percent": 50,
|
|
"immediate_payout": False
|
|
}
|
|
]
|
|
|
|
Milestone.make(milestones, self._proposal)
|
|
|
|
self._other_proposal = Proposal.create(status=ProposalStatus.DRAFT)
|
|
self._other_proposal.team.append(self.other_user)
|
|
db.session.add(self._other_proposal)
|
|
db.session.commit()
|
|
self._proposal_id = self._proposal.id
|
|
self._other_proposal_id = self._other_proposal.id
|
|
|
|
# always return fresh (avoid detached instance issues)
|
|
@property
|
|
def proposal(self):
|
|
return Proposal.query.filter_by(id=self._proposal_id).first()
|
|
|
|
@property
|
|
def other_proposal(self):
|
|
return Proposal.query.filter_by(id=self._other_proposal_id).first()
|
|
|
|
def make_proposal_reminder_task(self):
|
|
proposal_reminder = ProposalReminder(self.proposal.id)
|
|
proposal_reminder.make_task()
|
|
|
|
def stake_proposal(self):
|
|
# 1. submit
|
|
self.proposal.submit_for_approval()
|
|
# 2. get staking contribution
|
|
contribution = self.proposal.get_staking_contribution(self.user.id)
|
|
# 3. fake a confirmation
|
|
contribution.confirm(tx_id='tx', amount=str(PROPOSAL_STAKING_AMOUNT.normalize()))
|
|
db.session.add(contribution)
|
|
db.session.flush()
|
|
self.proposal.set_pending_when_ready()
|
|
|
|
|
|
class BaseCCRCreatorConfig(BaseUserConfig):
|
|
def setUp(self):
|
|
super().setUp()
|
|
self._ccr = CCR.create(
|
|
status=ProposalStatus.DRAFT,
|
|
title=test_ccr["title"],
|
|
content=test_ccr["content"],
|
|
brief=test_ccr["brief"],
|
|
target=test_ccr["target"],
|
|
user_id=self.user.id
|
|
)
|
|
self._ccr_id = self._ccr.id
|
|
db.session.commit()
|
|
|
|
# always return fresh (avoid detached instance issues)
|
|
@property
|
|
def ccr(self):
|
|
return CCR.query.filter_by(id=self._ccr_id).first()
|