X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/7ef4abfc9ef8f8835f83cec90040a8d91a659d00..fdb67a9ef3b41c31d4ae4122d49537860e483ec6:/apps/picture/tests/picture_import.py

diff --git a/apps/picture/tests/picture_import.py b/apps/picture/tests/picture_import.py
index 202acdd65..68c875391 100644
--- a/apps/picture/tests/picture_import.py
+++ b/apps/picture/tests/picture_import.py
@@ -2,11 +2,11 @@
 from __future__ import with_statement
 
 from os import path
-from django.test import TestCase
 from picture.models import Picture
+from catalogue.test_utils import WLTestCase
 
 
-class PictureTest(TestCase):
+class PictureTest(WLTestCase):
     
     def test_import(self):
         picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"))
@@ -22,3 +22,12 @@ class PictureTest(TestCase):
 
         picture.delete()
         
+
+    def test_import_2(self):
+        picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/pejzaz-i-miasto-krzyzanowski-chmury.xml"),
+                                        path.join(path.dirname(__file__), "files/pejzaz-i-miasto-krzyzanowski-chmury.jpg"),
+                                        overwrite=True)
+        cats = set([t.category for t in picture.tags])
+        assert 'genre' in cats
+        assert 'kind' in cats
+