X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/876c24227952faa17eda5d706cb0b3316d5ffe24..e7c8c7e1fd0fe2eb867b26ae205b75822204a45f:/apps/dictionary/tests.py

diff --git a/apps/dictionary/tests.py b/apps/dictionary/tests.py
index 0de7c5e6d..526819a01 100755
--- a/apps/dictionary/tests.py
+++ b/apps/dictionary/tests.py
@@ -5,7 +5,6 @@
 from django.core.files.base import ContentFile
 from catalogue.test_utils import *
 from catalogue.models import Book
-from dictionary.models import Note
 
 
 class DictionaryTests(WLTestCase):
@@ -13,19 +12,19 @@ class DictionaryTests(WLTestCase):
     def setUp(self):
         WLTestCase.setUp(self)
         self.book_info = BookInfoStub(
-            url=u"http://wolnelektury.pl/example/default-book",
-            about=u"http://wolnelektury.pl/example/URI/default_book",
-            title=u"Default Book",
             author=PersonStub(("Jim",), "Lazy"),
             kind="X-Kind",
             genre="X-Genre",
             epoch="X-Epoch",
+            **info_args(u"Default Book")
         )
 
     def test_book_with_footnote(self):
         BOOK_TEXT = """<utwor>
         <opowiadanie>
             <akap><pe><slowo_obce>rose</slowo_obce> --- kind of a flower.</pe></akap>
+            <akap><pe><slowo_obce>rose</slowo_obce> --- kind of a flower.</pe></akap>
+            <akap><pe><slowo_obce>rose</slowo_obce> (color) --- #FF007F.</pe></akap>
         </opowiadanie></utwor>
         """
 
@@ -33,16 +32,21 @@ class DictionaryTests(WLTestCase):
 
         self.assertEqual(
             len(self.client.get('/przypisy/').context['object_list']),
-            1,
+            2,
             'There should be a note on the note list.')
 
         self.assertEqual(
-            len(self.client.get('/przypisy/a/').context['object_list']),
+            len(self.client.get('/przypisy/?ltr=r').context['object_list']),
             0,
             'There should not be a note for the letter A.')
 
         self.assertEqual(
-            len(self.client.get('/przypisy/r/').context['object_list']),
-            1,
+            len(self.client.get('/przypisy/?ltr=r').context['object_list']),
+            2,
+            'There should be a note for the letter R.')
+
+        self.assertEqual(
+            len(self.client.get('/przypisy/?qual=color').context['object_list']),
+            2,
             'There should be a note for the letter R.')