def _login(use_cookie):
"""User login helper function."""
data = _get_request_data()
if 'email' not in data or 'password' not in data:
raise exception_handler.BadRequest(
'missing email or password in data'
)
if 'expires' not in data:
expire_timestamp = (
datetime.datetime.now() + app.config['REMEMBER_COOKIE_DURATION']
)
else:
expire_timestamp = util.parse_datetime(
data['expires'], exception_handler.BadRequest
)
data['expire_timestamp'] = expire_timestamp
user = auth_handler.authenticate_user(**data)
if not login_user(user, remember=data.get('remember', False)):
raise exception_handler.UserDisabled('failed to login: %s' % user)
user_log_api.log_user_action(user.id, request.path)
response_data = {'id': user.id}
user_api.record_user_token(
user, user.token, user.expire_timestamp
)
if not use_cookie:
response_data['token'] = user.token
response_data['expires'] = util.format_datetime(user.expire_timestamp)
return utils.make_json_response(200, response_data)
def test_get_user_object_from_token(self):
token = user_api.record_user_token(
self.user_object,
'test_token',
datetime.datetime.now() + datetime.timedelta(seconds=10000)
)
self.assertIsNotNone(token)
def test_get_user_object_from_token(self):
token = user_api.record_user_token(
'test_token',
datetime.datetime.now() + datetime.timedelta(seconds=10000),
user=self.user_object,
)
self.assertIsNotNone(token)
def test_record_user_token(self):
token = user_api.record_user_token(
self.user_object,
'test_token',
datetime.datetime.now() + datetime.timedelta(seconds=10000)
)
self.assertIsNotNone(token)
self.assertEqual(token['token'], 'test_token')
def test_record_user_token(self):
token = user_api.record_user_token(
'test_token',
datetime.datetime.now() + datetime.timedelta(seconds=10000),
user=self.user_object,
)
self.assertIsNotNone(token)
self.assertEqual(token['token'], 'test_token')
