def test_the_is_token_revoked_method_of_user_repository(app, mocker):
"""
GIVEN the TokenRepository instance
WHEN the is_token_revoked() method is call
THEN check session method calls
"""
from app.model import TokenRepository, Token
token_repository = TokenRepository()
token_repository.session = UnifiedAlchemyMagicMock()
token = Token()
token.id = 1
token.jti = "27d1b1a3-45b4-4a5f-83ed-b823f5ea1dbe"
token.token_type = "access"
token.user_identity = "test"
token.revoked = True
token.expires = datetime.now()
token_repository.session.add(token)
token_repository.session.commit()
decoded_token = mocker.MagicMock()
decoded_token['jti'] = token.jti
result = token_repository.is_token_revoked(decoded_token)
assert result
(token_repository.session.query.return_value.filter_by.
assert_called_once_with(jti=decoded_token['jti']))
def test_the_change_token_revoking_method_of_user_repository(app):
"""
GIVEN the TokenRepository instance
WHEN the save() method is call
THEN check session method calls and the token revoke value
"""
from app.model import TokenRepository, Token
token_repository = TokenRepository()
token_repository.session = UnifiedAlchemyMagicMock()
token = Token()
token.id = 1
token.jti = "27d1b1a3-45b4-4a5f-83ed-b823f5ea1dbe"
token.token_type = "access"
token.user_identity = "test"
token.revoked = False
token.expires = datetime.now()
token_repository.session.add(token)
token_repository.session.commit()
token_repository.change_token_revoking(1, "test", True)
(token_repository.session.query.return_value.filter_by.
assert_called_once_with(id=1, user_identity="test"))
assert token.revoked == True
def test_the_revoke_all_tokens_method_of_user_repository(app):
"""
GIVEN the TokenRepository instance
WHEN the revoke_all_tokens() method is call
THEN check session method calls
"""
from app.model import TokenRepository, Token
token_repository = TokenRepository()
token_repository.session = UnifiedAlchemyMagicMock()
acc_token = Token()
acc_token.id = 1
acc_token.jti = "27d1b1a3-45b4-4a5f-83ed-b823f5ea1dbe"
acc_token.token_type = "access"
acc_token.user_identity = "test"
acc_token.revoked = False
acc_token.expires = datetime.now()
token_repository.session.add(acc_token)
ref_token = Token()
ref_token.id = 2
ref_token.jti = "27d1b1a3-45b4-4a5f-83ed-b823f5ea1dbd"
ref_token.token_type = "refresh"
ref_token.user_identity = "test"
ref_token.revoked = False
ref_token.expires = datetime.now()
token_repository.session.add(ref_token)
token_repository.session.commit()
token_repository.revoke_all_tokens("test")
(token_repository.session.query.return_value.filter_by.
assert_called_once_with(user_identity="test"))
(token_repository.session.query.return_value.filter_by.return_value.update.
assert_called_once_with({Token.revoked: True}))