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(
"documents.tasks.consume_file",
file=temp_filename,
original_filename=att.filename,
force_title=title,
force_correspondent_id=correspondent.id if correspondent else None,
force_document_type_id=doc_type.id if doc_type else None,
force_tag_ids=[tag.id] if tag else None,
path=temp_filename,
override_filename=att.filename,
override_title=title,
override_correspondent_id=correspondent.id if correspondent else None,
override_document_type_id=doc_type.id if doc_type else None,
override_tag_ids=[tag.id] if tag else None,
task_name=f"Mail: {att.filename}"
)

View File

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