def test_encode_auth_token(self):
user = User(email='*****@*****.**', password='******', language='en_US')
db.session.add(user)
db.session.commit()
auth_token = user.encode_auth_token(user.id)
self.assertTrue(isinstance(auth_token, bytes))
def post(self):
"""
Register user
username min 5, max 50
password min 5, max 50
:return:
"""
try:
data = LoginRegisterParser.parse_args()
user = User.query.get(username=data['username'])
if user is None:
user = User(username=data['username'])
user.set_password(data['password'])
user.create()
data = {'token': user.encode_auth_token()}
return responses.SuccessResponse(data, status=201).send()
else:
raise CustomException(detail='A user with this data exist.')
except CustomException as e:
return responses.ErrorResponse(message=e.detail,
status=e.status_code).send()
def test_get_user_data(self):
with self.client:
user = User(
email="*****@*****.**",
password="******",
language="en_US",
phone="123456789"
)
db.session.add(user)
db.session.commit()
token = user.encode_auth_token(user.id)
response = self.client.get(
'/user/me',
headers=dict(
Authorization='Bearer ' + token.decode()
)
)
data = json.loads(response.data.decode())
self.assert200(response)
self.assertEqual(data['email'], '*****@*****.**')
self.assertEqual(data['phone'], '123456789')
