paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-10-30 03:56:23 -05:00
Code Issues Packages Projects Releases Wiki Activity

Tests for the consumer

Browse Source
This commit is contained in:
Daniel Quinn
2016-02-11 12:25:23 +00:00
parent cef4abc01d
commit ef1639208c
1 changed files with 76 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
src/documents/tests/test_consumer.py Normal file
View File

@@ -0,0 +1,76 @@
from django.test import TestCase
from ..consumer import Consumer
class TestAttachment(TestCase):
def test_guess_attributes_from_name(self):
consumer = Consumer()
suffixes = ("pdf", "png", "jpg", "jpeg", "gif")
tests = (
{
"path": "/path/to/Sender - Title - tag1,tag2,tag3.{}",
"result": {
"sender": "Sender",
"title": "Title",
"tags": ("tag1", "tag2", "tag3")
},
},
{
"path": "/path/to/Spaced Sender - Title - tag1,tag2,tag3.{}",
"result": {
"sender": "Spaced Sender",
"title": "Title",
"tags": ("tag1", "tag2", "tag3")
},
},
{
"path": "/path/to/Sender - Spaced Title - tag1,tag2,tag3.{}",
"result": {
"sender": "Sender",
"title": "Spaced Title",
"tags": ("tag1", "tag2", "tag3")
},
},
{
"path": "/path/to/Spaced Sender - Spaced Title - tag1,tag2.{}",
"result": {
"sender": "Spaced Sender",
"title": "Spaced Title",
"tags": ("tag1", "tag2")
},
},
{
"path": "/path/to/Dash-Sender - Title - tag1,tag2.{}",
"result": {
"sender": "Dash-Sender",
"title": "Title",
"tags": ("tag1", "tag2")
},
},
{
"path": "/path/to/Sender - Dash-Title - tag1,tag2.{}",
"result": {
"sender": "Sender",
"title": "Dash-Title",
"tags": ("tag1", "tag2")
},
},
{
"path": "/path/to/Dash-Sender - Dash-Title - tag1,tag2.{}",
"result": {
"sender": "Dash-Sender",
"title": "Dash-Title",
"tags": ("tag1", "tag2")
},
},
)
for test in tests:
for suffix in suffixes:
f = test["path"].format(suffix)
sender, title, tags, s = consumer._guess_attributes_from_name(f)
self.assertEqual(sender.name, test["result"]["sender"])
self.assertEqual(title, test["result"]["title"])
self.assertEqual(tags, test["result"]["tags"])
self.assertEqual(s, suffix)