def test_terms(self):
""" Tests acceptance of the terms """
# adding test user to the DB
user_to_add = User(username='******',
password=UserDTO._hash_password('test'),
accepted_terms=False)
user_to_add.save()
# setup test credentials
user_dto = UserDTO(username='******')
user_dto.set_password('test')
# check if login is possible
success, data = self.controller.login(user_dto)
self.assertFalse(success)
self.assertEqual(data, 'terms_not_accepted')
# login with accepted terms fields set
success, data = self.controller.login(user_dto, accept_terms=True)
self.assertTrue(success)
self.assertIsNotNone(data)
# login again to see if fields has been saved
success, data = self.controller.login(user_dto)
self.assertTrue(success)
self.assertIsNotNone(data)
def save(self):
user = User(
username= self.validated_data['username'],
fullname= self.validated_data['fullname'],
email= self.validated_data['email'],
)
password = self.validated_data['password']
password2 = self.validated_data['password2']
if password != password2:
raise serializers.ValidationError({'Password':'******'})
user.set_password(password)
user.save()
return user
def test_remove_user(self):
""" Test removing a user. """
# check that there is only one user in the system
users_in_controller = self.controller.load_users()
self.assertEqual(1, len(users_in_controller))
self.assertEqual('om', users_in_controller[0].username)
self.assertEqual(1, self.controller.get_number_of_users())
# create a new user to test with
user_to_add = User(username='******',
password=UserDTO._hash_password('test'),
accepted_terms=True)
user_to_add.save()
# creating equal credentials to use
user_dto = UserDTO(username='******')
user_dto.set_password('test')
# verify that the user has been added
users_in_controller = self.controller.load_users()
self.assertEqual(2, len(users_in_controller))
self.assertEqual('om', users_in_controller[0].username)
self.assertEqual('test', users_in_controller[1].username)
self.assertEqual(2, self.controller.get_number_of_users())
# verify that the new user can log in to the system
success, token = self.controller.login(user_dto, accept_terms=True)
self.assertTrue(success)
self.assertTrue(self.controller.check_token(token))
# remove the newly created user
self.controller.remove_user(user_dto)
# Verify that the user is logged out of the system
self.assertFalse(self.controller.check_token(token))
# verify that the user is deleted from the system
users_in_controller = self.controller.load_users()
self.assertEqual(1, len(users_in_controller))
self.assertEqual('om', users_in_controller[0].username)
self.assertEqual(1, self.controller.get_number_of_users())
# verify that the last user cannot be deleted.
try:
last_user_dto = UserDTO(username='******')
self.controller.remove_user(last_user_dto)
self.fail('Should have raised exception !')
except Exception as exception:
self.assertEqual(UserEnums.DeleteErrors.LAST_ACCOUNT,
str(exception))
def test_load_users(self):
""" Test getting all usernames. """
# get first list of users in the user controller
users_in_controller = self.controller.load_users()
self.assertEqual(1, len(users_in_controller))
self.assertEqual('om', users_in_controller[0].username)
user_to_add = User(username='******',
password=UserDTO._hash_password('test'),
accepted_terms=True)
user_to_add.save()
# check if the user has been added to the list
users_in_controller = self.controller.load_users()
self.assertEqual(2, len(users_in_controller))
self.assertEqual('om', users_in_controller[0].username)
self.assertEqual('test', users_in_controller[1].username)
# check if the number of users is correct
num_users = self.controller.get_number_of_users()
self.assertEqual(2, num_users)
def _migrate(cls):
# type: () -> None
old_sqlite_db = constants.get_config_database_file()
if os.path.exists(old_sqlite_db):
import sqlite3
connection = sqlite3.connect(old_sqlite_db,
detect_types=sqlite3.PARSE_DECLTYPES,
check_same_thread=False,
isolation_level=None)
cursor = connection.cursor()
for row in cursor.execute(
'SELECT id, username, password, accepted_terms FROM users;'
):
username = row[1]
user = User.get_or_none(username=username)
if user is None:
user = User(username=username,
password=row[2],
accepted_terms=row[3])
user.save()
cursor.execute('DROP TABLE users;')