X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/a70cbcf401eb1ebc0009cc1ce9d0d8e664cc9642..4298f7948e8963176debf5877fce30c49ae4e3ad:/src/picture/tests/test_picture_import.py?ds=sidebyside

diff --git a/src/picture/tests/test_picture_import.py b/src/picture/tests/test_picture_import.py
new file mode 100644
index 000000000..022b33c62
--- /dev/null
+++ b/src/picture/tests/test_picture_import.py
@@ -0,0 +1,42 @@
+# -*- 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 os import path
+from picture.models import Picture
+from catalogue.test_utils import WLTestCase
+
+
+class PictureTest(WLTestCase):
+
+    def test_import(self):
+        picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"))
+
+        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 == {(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()
+
+    def test_import_with_explicit_image(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"))
+
+        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