Fix tests

2024-10-05 07:41:59 -07:00 · 2024-10-05 07:41:59 -07:00 · 61723528f2
commit 61723528f2
parent 3386eaee6d
2 changed files with 6 additions and 2 deletions
--- a/src/documents/tests/test_migration_workflows.py
+++ b/src/documents/tests/test_migration_workflows.py
@ -8,7 +8,7 @@ class TestMigrateWorkflow(TestMigrations):
    dependencies = (
        (
            "paperless_mail",
-            "0026_mailrule_enabled",
+            "0027_mailaccount_expiration_mailaccount_refresh_token",
        ),
    )

--- a/src/paperless_mail/mail.py
+++ b/src/paperless_mail/mail.py
@ -572,7 +572,11 @@ class MailAccountHandler(LoggingMixin):
                account.imap_port,
                account.imap_security,
            ) as M:
-                if account.is_token and account.expiration < timezone.now():
+                if (
+                    account.is_token
+                    and account.expiration is not None
+                    and account.expiration < timezone.now()
+                ):
                    self.log.debug(f"Attempting to refresh token for account {account}")
                    success = self.refresh_token(account)
                    if not success: