def test_encode_auth_token(self): user = User( first_name = "John", last_name = "Doe", password = "******", date_registered = datetime.datetime.utcnow() ) db.session.add(user) db.session.commit() auth_token = user.encode_auth_token(user.id) self.assertTrue(isinstance(auth_token, bytes))
def test_get(self): user = self.user_dao.get(72) role = user.role dict_ = user.to_dict() user2 = User() user2.from_dict(**dict_) print(user2)
def create_user(self, user: CreateUserDto): fake_hashed_password = user.password + "notreallyhashed" db_user = User(email=user.email, hashed_password=fake_hashed_password) self.db.add(db_user) self.db.commit() self.db.refresh(db_user) return db_user
def get_logged_in_user(new_request): auth_token = new_request.headers.get("Authorization") if auth_token: resp = User.decode_auth_token(auth_token) if isinstance(resp, str): user = User.query.filter_by(public_id=resp).first() if user: response_object = { "status": "success", "data":{ "user_id": resp, "email": user.email, "admin": user.is_admin, "registered_on": str(user.date_registered) } } return response_object, 200 response_object = { "status": "fail", "message": resp } return response_object, 401 else: response_object = { "status": "fail", "message": "Provide a valid auth token" } return response_object, 401
def logout_user(data): if data: auth_token = data.split(" ")[1] else: auth_token = "" if auth_token: resp = User.decode_auth_token(auth_token) if not isinstance(resp, str): return save_token(token=auth_token) else: response_object = { "status": "fail", "message": resp } return response_object, 401 else: response_object = { "status": "fail", "message": "Provide a valid auth token." } return response_object, 403
def save_new_patient_and_booking(data): if booking_available(data): user = User.query.filter_by(email=data['email']).first() print("user is", user) if not user: dob_data = data['dob'].split("/") dob = date(int(dob_data[2]), int(dob_data[1]), int(dob_data[0])) # Create a new new user = User(first_name=data['first name'], last_name=data['last name'], email=data['email'], phone=data['phone'], dob=dob, gender=data['gender'], password=data['password'], street=data['address'], suburb=data['suburb'], state=data['state'], postcode=data['postcode']) db.session.add(user) user_id = db.session.query(db.func.max(User.id)).scalar() booking_outcome = save_new_booking(data, user_id) return booking_outcome else: response_object = { 'status': 'fail', 'message': 'User already registered. Please log in.' } return response_object, 409
def test_create(self): user = User(username='******', password='******', role_id=1, email='*****@*****.**', edu_level=EducationLevel.BACHELOR) self.user_dao.insert(user)
def test_create_users(self): for i in range(50): username = '******' % i edu_level = EducationLevel.BACHELOR if i % 2 == 0 else EducationLevel.MASTER email = '*****@*****.**' % i user = User(username=username, password='', email=email, edu_level=edu_level, role_id=1) self.user_dao.insert(user)
def save_new_user(data): user = User.query.filter_by(email=data['email']).first() if not user: new_user = User(public_id=str(uuid.uuid4()), email=data['email'], username=data['username'], password=data['password'], registered_on=datetime.datetime.utcnow()) save_changes(new_user) response_object = { 'status': 'success', 'message': 'Successfully registered.' } return response_object, 201 else: response_object = { 'status': 'fail', 'message': 'User already exists. Please Log in.', } return response_object, 409
def get_page_by_filter(self, page, per_page, sort=[], filter=None): # sort sort_ = [] for e in sort: sort_.append(User.get_sort_filter(e.field, e.order)) # filter filter_ = {} if filter is not None: if filter.username is not None and filter.username != '': filter_['username'] = User.username.like('%{}$'.format( filter.username)) if filter.edu_level is not None: filter_['edu_level'] = filter.edu_level if filter.role_id is not None: filter_['role_id'] = filter.role_id pagination = self.dao.get_page_by_filter(page, per_page, tuple(sort_), **filter_) return pagination
def create_admin(data): admin = User.query.filter_by(email=data['email']).first() if not admin: new_admin = User( public_id=str(uuid.uuid4()), email=data['email'], first_name=data['first_name'], last_name=data['last_name'], password=data['password'], date_registered=datetime.datetime.utcnow(), is_admin = 1 ) save_changes(new_admin) return generate_token(new_admin) else: response_object = { 'status': 'fail', 'message': 'User already exists. Please Log in.' } return response_object, 409
def save_new_user(data): user = User.query.filter_by(login=data['login']).first() if not user: new_user = User(public_id=str(uuid.uuid4()), login=data['login'], password=data['password'], firstname=data['firstname'], lastname=data['lastname'], git_login=data['git_login'], git_password=data['git_password'], registered_on=datetime.datetime.utcnow()) save_changes(new_user) response_object = { 'status': 'success', 'message': 'Successfully registered.' } return response_object, 201 else: response_object = { 'status': 'fail', 'message': 'Project already exists.' } return response_object, 409
def update(self, user_form: UserForm): self.user_service.update(User(**user_form.data)) return Response.success()
def create(self, user_form: UserForm(exclude=['id'])): self.user_service.create(User(**user_form.data)) return Response.success()
def test_init(self): user = User(username='******', password='******') self.assertEqual(user.username, "admin") self.assertNotEqual(user.password, None) self.assertNotEqual(user.salt, None)
def test_login(self): user = User(username='******', password='******')