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()
