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

Include eml in test_handle_encrypted_message

Browse Source
This commit is contained in:
Daniel Bankmann 2024-08-12 22:44:45 +02:00 committed by shamoon
parent 5e40321883
commit 243daf7469
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/paperless_mail/tests/test_mail.py
View File

@ -1385,6 +1385,7 @@ class TestMail(
subject="the message title", subject="the message title",
from_="Myself", from_="Myself",
attachments=2, attachments=2,
body="Test mail",
) )
encrypted_message = self.messageEncryptor.encrypt(message) encrypted_message = self.messageEncryptor.encrypt(message)
@ -1392,18 +1393,25 @@ class TestMail(
account = MailAccount.objects.create() account = MailAccount.objects.create()
rule = MailRule( rule = MailRule(
assign_title_from=MailRule.TitleSource.FROM_FILENAME, assign_title_from=MailRule.TitleSource.FROM_FILENAME,
consumption_scope=MailRule.ConsumptionScope.EVERYTHING,
account=account, account=account,
) )
rule.save() rule.save()
result = self.mail_account_handler._handle_message(encrypted_message, rule) result = self.mail_account_handler._handle_message(encrypted_message, rule)
self.assertEqual(result, 2) self.assertEqual(result, 3)
self._queue_consumption_tasks_mock.assert_called() self._queue_consumption_tasks_mock.assert_called()
self.assert_queue_consumption_tasks_call_args( self.assert_queue_consumption_tasks_call_args(
[ [
[
{
"override_title": message.subject,
"override_filename": f"{message.subject}.eml",
},
],
[ [
{"override_title": "file_0", "override_filename": "file_0.pdf"}, {"override_title": "file_0", "override_filename": "file_0.pdf"},
{"override_title": "file_1", "override_filename": "file_1.pdf"}, {"override_title": "file_1", "override_filename": "file_1.pdf"},