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)