Make tests pass

This commit is contained in:
Tikitu de Jager
2016-03-07 21:37:18 +02:00
parent 4065d14fab
commit ad07eec3e1
2 changed files with 22 additions and 8 deletions

View File

@@ -1,12 +1,11 @@
from django.test import TestCase
from ..consumer import Consumer
from ..models import FileInfo
class TestAttachment(TestCase):
TAGS = ("tag1", "tag2", "tag3")
CONSUMER = Consumer()
SUFFIXES = (
"pdf", "png", "jpg", "jpeg", "gif",
"PDF", "PNG", "JPG", "JPEG", "GIF",
@@ -16,14 +15,14 @@ class TestAttachment(TestCase):
def _test_guess_attributes_from_name(self, path, sender, title, tags):
for suffix in self.SUFFIXES:
f = path.format(suffix)
results = self.CONSUMER._guess_attributes_from_name(f)
self.assertEqual(results[0].name, sender, f)
self.assertEqual(results[1], title, f)
self.assertEqual(tuple([t.slug for t in results[2]]), tags, f)
file_info = FileInfo.from_path(f)
self.assertEqual(file_info.correspondent.name, sender, f)
self.assertEqual(file_info.title, title, f)
self.assertEqual(tuple([t.slug for t in file_info.tags]), tags, f)
if suffix.lower() == "jpeg":
self.assertEqual(results[3], "jpg", f)
self.assertEqual(file_info.suffix, "jpg", f)
else:
self.assertEqual(results[3], suffix.lower(), f)
self.assertEqual(file_info.suffix, suffix.lower(), f)
def test_guess_attributes_from_name0(self):
self._test_guess_attributes_from_name(