lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mock httpx_oauth refresh to actually test paperless one

Browse Source
This commit is contained in:
shamoon 2024-10-07 11:25:37 -07:00
parent f9671ccbd8
commit 1eb185e7c7
2 changed files with 22 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/paperless_mail/oauth.py
View File

@ -101,10 +101,6 @@ class PaperlessMailOAuth2Manager:
""" """
logger = logging.getLogger("paperless_mail") logger = logging.getLogger("paperless_mail")
logger.debug(f"Attempting to refresh oauth token for account {account}") logger.debug(f"Attempting to refresh oauth token for account {account}")
if not account.refresh_token:
logger.error(f"Account {account}: No refresh token available.")
return False
try: try:
result: OAuth2Token result: OAuth2Token
if account.account_type == MailAccount.MailAccountType.GMAIL_OAUTH: if account.account_type == MailAccount.MailAccountType.GMAIL_OAUTH:

33
src/paperless_mail/tests/test_mail_oauth.py
View File

@ -186,11 +186,11 @@ class TestMailOAuth(
@mock.patch("paperless_mail.mail.get_mailbox") @mock.patch("paperless_mail.mail.get_mailbox")
@mock.patch( @mock.patch(
"paperless_mail.oauth.PaperlessMailOAuth2Manager.refresh_account_oauth_token", "httpx_oauth.oauth2.BaseOAuth2.refresh_token",
) )
def test_refresh_token_on_handle_mail_account( def test_refresh_token_on_handle_mail_account(
self, self,
mock_refresh_account_oauth_token, mock_refresh_token,
mock_get_mailbox, mock_get_mailbox,
): ):
""" """
@ -216,13 +216,21 @@ class TestMailOAuth(
expiration=timezone.now() - timedelta(days=1), expiration=timezone.now() - timedelta(days=1),
) )
mock_refresh_account_oauth_token.return_value = True mock_refresh_token.return_value = {
"access_token": "test_access_token",
"refresh_token": "test_refresh_token",
"expires_in": 3600,
}
self.mail_account_handler.handle_mail_account(mail_account) self.mail_account_handler.handle_mail_account(mail_account)
mock_refresh_account_oauth_token.assert_called_once() mock_refresh_token.assert_called_once()
mock_refresh_account_oauth_token.reset_mock() mock_refresh_token.reset_mock()
mock_refresh_account_oauth_token.return_value = True mock_refresh_token.return_value = {
"access_token": "test_access_token",
"refresh_token": "test_refresh",
"expires_in": 3600,
}
outlook_mail_account = MailAccount.objects.create( outlook_mail_account = MailAccount.objects.create(
name="Test Outlook Mail Account", name="Test Outlook Mail Account",
username="test_username", username="test_username",
@ -235,15 +243,15 @@ class TestMailOAuth(
) )
self.mail_account_handler.handle_mail_account(outlook_mail_account) self.mail_account_handler.handle_mail_account(outlook_mail_account)
mock_refresh_account_oauth_token.assert_called_once() mock_refresh_token.assert_called_once()
@mock.patch("paperless_mail.mail.get_mailbox") @mock.patch("paperless_mail.mail.get_mailbox")
@mock.patch( @mock.patch(
"paperless_mail.oauth.PaperlessMailOAuth2Manager.refresh_account_oauth_token", "httpx_oauth.oauth2.BaseOAuth2.refresh_token",
) )
def test_refresh_token_on_handle_mail_account_fails( def test_refresh_token_on_handle_mail_account_fails(
self, self,
mock_refresh_account_oauth_token, mock_refresh_token,
mock_get_mailbox, mock_get_mailbox,
): ):
""" """
@ -270,10 +278,13 @@ class TestMailOAuth(
expiration=timezone.now() - timedelta(days=1), expiration=timezone.now() - timedelta(days=1),
) )
mock_refresh_account_oauth_token.return_value = False mock_refresh_token.return_value = {
"error": "test_error",
}
# returns 0 processed mails
self.assertEqual( self.assertEqual(
self.mail_account_handler.handle_mail_account(mail_account), self.mail_account_handler.handle_mail_account(mail_account),
0, 0,
) )
mock_refresh_account_oauth_token.assert_called_once() mock_refresh_token.assert_called_once()