172 lines
6.1 KiB
Python
172 lines
6.1 KiB
Python
import copy
|
|
import json
|
|
|
|
from animal_case import animalify
|
|
from grant.proposal.models import Proposal
|
|
from grant.user.models import User, user_schema, db
|
|
from mock import patch
|
|
|
|
from ..config import BaseUserConfig
|
|
from ..test_data import test_team, test_proposal, test_user
|
|
|
|
|
|
class TestAPI(BaseUserConfig):
|
|
@patch('grant.email.send.send_email')
|
|
def test_create_user(self, mock_send_email):
|
|
mock_send_email.return_value.ok = True
|
|
# Delete the user config user
|
|
db.session.delete(self.user)
|
|
db.session.commit()
|
|
|
|
response = self.app.post(
|
|
"/api/v1/users/",
|
|
data=json.dumps(test_user),
|
|
content_type='application/json'
|
|
)
|
|
self.assertStatus(response, 201)
|
|
|
|
# User
|
|
user_db = User.get_by_email(test_user["emailAddress"])
|
|
self.assertEqual(user_db.display_name, test_user["displayName"])
|
|
self.assertEqual(user_db.title, test_user["title"])
|
|
self.assertEqual(user_db.email_address, test_user["emailAddress"])
|
|
|
|
def test_get_all_users(self):
|
|
users_get_resp = self.app.get(
|
|
"/api/v1/users/"
|
|
)
|
|
self.assert200(users_get_resp)
|
|
users_json = users_get_resp.json
|
|
self.assertEqual(users_json[0]["displayName"], self.user.display_name)
|
|
|
|
def test_get_single_user_by_id(self):
|
|
users_get_resp = self.app.get(
|
|
"/api/v1/users/{}".format(self.user.id)
|
|
)
|
|
|
|
users_json = users_get_resp.json
|
|
self.assertEqual(users_json["avatar"]["imageUrl"], self.user.avatar.image_url)
|
|
self.assertEqual(users_json["socialMedias"][0]["service"], 'GITHUB')
|
|
self.assertEqual(users_json["socialMedias"][0]["username"], 'groot')
|
|
self.assertEqual(users_json["socialMedias"][0]["url"], self.user.social_medias[0].social_media_link)
|
|
self.assertEqual(users_json["displayName"], self.user.display_name)
|
|
|
|
def test_user_auth_success(self):
|
|
user_auth_resp = self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": self.user_password
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
print(user_auth_resp.headers)
|
|
self.assertEqual(user_auth_resp.json['emailAddress'], self.user.email_address)
|
|
self.assertEqual(user_auth_resp.json['displayName'], self.user.display_name)
|
|
|
|
def test_user_auth_required(self):
|
|
login_resp = self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": self.user_password
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
print(login_resp.headers)
|
|
# should have session cookie now
|
|
me_resp = self.app.get(
|
|
"/api/v1/users/me",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": self.user_password
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
print(me_resp.headers)
|
|
self.assert200(me_resp)
|
|
|
|
def test_user_auth_required_fail(self):
|
|
me_resp = self.app.get(
|
|
"/api/v1/users/me",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": self.user_password
|
|
}),
|
|
|
|
content_type="application/json"
|
|
)
|
|
print(me_resp.json)
|
|
print(me_resp.headers)
|
|
self.assert401(me_resp)
|
|
|
|
def test_user_auth_bad_password(self):
|
|
user_auth_resp = self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": self.user.email_address,
|
|
"password": "badpassword"
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
self.assert403(user_auth_resp)
|
|
self.assertTrue(user_auth_resp.json['message'] is not None)
|
|
|
|
def test_user_auth_bad_email(self):
|
|
user_auth_resp = self.app.post(
|
|
"/api/v1/users/auth",
|
|
data=json.dumps({
|
|
"email": "bademail@bad.com",
|
|
"password": "somepassword"
|
|
}),
|
|
content_type="application/json"
|
|
)
|
|
self.assert400(user_auth_resp)
|
|
self.assertTrue(user_auth_resp.json['message'] is not None)
|
|
|
|
def test_create_user_duplicate_400(self):
|
|
# self.user is identical to test_user, should throw
|
|
response = self.app.post(
|
|
"/api/v1/users/",
|
|
data=json.dumps(test_user),
|
|
content_type='application/json'
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 409)
|
|
|
|
@patch('grant.user.views.remove_avatar')
|
|
def test_update_user_remove_social_and_avatar(self, mock_remove_avatar):
|
|
self.login_default_user()
|
|
updated_user = animalify(copy.deepcopy(user_schema.dump(self.user)))
|
|
updated_user["displayName"] = 'new display name'
|
|
updated_user["avatar"] = {}
|
|
updated_user["socialMedias"] = []
|
|
|
|
user_update_resp = self.app.put(
|
|
"/api/v1/users/{}".format(self.user.id),
|
|
data=json.dumps(updated_user),
|
|
headers=self.headers,
|
|
content_type='application/json'
|
|
)
|
|
self.assert200(user_update_resp, user_update_resp.json)
|
|
|
|
user_json = user_update_resp.json
|
|
self.assertFalse(user_json["avatar"])
|
|
self.assertFalse(len(user_json["socialMedias"]))
|
|
self.assertEqual(user_json["displayName"], updated_user["displayName"])
|
|
self.assertEqual(user_json["title"], updated_user["title"])
|
|
mock_remove_avatar.assert_called_with(test_user["avatar"]["link"], 1)
|
|
|
|
def test_update_user_400_when_required_param_not_passed(self):
|
|
self.login_default_user()
|
|
updated_user = animalify(copy.deepcopy(user_schema.dump(self.user)))
|
|
updated_user["displayName"] = 'new display name'
|
|
del updated_user["avatar"]
|
|
user_update_resp = self.app.put(
|
|
"/api/v1/users/{}".format(self.user.id),
|
|
data=json.dumps(updated_user),
|
|
headers=self.headers,
|
|
content_type='application/json'
|
|
)
|
|
self.assert400(user_update_resp)
|