60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from django.test import TestCase
|
|
from tempfile import TemporaryDirectory
|
|
|
|
from library.models import Collection, Work
|
|
from library.indexer import index_works, model_search
|
|
from library.indexer import whoosh
|
|
|
|
JAZZ_STANDARDS = (
|
|
["But Not For Me", "Gershwin, George & Ira"],
|
|
["Autumn Leaves", "Kosma, Joseph"],
|
|
["Best of Gershwin", "Compilation"],
|
|
)
|
|
|
|
CLASSICAL_WORKS = (
|
|
["Symphony No.5", "Beethoven, L"],
|
|
["March from Aieda", "Verdi"],
|
|
)
|
|
|
|
|
|
class WhooshIndexTestCase(TestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
jazz = Collection.objects.create(name="Jazz Standards", prefix="jazz")
|
|
classical = Collection.objects.create(
|
|
name="Classical Music", prefix="classical"
|
|
)
|
|
|
|
for name, composer in JAZZ_STANDARDS:
|
|
jazz.works.create(name=name, composer=composer)
|
|
|
|
for name, composer in CLASSICAL_WORKS:
|
|
classical.works.create(name=name, composer=composer)
|
|
|
|
def test_setup(self):
|
|
self.assertEqual(Collection.objects.all().count(), 2)
|
|
self.assertEqual(Work.objects.all().count(), 5)
|
|
|
|
def test_indexer(self):
|
|
|
|
expected = [
|
|
("beethoven", [], ["Symph"]),
|
|
("Ira", [], ["But N"]),
|
|
("bethoven", [], ["Symph"]),
|
|
("George", [1], ["But N"]),
|
|
("George", [2], []),
|
|
("George", [1, 2], ["But N"]),
|
|
("not for me", [], ["But N"]),
|
|
("Gershwin", [], ["Best ", "But N"]),
|
|
("composer:Gershwin", [], ["But N"]),
|
|
]
|
|
|
|
with TemporaryDirectory() as d:
|
|
whoosh.index_path = d
|
|
index_works(Work.objects.all())
|
|
|
|
for query, collections, result in expected:
|
|
self.assertListEqual(
|
|
[x.name[:5] for x in model_search(query, collections)], result
|
|
)
|