def test_case_preserving_username(app):
"""Test that username preserves the case."""
with app.app_context():
with db.session.begin_nested():
u1 = User(email='*****@*****.**')
db.session.add(u1)
db.session.add(UserProfile(user=u1, username="******"))
db.session.commit()
profile = UserProfile.get_by_username('info')
assert profile.username == 'InFo'
def test_case_insensitive_username(app):
"""Test case-insensitive uniqueness."""
with app.app_context():
with db.session.begin_nested():
u1 = User(email='*****@*****.**')
u2 = User(email='*****@*****.**')
db.session.add(u1, u2)
profile1 = UserProfile(user=u1, username="******")
profile2 = UserProfile(user=u2, username="******")
db.session.add(profile1)
db.session.add(profile2)
pytest.raises(IntegrityError, db.session.commit)
def test_delete_cascade(app):
"""Test that deletion of user, also removes profile."""
with app.app_context():
with db.session.begin_nested():
u = User(email='*****@*****.**')
db.session.add(u)
p = UserProfile(user=u, username="******")
db.session.add(p)
db.session.commit()
assert UserProfile.get_by_userid(u.id) is not None
db.session.delete(u)
db.session.commit()
assert UserProfile.get_by_userid(u.id) is None
def get_user_info_by_username(username):
"""Get user information by username.
Query database to get user id by using username
Get email from database using user id
Pack response data: user id, user name, email
parameter:
username: The username
return: response pack
"""
result = dict()
try:
user = UserProfile.get_by_username(username)
user_id = user.user_id
metadata = MetaData()
metadata.reflect(bind=db.engine)
table_name = 'accounts_user'
user_table = Table(table_name, metadata)
record = db.session.query(user_table)
data = record.all()
for item in data:
if item[0] == user_id:
result['username'] = username
result['user_id'] = user_id
result['email'] = item[1]
return result
return None
except Exception as e:
result['error'] = str(e)
def get_user_information(user_id):
"""
Get user information user_id.
Query database to get email by using user_id
Get username from database using user id
Pack response data: user id, user name, email
parameter:
user_id: The user_id
return: response
"""
result = {'username': '', 'email': ''}
user_info = UserProfile.get_by_userid(user_id)
if user_info is not None:
result['username'] = user_info.get_username
metadata = MetaData()
metadata.reflect(bind=db.engine)
table_name = 'accounts_user'
user_table = Table(table_name, metadata)
record = db.session.query(user_table)
data = record.all()
for item in data:
if item[0] == user_id:
result['email'] = item[1]
return result
return result
def get_user_info_by_email(email):
"""
Get user information by email.
Query database to get user id by using email
Get username from database using user id
Pack response data: user id, user name, email
parameter:
email: The email
return: response
"""
result = dict()
try:
metadata = MetaData()
metadata.reflect(bind=db.engine)
table_name = 'accounts_user'
user_table = Table(table_name, metadata)
record = db.session.query(user_table)
data = record.all()
for item in data:
if item[1] == email:
user = UserProfile.get_by_userid(item[0])
if user is None:
result['username'] = ""
else:
result['username'] = user.get_username
result['user_id'] = item[0]
result['email'] = email
return result
return None
except Exception as e:
result['error'] = str(e)
def parse_ranking_results(results,
display_rank,
list_name='all',
title_key='title',
count_key=None,
pid_key=None,
search_key=None,
date_key=None):
"""Parse the raw stats results to be usable by the view."""
ranking_list = []
if pid_key:
url = '../records/{0}'
key = pid_key
elif search_key:
url = '../search?page=1&size=20&search_type=1&q={0}'
key = search_key
else:
url = None
if results and list_name in results:
rank = 1
count = 0
date = ''
for item in results[list_name]:
t = {}
if count_key:
if not count == int(item[count_key]):
rank = len(ranking_list) + 1
count = int(item[count_key])
t['rank'] = rank
t['count'] = count
elif date_key:
new_date = \
datetime.utcfromtimestamp(
item[date_key]).strftime('%Y-%m-%d')
if new_date == date:
t['date'] = ''
else:
t['date'] = new_date
date = new_date
title = item[title_key]
if title_key == 'user_id':
user_info = UserProfile.get_by_userid(title)
if user_info:
title = user_info.username
else:
title = 'None'
t['title'] = title
t['url'] = url.format(item[key]) if url and key in item else None
ranking_list.append(t)
if len(ranking_list) == display_rank:
break
return ranking_list
def validate_user(username, email):
"""Validate user information.
Get user id from database using username
Get user id from database using email
Compare 2 user id to validate user information
Pack responde data:
results: user information (username, user id, email)
validation: username is match with email or not
error: null if no error occurs
param:
username: The username
email: The email
return: response data
"""
result = {'results': '', 'validation': False, 'error': ''}
try:
user = UserProfile.get_by_username(username)
user_id = 0
metadata = MetaData()
metadata.reflect(bind=db.engine)
table_name = 'accounts_user'
user_table = Table(table_name, metadata)
record = db.session.query(user_table)
data = record.all()
for item in data:
if item[1] == email:
user_id = item[0]
break
if user.user_id == user_id:
user_info = dict()
user_info['username'] = username
user_info['user_id'] = user_id
user_info['email'] = email
result['results'] = user_info
result['validation'] = True
return result
except Exception as e:
result['error'] = str(e)
return result
def test_userprofiles(app):
"""Test UserProfile model."""
profile = UserProfile()
# Check the username validator works on the model
profile.username = test_usernames['valid']
with pytest.raises(ValueError):
profile.username = test_usernames['invalid_characters']
with pytest.raises(ValueError):
profile.username = test_usernames['invalid_begins_with_number']
# Test non-validated attributes
profile.first_name = 'Test'
profile.last_name = 'User'
assert profile.first_name == 'Test'
assert profile.last_name == 'User'
def test_profile_updating(base_app):
base_app.config.update(USERPROFILES_EXTEND_SECURITY_FORMS=True)
WekoUserProfiles(base_app)
app = base_app
with app.app_context():
user = User(email='lollorosso', password='******')
db.session.add(user)
db.session.commit()
assert user.profile is None
profile = UserProfile(
username='******',
full_name='Test T. User'
)
user.profile = profile
user.profile.username = '******'
assert user.profile.username == 'Different_Name'
assert profile.username == 'Different_Name'
def get_list_username():
"""Get list username.
Query database to get all available username
return: list of username
"""
current_user_id = current_user.get_id()
user_index = 1
result = list()
while True:
try:
if (int(current_user_id) == user_index):
pass
else:
user_info = UserProfile.get_by_userid(user_index)
result.append(user_info.get_username)
user_index = user_index + 1
except Exception:
break
return result