84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
import json
|
|
from mock import patch
|
|
|
|
from grant.proposal.models import Proposal, ProposalTeamInvite, db
|
|
from ..config import BaseProposalCreatorConfig
|
|
from ..test_data import test_proposal, test_user
|
|
|
|
|
|
class TestAPI(BaseProposalCreatorConfig):
|
|
def test_create_invite_by_email_address(self):
|
|
invite_res = self.app.post(
|
|
"/api/v1/proposals/{}/invite".format(self.proposal.id),
|
|
data=json.dumps({"address": "test@test.test"}),
|
|
headers=self.headers,
|
|
content_type='application/json'
|
|
)
|
|
self.assertStatus(invite_res, 201)
|
|
|
|
# Rejects if not authorized
|
|
def test_no_auth_create_invite_fails(self):
|
|
invite_res = self.app.post(
|
|
"/api/v1/proposals/{}/invite".format(self.proposal.id),
|
|
data=json.dumps({"address": "0x8B0B72F8bDE212991135668922fD5acE557DE6aB"}),
|
|
content_type='application/json'
|
|
)
|
|
self.assertStatus(invite_res, 401)
|
|
|
|
# Rejects if non-existant proposal id
|
|
def test_invalid_proposal_create_invite_fails(self):
|
|
invite_res = self.app.post(
|
|
"/api/v1/proposals/12345/invite",
|
|
data=json.dumps({"address": "0x8B0B72F8bDE212991135668922fD5acE557DE6aB"}),
|
|
headers=self.headers,
|
|
content_type='application/json'
|
|
)
|
|
self.assertStatus(invite_res, 404)
|
|
|
|
def test_delete_invite(self):
|
|
address = "0x8B0B72F8bDE212991135668922fD5acE557DE6aB"
|
|
invite = ProposalTeamInvite(
|
|
proposal_id=self.proposal.id,
|
|
address="0x8B0B72F8bDE212991135668922fD5acE557DE6aB"
|
|
)
|
|
db.session.add(invite)
|
|
db.session.commit()
|
|
|
|
delete_res = self.app.delete(
|
|
"/api/v1/proposals/{}/invite/{}".format(self.proposal.id, address),
|
|
headers=self.headers
|
|
)
|
|
self.assertStatus(delete_res, 202)
|
|
|
|
# Rejects if unknown proposal
|
|
def test_invalid_invite_delete_invite(self):
|
|
delete_res = self.app.delete(
|
|
"/api/v1/proposals/{}/invite/12345".format(self.proposal),
|
|
headers=self.headers
|
|
)
|
|
self.assertStatus(delete_res, 404)
|
|
|
|
# Rejects if not authorized
|
|
def test_no_auth_delete_invite_fails(self):
|
|
delete_res = self.app.delete(
|
|
"/api/v1/proposals/{}/invite/12345".format(self.proposal)
|
|
)
|
|
self.assertStatus(delete_res, 401)
|
|
|
|
# Rejects if the invite was already accepted
|
|
def test_accepted_invite_delete_invite(self):
|
|
address = "0x8B0B72F8bDE212991135668922fD5acE557DE6aB"
|
|
invite = ProposalTeamInvite(
|
|
proposal_id=self.proposal.id,
|
|
address="0x8B0B72F8bDE212991135668922fD5acE557DE6aB",
|
|
accepted=True
|
|
)
|
|
db.session.add(invite)
|
|
db.session.commit()
|
|
|
|
delete_res = self.app.delete(
|
|
"/api/v1/proposals/{}/invite/{}".format(self.proposal.id, address),
|
|
headers=self.headers
|
|
)
|
|
self.assertStatus(delete_res, 403)
|