def setUp(self): """ Test Setup """ self.browser = Browser("phantomjs") self.browser.driver.set_window_size(1440, 900) # set browser size. # set up the tables in the database Base.metadata.create_all(engine) # create 2 example users self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit() self.user2 = models.User(name="Ryan", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user2) session.commit() # Create some sample posts content = "Acceptance testing content" for i in range(5): post = models.Post(title="Acceptance test post #{}".format(i + 1), content=content, author=self.user) session.add(post) session.commit() """ gives the ability to start and run other code simultaneously with your own scripts """ self.process = multiprocessing.Process(target=app.run) self.process.start() # pause for 1 second time.sleep(1)
def create(request: schemas.User, db: Session): new_user = models.User(name=request.name, email=request.email, password=Hash.bcrypt(request.password)) db.add(new_user) db.commit() db.refresh(new_user) return new_user
def create_user(request: schemas.User, db: Session = Depends(get_db)): user_detail = request.dict() password = user_detail.get('password') user_detail['password'] = get_hashed_pass(password) new_user = models.User(**user_detail) db.add(new_user) db.commit() db.refresh(new_user) return {"detail": "user created succesfully"}
def generate_user(): user = models.User() user.name = fake.name() user.email = fake.ascii_email() models.UserHandler(user).add_pwd("password") print("Created new user {}".format(user)) return user
def register_user(db: Session, username, email, password): pass_hash_obj = Hash(password) hashed_password = pass_hash_obj.hash_password(password) new_user = models.User(username=username, email=email, password=hashed_password) db.add(new_user) db.commit() db.refresh(new_user) return new_user
def create(request: schemas.User, db: Session): hashed_password = hash.Hash.bcrypt(request.password) new_user = models.User(name=request.name, username=request.username, password=hashed_password) db.add(new_user) db.commit() db.refresh(new_user) return new_user
def setUp(self): """ Test setup """ self.client = app.test_client() # Set up the tables in the database Base.metadata.create_all(engine) # Create two example users self.alice = models.User(name = "Alice" , email = "*****@*****.**" , password = generate_password_hash("test") ) self.bob = models.User(name = "Bob" , email = "*****@*****.**" , password = generate_password_hash("test") ) session.add(self.alice) session.add(self.bob) session.commit()
def setUp(self): """ Test setup """ self.client = app.test_client() # set up the tables in the database Base.metadata.create_all(engine) # create an example user self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit()
def setUp(self): """ Test setup """ # Create a test client that allows inspection of app views/responses self.client = app.test_client() # Set up the tables in the testing database Base.metadata.create_all(engine) # Create an example user in the testing database self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit()
def setUp(self): """ Test setup """ self.browser = Browser("phantomjs") # Set up the tables in the database Base.metadata.create_all(engine) # Create an example user self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit() self.process = multiprocessing.Process(target=app.run) self.process.start() time.sleep(1)
def post(self): json_data = request.get_json(force=True) try: id = models.User.query.order_by(models.User.id.desc()).first().id new_id = int(id) + 1 except: new_id = 1 print('new_id', new_id) res = models.User(id=new_id, username=json_data['username'], email=json_data['email'], password=json_data['password']) db.session.add(res) db.session.commit() db.session.close() return json_data, 200
def setUp(self): """Test setup""" #create test client using the app.test_client function #--> allows us to make requests to views and inspect the responses we get from the app self.client = app.test_client() #set up the tables in the database Base.metadata.create_all(engine) #create an example user --> use this user to login and be the author of our test post self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) #add user to the database session.add(self.user) session.commit()
def register(): if current_user.is_authenticated: return redirect(url_for('home')) form = forms.RegistrationForm() if form.validate_on_submit(): hashed = crypto.generate_password_hash( form.password.data).decode('utf-8') user = models.User(username=form.username.data, email=form.email.data, password=hashed) db.session.add(user) db.session.commit() flash( u'Account created for {0}, Please login!'.format( form.username.data), 'success') return redirect(url_for('login')) return render_template('reg.html', title='register', form=form)
def add_user(): userform = forms.NewUserForm() if userform.validate_on_submit(): name = userform.name.data pwd = userform.password.data user = models.User(name=name) handler = models.UserHandler(user) handler.add_pwd(pwd) if not handler.add_to_db(): flask.flash("Unable to create user") return flask.redirect(flask.url_for('add_user')) else: flask.flash("Welcome {}!".format(user.name)) return flask.redirect(flask.url_for('homepage')) return flask.render_template("newuser.html", userform=userform)
def register(request): email = request.POST['email'] username = request.POST['username'] password = request.POST['password'] reward = request.POST['reward'] verification_code = request.POST["verificationCode"] cache_verification_code = cache.get(email) # 验证信息 re_email = re.search( r'^([a-zA-Z]|[0-9])(\w|)+@[a-zA-Z0-9]+\.([a-zA-Z]{2,4})$', email) if re_email: pass else: return Response('error_email') re_username = re.search(r'^[a-zA-Z0-9_-]{4,16}$', username) if re_username: pass else: return Response('error_username') re_password = re.search( r'^.*(?=.{8,})(?=.*\d)(?=.*[a-zA-Z])(?=.*[!@#$%^&*?.]).*$', password) re_reward = re.search( r'^.*(?=.{8,})(?=.*\d)(?=.*[a-zA-Z])(?=.*[!@#$%^&*?.]).*$', reward) if re_password and re_reward: pass else: return Response('error_password') if password != reward: return Response('密码不一致') if cache_verification_code != verification_code: return Response("验证码错误") new_password = make_password(password) user = models.User(email=email, username=username, password=new_password) user.save() Token.objects.get_or_create(user=user) token = Token.objects.get(user=user) userinfo = { 'username': user.username, 'token': token.key, } return Response(userinfo)
def login_check(request): # 添加超级用户 if not models.User.objects.filter(user_type='admin'): import hashlib adminUser = models.User() adminUser.user_id = 'administrator' adminUser.user_password = hashlib.md5(b'123456').hexdigest() adminUser.create_time = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()) adminUser.user_type = 'admin' adminUser.save() return func(request) if not request.session.has_key('expire_time'): return HttpResponseRedirect('/administrator/login/') else: if request.session.get('expire_time')<time.time(): # request.sesstion.clear() # request.sesstion.flush() return render(request, 'administrator/login.html', {'code': 0, 'msg': '登录已过期,请重新登录'}) else: return func(request)
def register(request): """ ajax 注册验证 :param request: :return: 验证信息 """ user_name = request.POST.get('user-name') user_phone = request.POST.get('user-phone') user_pwd = request.POST.get('user-pwd') user_count = models.User.objects.filter(userName=user_name).count() if user_count > 0: json_text = {'code': '1', 'msg': '昵称已被使用,换一个吧'} return HttpResponse(json.dumps(json_text)) else: user = models.User( userName=user_name, pwd=user_pwd, ) user.save() models.UserInfo.objects.create(userPhone=user_phone, user=user) json_text = {'code': '0', 'msg': '注册成功点击登录'} return HttpResponse(json.dumps(json_text))
def setUp(self): """Test setup""" #create an instance of the splinter.Browser class, using the (PhantomJS driver) <-- use for our automation self.browser = Browser("phantomjs") #set up the tables in the database Base.metadata.create_all(engine) #create an example user self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit() """ Understand multiprocessing from lesson, and more """ #use multiprocessing module in order to start the Flask test server running #since we need a server up to run our app, since our test will be visiting the actual site self.process = multiprocessing.Process(target=app.run) self.process.start() time.sleep(1)
def setUp(self): """ Test setup """ # Unlike integration test, create browser instance rather than app instance # PhantomJS used for browser automation self.browser = Browser("phantomjs") # Set up the tables in the database Base.metadata.create_all(engine) # Create an example user self.user = models.User(name="Alice", email="*****@*****.**", password=generate_password_hash("test")) session.add(self.user) session.commit() # Use 'multiprocessing' module to run test server since test is going to actual website self.process = multiprocessing.Process(target=app.run) self.process.start() time.sleep(1) # Set url path self.url_path = "http://127.0.0.1:5000"
def createUser(request: schemas.Blog, db: Session): new_user = models.User(username=request.username, email=request.email, password=Hash.passwordHash(request.password)) db.add(new_user) db.commit() db.refresh(new_user) return new_user