adjustments due to refactor

This commit is contained in:
Jonas Winkler 2020-11-17 17:12:34 +01:00
parent 829785268e
commit b366d1b88f
2 changed files with 10 additions and 10 deletions

View File

@ -213,12 +213,12 @@ def handle_message(message, rule):
async_task( async_task(
"documents.tasks.consume_file", "documents.tasks.consume_file",
file=temp_filename, path=temp_filename,
original_filename=att.filename, override_filename=att.filename,
force_title=title, override_title=title,
force_correspondent_id=correspondent.id if correspondent else None, override_correspondent_id=correspondent.id if correspondent else None,
force_document_type_id=doc_type.id if doc_type else None, override_document_type_id=doc_type.id if doc_type else None,
force_tag_ids=[tag.id] if tag else None, override_tag_ids=[tag.id] if tag else None,
task_name=f"Mail: {att.filename}" task_name=f"Mail: {att.filename}"
) )

View File

@ -89,11 +89,11 @@ class TestMail(TestCase):
args1, kwargs1 = m.call_args_list[0] args1, kwargs1 = m.call_args_list[0]
args2, kwargs2 = m.call_args_list[1] args2, kwargs2 = m.call_args_list[1]
self.assertEqual(kwargs1['force_title'], "test1") self.assertEqual(kwargs1['override_title'], "test1")
self.assertEqual(kwargs1['original_filename'], "test1.pdf") self.assertEqual(kwargs1['override_filename'], "test1.pdf")
self.assertEqual(kwargs2['force_title'], "test2") self.assertEqual(kwargs2['override_title'], "test2")
self.assertEqual(kwargs2['original_filename'], "test2.pdf") self.assertEqual(kwargs2['override_filename'], "test2.pdf")
@mock.patch("paperless_mail.mail.async_task") @mock.patch("paperless_mail.mail.async_task")
def test_handle_empty_message(self, m): def test_handle_empty_message(self, m):