def setup(self, session):
"""
SETUP funcion for TestUser class, this function is executed for each function in this TestClass
"""
user_types = []
user_types.append(UserType("ADMIN", 2))
user_types.append(UserType("GUEST", 0))
user_types.append(UserType("USER", 1))
user_type_db = UserType.query.all()
for user in user_type_db:
for seedUser in user_types:
if (user.name == seedUser.name):
user_types.remove(seedUser)
for user_type in user_types:
self.user_type_list.append(user_type)
session.add(user_type)
session.commit()
user = User("Admin", "*****@*****.**", "SuperSecret", None, None,
True)
user.user_type_id = UserType.get_type_by_name('ADMIN').id
user.activated = True
session.add(user)
session.commit()
ttoken = str(user.id).zfill(10) + str(uuid.uuid4())
#Check if token is unique
while (UserToken.query.filter_by(token=ttoken).count() != 0):
ttoken = str(user.id).zfill(10) + str(uuid.uuid4())
ttoken = UserToken(user_id=user.id, token=ttoken)
ttoken.update = datetime.utcnow()
ttoken.created = datetime.utcnow()
session.add(ttoken)
session.commit()
self.token_list.append(ttoken.token)
self.user_list.append(user.id)
user = User("Test", "*****@*****.**", "SuperSecret", None, None, True)
session.add(user)
session.commit()
ttoken = str(user.id).zfill(10) + str(uuid.uuid4())
#Check if token is unique
while (UserToken.query.filter_by(token=ttoken).count() != 0):
ttoken = str(user.id).zfill(10) + str(uuid.uuid4())
ttoken = UserToken(user_id=user.id, token=ttoken)
ttoken.update = datetime.utcnow()
ttoken.created = datetime.utcnow()
session.add(ttoken)
session.commit()
self.token_list.append(ttoken.token)
self.user_list.append(user.id)
yield
session.query(User).delete()
session.query(UserType).delete()
session.query(UserToken).delete()
session.commit()
def setup(self, session):
"""
SETUP funcion for TestUser class, this function is executed for each function in this TestClass
"""
user_types = []
user_types.append(UserType("ADMIN", 2))
user_types.append(UserType("GUEST", 0))
user_types.append(UserType("USER", 1))
user_type_db = UserType.query.all()
for user in user_type_db:
for seedUser in user_types:
if (user.name == seedUser.name):
user_types.remove(seedUser)
for user_type in user_types:
session.add(user_type)
session.commit()
user = User("Admin", "*****@*****.**", "SuperSecret", None, None,
True)
user.user_type_id = UserType.get_type_by_name('ADMIN').id
user.activated = True
session.add(user)
session.commit()
self.userList.append(user)
yield
session.query(User).delete()
session.query(UserType).delete()
session.commit()
self.userList[:] = []
def test_get_user_all(self, session):
"""Test method get_user_all in User"""
users = User.get_user_all()
assert len(users) == len(self.userList)
assert users[0].id == self.userList[0].id
assert users[0].name == self.userList[0].name
assert users[0].email == self.userList[0].email
assert users[0].password == self.userList[0].password
user = User("DUMMY", "*****@*****.**", "hackPr00f", None, None,
True)
user.user_type_id = UserType.get_type_by_name('USER').id
user.activated = True
session.add(user)
session.commit()
self.userList.append(user)
users = User.get_user_all()
assert len(users) == len(self.userList)
session.query(User).delete()
session.commit()
users = User.get_user_all(False)
assert users == []
#Test something that really shouldn't be there
with pytest.raises(DbException) as err:
User.get_user_all()
assert str(err.value) == "(404, 'No user found.')"
def __userInsert():
userList = []
userAdmin = User("Admin", "*****@*****.**", "SuperSecret", None, None,
True)
userAdmin.user_type_id = UserType.get_type_by_name('ADMIN').id
userAdmin.activated = True
userList.append(userAdmin)
userEntities = User.query.all()
for user in userEntities:
for seedUser in userList:
if (user.name == seedUser.name):
userList.remove(seedUser)
for seedUser in userList:
db.session.add(seedUser)
db.session.commit()
print("Inserting: ")
print(seedUser.name)
def test_update_user_by_id_from_array(self, session):
"""Test method update_user_by_id_from_array in User"""
array = {
'name': 'kektimusMaximusPrime',
'email': '*****@*****.**',
'password': '******',
'user_type_id': 2
}
#Test something that really shouldn't be there
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(0, array)
assert str(err.value) == "(404, 'User not found.')"
user = User("DUMMY", "*****@*****.**", "hackPr00f", None, None,
True)
user.user_type_id = UserType.get_type_by_name('GUEST').id
user.activated = True
session.add(user)
session.commit()
self.userList.append(user)
assert len(self.userList) == 2
User.update_user_by_id_from_array(self.userList[1].id, array)
user = User.get_user_by_id(self.userList[1].id)
assert user.id == self.userList[1].id
assert user.name == 'kektimusMaximusPrime'
assert user.email == '*****@*****.**'
assert user.confirm_password('passwd')
assert user.user_type_id == 2
array = {}
User.update_user_by_id_from_array(self.userList[1].id, array)
user = User.get_user_by_id(self.userList[1].id)
assert user.id == self.userList[1].id
assert user.name == 'kektimusMaximusPrime'
assert user.email == '*****@*****.**'
assert user.confirm_password('passwd')
assert user.user_type_id == 2
array = {'kek': 123, 'token': '01235'}
User.update_user_by_id_from_array(self.userList[1].id, array)
user = User.get_user_by_id(self.userList[1].id)
assert user.id == self.userList[1].id
assert user.name == 'kektimusMaximusPrime'
assert user.email == '*****@*****.**'
assert user.confirm_password('passwd')
assert user.user_type_id == 2
array = {'name': 'kektimusMaximusPrime'}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'username')"
array = {'email': '*****@*****.**'}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'email')"
array = {'user_type_id': 200}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'usertype')"
array = {'name': 'kektimusMaximusPrime', 'password': '******'}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'username')"
user = User.get_user_by_id(self.userList[0].id)
assert not user.confirm_password('passwd')
array = {'email': '*****@*****.**', 'password': '******'}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'email')"
user = User.get_user_by_id(self.userList[0].id)
assert not user.confirm_password('passwd')
array = {'user_type_id': 200, 'password': '******'}
with pytest.raises(DbException) as err:
User.update_user_by_id_from_array(self.userList[0].id, array)
assert str(err.value) == "(400, 'usertype')"
user = User.get_user_by_id(self.userList[0].id)
assert not user.confirm_password('passwd')
def setup(self, session):
"""
SETUP funcion for TestUser class, this function is executed for each function in this TestClass
"""
user_types = []
user_types.append(UserType("ADMIN", 2))
user_types.append(UserType("GUEST", 0))
user_types.append(UserType("USER", 1))
user_type_db = UserType.query.all()
for user in user_type_db:
for seedUser in user_types:
if (user.name == seedUser.name):
user_types.remove(seedUser)
for user_type in user_types:
session.add(user_type)
session.commit()
user = User("GUEST", "*****@*****.**", "GUEST", None, None, True)
user.user_type_id = UserType.get_type_by_name('GUEST').id
user.activated = True
user1 = User("USER", "*****@*****.**", "USER", None, None, True)
user1.user_type_id = UserType.get_type_by_name('USER').id
user1.activated = True
user2 = User("Admin", "*****@*****.**", "SuperSecret", None, None,
True)
user2.user_type_id = UserType.get_type_by_name('ADMIN').id
user2.activated = True
session.add(user)
session.add(user1)
session.add(user2)
session.commit()
self.userList.append(user)
self.userList.append(user1)
self.userList.append(user2)
#make initial project
session.add(Project.create_project('DummyProject_01'))
session.add(Project.create_project('DummyProject_02'))
session.commit()
self.projectList = Project.query.all()
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'M11'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'M12'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'M13'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'read'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'write'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'share'))
session.add(
Module(Project.get_project_by_name('DummyProject_01').id, 'own'))
session.commit()
self.moduleList = Module.query.all()
perm0 = ModulePerm(self.userList[0].id, self.moduleList[0].id, 15)
perm1 = ModulePerm(self.userList[0].id, self.moduleList[1].id, 8)
perm2 = ModulePerm(self.userList[0].id, self.moduleList[2].id, 0)
session.add(perm0)
session.add(perm1)
session.add(perm2)
session.commit()
self.permList.append(perm0)
self.permList.append(perm1)
self.permList.append(perm2)
yield
session.query(ModulePerm).delete()
self.permList[:] = []
session.query(Module).delete()
session.query(Project).delete()
session.commit()
self.moduleList[:] = []
self.projectList[:] = []
session.query(User).delete()
session.query(UserType).delete()
session.commit()
self.userList[:] = []
