mirror of https://gitlab.com/SIGBUS/nyaa.git
104 lines
4.2 KiB
Python
104 lines
4.2 KiB
Python
import unittest
|
|
from hashlib import md5
|
|
from urllib.parse import urlencode
|
|
|
|
import flask
|
|
|
|
from tests import NyaaTestCase
|
|
from nyaa import models
|
|
|
|
|
|
class TestUserModel(NyaaTestCase):
|
|
def setUp(self):
|
|
self.user = models.User(username='test_user', email='user@test.com', password='passwd_hash')
|
|
|
|
def test_basic(self):
|
|
""" Test basic attributes """
|
|
self.assertEqual(repr(self.user), '<User \'test_user\'>')
|
|
self.assertEqual(self.user.username, 'test_user')
|
|
self.assertEqual(self.user.email, 'user@test.com')
|
|
# expected failure
|
|
# self.assertEqual(self.user.password_hash, 'passwd_hash')
|
|
self.assertEqual(self.user.status, models.UserStatusType.INACTIVE)
|
|
self.assertEqual(self.user.level, models.UserLevelType.REGULAR)
|
|
|
|
def test_userlevel_str(self):
|
|
""" Test user level strings and colors """
|
|
self.user.level = models.UserLevelType.REGULAR
|
|
self.assertEqual(self.user.userlevel_str, 'User')
|
|
self.assertEqual(self.user.userlevel_color, 'default')
|
|
|
|
self.user.level = models.UserLevelType.TRUSTED
|
|
self.assertEqual(self.user.userlevel_str, 'Trusted')
|
|
self.assertEqual(self.user.userlevel_color, 'success')
|
|
|
|
self.user.level = models.UserLevelType.MODERATOR
|
|
self.assertEqual(self.user.userlevel_str, 'Moderator')
|
|
self.assertEqual(self.user.userlevel_color, 'purple')
|
|
|
|
# The color for Superadmins is the same as Moderators
|
|
self.user.level = models.UserLevelType.SUPERADMIN
|
|
self.assertEqual(self.user.userlevel_str, 'Administrator')
|
|
self.assertEqual(self.user.userlevel_color, 'purple')
|
|
|
|
def test_userstatus_str(self):
|
|
""" Test user status strings """
|
|
self.user.status = models.UserStatusType.INACTIVE
|
|
self.assertEqual(self.user.userstatus_str, 'Inactive')
|
|
self.user.status = models.UserStatusType.ACTIVE
|
|
self.assertEqual(self.user.userstatus_str, 'Active')
|
|
self.user.status = models.UserStatusType.BANNED
|
|
self.assertEqual(self.user.userstatus_str, 'Banned')
|
|
|
|
def test_is_x_properties(self):
|
|
""" Test user.is_x properties """
|
|
self.user.status = models.UserStatusType.BANNED
|
|
self.assertTrue(self.user.is_banned)
|
|
self.assertFalse(self.user.is_trusted)
|
|
self.assertFalse(self.user.is_moderator)
|
|
self.assertFalse(self.user.is_superadmin)
|
|
self.user.status = models.UserStatusType.INACTIVE
|
|
|
|
self.user.level = models.UserLevelType.REGULAR
|
|
self.assertFalse(self.user.is_banned)
|
|
self.assertFalse(self.user.is_trusted)
|
|
self.assertFalse(self.user.is_moderator)
|
|
self.assertFalse(self.user.is_superadmin)
|
|
|
|
self.user.level = models.UserLevelType.TRUSTED
|
|
self.assertFalse(self.user.is_banned)
|
|
self.assertTrue(self.user.is_trusted)
|
|
self.assertFalse(self.user.is_moderator)
|
|
self.assertFalse(self.user.is_superadmin)
|
|
|
|
# Moderators and above are also trusted
|
|
self.user.level = models.UserLevelType.MODERATOR
|
|
self.assertFalse(self.user.is_banned)
|
|
self.assertTrue(self.user.is_trusted)
|
|
self.assertTrue(self.user.is_moderator)
|
|
self.assertFalse(self.user.is_superadmin)
|
|
|
|
# Superadmins also show up as Moderators
|
|
self.user.level = models.UserLevelType.SUPERADMIN
|
|
self.assertFalse(self.user.is_banned)
|
|
self.assertTrue(self.user.is_trusted)
|
|
self.assertTrue(self.user.is_moderator)
|
|
self.assertTrue(self.user.is_superadmin)
|
|
|
|
def test_gravatar_url(self):
|
|
""" Test user Gravatar URL """
|
|
with self.request_context():
|
|
expected_gravatar_url = 'https://www.gravatar.com/avatar/{email_hash}?{params}'.format(
|
|
email_hash=md5(self.user.email.encode('utf-8').lower()).hexdigest(),
|
|
params=urlencode({
|
|
's': 120,
|
|
'd': flask.url_for('static', filename='img/avatar/default.png', _external=True),
|
|
'r': 'pg' if self.app.config['SITE_FLAVOR'] == 'nyaa' else 'x',
|
|
})
|
|
)
|
|
self.assertEqual(self.user.gravatar_url(), expected_gravatar_url)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|