def test_decode_token_method_raises_expection_with_an_expired_token(
self, mock_jwt_decode):
with self.assertRaises(Exception) as e:
mock_jwt_decode.side_effect = jwt.ExpiredSignature()
Auth.decode_token('tdtddyd')
self.assertEqual(e.exception.args[0], 'Token is Expired')
def test_user_returns_valid_user(self, mock_get_token):
with self.app.app_context():
mock_get_token.return_value = self.get_valid_token()
base_controller = BaseController(self.request_context.request)
return_value = base_controller.user()
assert return_value.get("id") == Auth.decode_token(
self.get_valid_token())['UserInfo']["id"]
assert return_value.get("first_name") == Auth.decode_token(
self.get_valid_token())['UserInfo']["first_name"]
assert return_value.get("last_name") == Auth.decode_token(
self.get_valid_token())['UserInfo']["last_name"]
assert return_value.get("firstName") == Auth.decode_token(
self.get_valid_token())['UserInfo']["firstName"]
assert return_value.get("lastName") == Auth.decode_token(
self.get_valid_token())['UserInfo']["lastName"]
assert return_value.get("email") == Auth.decode_token(
self.get_valid_token())['UserInfo']["email"]
assert return_value.get("name") == Auth.decode_token(
self.get_valid_token())['UserInfo']["name"]
assert return_value.get("picture") == Auth.decode_token(
self.get_valid_token())['UserInfo']["picture"]
assert return_value.get("roles") == Auth.decode_token(
self.get_valid_token())['UserInfo']["roles"]
def test_user_method_return_list_of_user_data_based_on_supplied_keys(self):
with self.app.test_request_context(path='/api/v1/vendors',
method='GET',
headers=self.headers()) as request:
decoded = Auth.decode_token(self.get_valid_token())
values = Auth.user('id', 'first_name', 'last_name', 'email')
id, first_name, last_name, email = values
self.assertIsInstance(values, list)
self.assertEquals(decoded['UserInfo']['id'], id)
self.assertEquals(decoded['UserInfo']['first_name'], first_name)
self.assertEquals(decoded['UserInfo']['last_name'], last_name)
self.assertEquals(decoded['UserInfo']['email'], email)
def user_first_and_last_name():
return (Auth.decode_token(
BaseTestCase.get_valid_token())['UserInfo']['firstName'],
Auth.decode_token(
BaseTestCase.get_valid_token())['UserInfo']['lastName'])
def user_email():
return \
Auth.decode_token(BaseTestCase.get_valid_token())['UserInfo'][
'email']
def user_id():
return \
Auth.decode_token(BaseTestCase.get_valid_token())['UserInfo']['id']
def test_decode_token_returns_dict_on_valid_token(self):
token = Auth.decode_token(self.get_valid_token())
if type(token) is dict:
assert True
else:
assert False
def test_decode_token_throws_exception_on_invalid_token(self):
try:
Auth.decode_token(self.get_invalid_token())
assert False
except Exception as e:
assert True
def user_first_and_last_name():
return (
Auth.decode_token(BaseTestCase.get_valid_token())["UserInfo"]["firstName"],
Auth.decode_token(BaseTestCase.get_valid_token())["UserInfo"]["lastName"],
)
def user_email():
return Auth.decode_token(BaseTestCase.get_valid_token())["UserInfo"]["email"]
def user_id():
return Auth.decode_token(BaseTestCase.get_valid_token())["UserInfo"]["id"]
