def https_login(session): pem_file_path = os.path.join(ROOT_DIR, 'server.pem') valid_token = User.encode_auth_token(1).decode('utf-8') result = session.post(HTTPS_URI(LOGIN_PATH), json=dict(token=valid_token), verify=pem_file_path) print(result.text)
def register_user(): user_data = request.get_json() user = User.query.filter_by(email=user_data.get('email')).first() if not user: try: #Set data to be entered into the user's table user = User(email=user_data.get('email'), password=user_data.get('password'), phone=user_data.get('phone')) #insert the data db.session.add(user) db.session.commit() # generate JWT token auth_token = user.encode_auth_token(user.id) responseObject = { 'status': 'success', 'message': 'Successfully registered.', 'auth_token': auth_token.decode() } return make_response(jsonify(responseObject)), 201 except Exception as e: responseObject = { 'status': 'fail', 'message': 'Some error occurred. Please try again.' } return make_response(jsonify(responseObject)), 401 else: responseObject = { 'status': 'fail', 'message': 'User already exists. Please Log in.', } return make_response(jsonify(responseObject)), 202
def setUpClass(cls): cls.app = cls.create_app() cls.client = socketio.test_client(app=cls.app) cls.client.testing = True cls.http_client = cls.app.test_client() thing = User(name='thing') cls.valid_token = thing.encode_auth_token(1).decode('utf-8')
def test_encode_auth_token(self): user = User( email='*****@*****.**', password='******', registered_on=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 init_db(self): self.expired_token = ( 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.' 'eyJleHAiOjE1MjA2OTI5MzEsImlhdCI6MTUxODEwMDkzMSwic3ViIjoxfQ.' '4uuIsAIpxng-dcPQxXQvN3vjibxxeJpQFcaqBw0z4Eg') thing1 = User(name='thing1') thing2 = User(name='thing2') self.valid_token = User.encode_auth_token(1).decode('utf-8') db.session.add(thing1) db.session.add(thing2) db.session.commit()