64 lines
2.2 KiB

from django.test import TestCase
from django.urls import reverse
import seminar.models as m
import seminar.views as v
from seminar.utils import sync_skoly
class OrgSkolyAutocompleteTestCase(TestCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
sync_skoly('https://mam.mff.cuni.cz/')
# Správné školy podle toho, co orgové poslali: (prefix, ID školy)
# NOTE: Pozor, jedná se o databázové indexy. Pokud se to někdy rozbije, bude potřeba je přepsat nebo předělat na IZO
cls.spravna_data = [
('gymnázium kolín', 53),
('kolín', 53),
('gasoš', 96),
('Rokycany', 96),
('gasoš Rokycany', 96),
('SPŠE Pardubice', 815),
('Jaroše', 164),
("Gymnázium, Brno, tř. Kpt. Jaroše", 164),
("Jírovcova", 157),
('České Budějovice', 157),
("Gymnázium, České Budějovice, Jírovcova 8", 157),
("první soukromé", 2),
("Gymnázium Elgartova", 147),
("Jihlava", 45),
('Milevsko', 223),
("Gymnázium Milevsko", 223),
("Gymnázium Žamberk", 257),
("Žamberk", 257),
("Mendelovo", 201),
("Omská", 1038),
("Omsk", 1038),
("machar", 15),
("J. S. Machar", 15),
("Brandýs", 15),
]
def test_view_funguje(self):
"""Jen se pokusí udělat na ten view dotaz a kouká na odpověď"""
resp = self.client.get(reverse('autocomplete_skola'))
self.assertEqual(resp.status_code, 200)
def test_skoly_vraceny(self):
"""Testuje, že pro každého orga je jeho škola ve výsledném QuerySetu"""
for pfx, id in self.spravna_data:
with self.subTest(prefix=pfx, spravne_id=id):
spravna_skola = m.Skola.objects.get(id=id)
# Zeptáme se view, co si myslí
resp = self.client.get(reverse('autocomplete_skola')+'?q='+pfx).json()
ids = [int(x['id']) for x in resp['results']]
self.assertIn(spravna_skola.id, ids, f"Škola nenalezena v odpovědi")
def test_skoly_pocet(self):
"""Testuje, že se pro dané prefixy nevrací moc škol"""
limit = 20
for pfx, id in self.spravna_data:
# Kopírování …
with self.subTest(prefix=pfx, spravne_id=id):
resp = self.client.get(reverse('autocomplete_skola')+'?q='+pfx).json()
self.assertLessEqual(len(resp['results']), limit, f"Dotaz vrací moc škol")