Fixes #3396: More pictures info.

[wolnelektury.git] / apps / picture / tests / picture_import.py

diff --git a/apps/picture/tests/picture_import.py b/apps/picture/tests/picture_import.py
index 91fb35f..7b1f648 100644 (file)
--- a/apps/picture/tests/picture_import.py
+++ b/apps/picture/tests/picture_import.py
@@ -1,24 +1,28 @@
 # -*- coding: utf-8 -*-
 # -*- coding: utf-8 -*-

+# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#

 from __future__ import with_statement
 
 from __future__ import with_statement
 

-from django.core.files.base import ContentFile, File
-from catalogue.test_utils import *
-from catalogue import models
-from librarian import WLURI
+from os import path

 from picture.models import Picture
 from picture.models import Picture

+from catalogue.test_utils import WLTestCase

 
 

-from nose.tools import raises
-import tempfile
-from os import unlink, path, makedirs

 
 

+class PictureTest(WLTestCase):

 
 

-class PictureTest(TestCase):
-    

     def test_import(self):
         picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"))
 
     def test_import(self):
         picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"))
 

-        motifs = set([tag.name for tag in picture.tags if tag.category == 'theme'])
-        assert motifs == set([u'nieporządek']), 'theme tags are wrong. %s' % motifs
+        themes = set()
+        for area in picture.areas.all():
+            themes.update([(tag.category, tag.name)
+                for tag in area.tags if tag.category in (u'theme', u'thing')])
+        assert themes == set([(u'theme', u'nieporządek'), (u'thing', u'kosmos')]), \
+            'Bad themes on Picture areas: %s' % themes
+
+        pic_themes = set([tag.name for tag in picture.tags if tag.category in ('theme', 'thing')])
+        assert not pic_themes, 'Unwanted themes set on Pictures: %s' % pic_themes

 
         picture.delete()
 
 
         picture.delete()
 


@@ -27,4 +31,13 @@ class PictureTest(TestCase):
                                         path.join(path.dirname(__file__), "files/kandinsky-composition-viii.png"))
 
         picture.delete()
                                         path.join(path.dirname(__file__), "files/kandinsky-composition-viii.png"))
 
         picture.delete()

-        
+
+
+    def test_import_2(self):
+        picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"),
+                                        path.join(path.dirname(__file__), "files/kandinsky-composition-viii.png"),
+                                        overwrite=True)
+        cats = set([t.category for t in picture.tags])
+        assert 'epoch' in cats
+        assert 'kind' in cats
+