def testSignUpUser(self):
# 注册成功
rv = self.signUp("tom2333", "tom%@", "1")
user = User_helper.query_first_name_password(Users("tom2333", "tom%@"))
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("tom2333", "tom%@"))
self.assertIn("User registration successful!", rv.data)
def test_recise_name(self):
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.update_user_name_by_user_extra(
Users("tomeUpdate", "tom jobn"))
self.assertTrue(isSuccess)
userList = self.helper.query_all_with_user_name(
Users("tomeUpdate", "tom job"))
self.assertEqual(len(userList), 1)
def testSignUpUserLose(self):
# 注册失败
self.signUp("tom2333", "tom12333", "1")
rv = self.signUp("tom2333", "tom12333", "1")
user = User_helper.query_first_name_password(
Users("tom2333", "tom12333"))
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("tom2333", "tom12333"))
self.assertIn("User registration failed!", rv.data)
def test_query_one(self):
# 查询相同name数据，取一个 测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn2"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn3"))
self.assertTrue(isSuccess)
userList = self.helper.query_one(Users("tome", "tom jobn4"))
self.assertEqual(len(userList), 1)
def test_Change(self):
# 修改密码成功
self.signUp("Jar123", "Jar1234", "1")
rv = self.Login("Jar123", "Jar1234")
self.assertIn("Login Success", rv.data)
rv = self.Change("Jar123", "Jar1234", "Jar456")
user = User_helper.query_first_name_password(Users("Jar123", "Jar456"))
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("Jar123", "Jar456"))
self.assertIn('Change Success', rv.data)
def test_userLogout(self):
# 注销测试
self.signUp("Jar123", "Jar1234", "1")
rv = self.Login("Jar123", "Jar1234")
user = User_helper.query_first_name_password(Users(
"Jar123", "Jar1234"))
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("Jar123", "Jar1234"))
self.assertIn("Login Success", rv.data)
rv = self.logout()
self.assertIn(b'You were logged out', rv.data)
def test_query_all_name_password(self):
# 查询所有相同name，password数据测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn2"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn3"))
self.assertTrue(isSuccess)
userList = self.helper.query_all_with_user_name_password(
Users("tome", "tom jobn"))
self.assertEqual(len(userList), 1)
def test_allArticle(self):
# 发表文章
self.signUp("Jar123", "Jar1234", "1")
rv = self.Login("Jar123", "Jar1234")
self.assertIn("Login Success", rv.data)
rv = self.addArticle("fish", "fish is good!")
user = User_helper.query_first_name_password(Users(
"Jar123", "Jar1234"))
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("Jar123", "Jar1234"))
Article_helper.delete(Article("fish", "fish is good!"))
self.assertIn("text published successfully", rv.data)
def test_query_all(self):
# 查询所有相同name数据测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn2"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn3"))
self.assertTrue(isSuccess)
userList = self.helper.query_all_with_user_name(
Users("tome", "tom jobn"))
# print userList
self.assertEqual(len(userList), 3)
def testLoginUser(self):
# 登陆成功
user = User_helper.query_first_name_password(Users("tom2333", "%@123"))
if user:
User_Role_helper.delete(User_Role(user.id, "1"))
User_helper.delete(Users("tom2333", "%@123"))
self.signUp("tom2333", "%@123", "1")
rv = self.Login("tom2333", "%@123")
self.assertIn("Login Success", rv.data)
def test_query_one_name_password(self):
# 查询一个相同name，password数据测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn2"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn3"))
self.assertTrue(isSuccess)
userList = self.helper.query_first_name_password(
Users("tome", "tom jobn"))
# print userList
self.assertIsNotNone(userList)
def Change():
# 修改密码
user_name = request.form['username']
old_password = request.form['old_password']
new_password = request.form['new_password']
userss = Users(user_name, old_password)
users=Users(user_name, new_password)
user = User_helper.query_first_name_password(userss)
if user>0 :
users=User_helper.update_user_extra_by_user_name(users)
if users>0:
return render_template('changeSuccess.html')
else:
return render_template('changeFail.html')
else:
return render_template('changeFail.html')
def testSignUpUser(self):
rv = self.signUp("tom2333", "tom12333")
print type(rv)
print rv.data
user_orm_helper.delete(Users("tom2333", "tom12333"))
self.assertIn("Sign Up Success!", rv.data)
def signUp():
user_name = request.form['username']
pass_word = request.form['password']
user = Users(user_name, pass_word)
isSuccess = user_orm_helper.addUser(user)
if isSuccess:
return render_template('signUpSuccess.html')
else:
return render_template('signUpLose.html')
def login():
"""User login/authentication/session management."""
user_name = request.form['username']
pass_word = request.form['password']
user = Users(user_name, pass_word)
isSuccess = user_orm_helper.query_all_with_user_name_password(user)
if len(isSuccess) > 0:
session['logged_in'] = True
return render_template('loginSuccess.html')
else:
return render_template('loginFail.html')
def signUp():
# 注册
user_name = request.form['username']
pass_word = request.form['password']
role_Id = request.form['roleId']
user = Users(user_name, pass_word)
isSuccess = User_helper.addUser(user)
user = User_helper.query_first_name_password(user)
if isSuccess:
addUserRoleSuccess=User_Role_helper.addUserRole(User_Role(user.id, int(role_Id)))
if addUserRoleSuccess:
return render_template('signUpSuccess.html')
else:
return render_template('signUpLose.html')
else:
return render_template('signUpLose.html')
def login():
"""登录."""
user_name = request.form['username']
pass_word = request.form['password']
user = Users(user_name, pass_word)
loginUser = User_helper.query_first_name_password(user)
user_Role = User_Role_helper.query_first_userId(loginUser.id if loginUser else None)
rId=int(user_Role.roleId if user_Role else "404")
role = Role_helper.queryRoleByRoleId(rId)
session['user_id']=user_name
session[user_name + 'Role'] = role.function if role else "None"
sessions = session[user_name + 'Role']
if sessions == "CRUD":
return render_template('loginSuccess.html')
elif sessions == "CR":
return render_template('loginSuccess.html')
elif sessions == "R":
return render_template('loginSuccess.html')
elif sessions == "None":
return render_template('loginFail.html')
def test_add(self):
# 插入测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertFalse(isSuccess)
def test_delete(self):
# 删除测试
isSuccess = self.helper.addUser(Users("tome", "tom jobn"))
self.assertTrue(isSuccess)
isSucess = self.helper.delete(Users("tome", "tom jobn"))
self.assertTrue(isSucess)
def test_query_none(self):
# 查询相同name数据，取第一个 测试
userList = self.helper.query_first_with_user_name(
Users("tom", "tom job"))
self.assertEqual(userList, None)
