diff --git a/personalni/tests.py b/personalni/tests.py index 31aac8e8..4c042ba0 100644 --- a/personalni/tests.py +++ b/personalni/tests.py @@ -2,7 +2,7 @@ from django.test import TestCase, RequestFactory from django.contrib.auth.models import User, Group from django.contrib.admin.sites import AdminSite -from personalni.admin import OsobaAdmin +from personalni.admin import OsobaAdmin, udelej_orgem # Tohle bude peklo, až jednou ty modely fakt rozstřelíme… Možná vyrobit various.all_models, které půjdou importovat jako m? :-) import seminar.models as m @@ -42,7 +42,7 @@ class DelaniOrguTest(TestCase): # Pak orga uděláme… qs = m.Osoba.objects.filter(id=self.nova_osoba.id) - self.admin.udelej_orgem(self.request, qs) + udelej_orgem(self.admin, self.request, qs) # A pak už to org má být. self.nova_osoba.refresh_from_db() @@ -56,7 +56,7 @@ class DelaniOrguTest(TestCase): self.assertIsNotNone(novy_org.organizuje_od) def test_pridani_stareho_orga(self): - self.admin.udelej_orgem(self.request, m.Osoba.objects.filter(id=self.stary_org.osoba.id)) # Ugly + udelej_orgem(self.admin, self.request, m.Osoba.objects.filter(id=self.stary_org.osoba.id)) # Ugly # Když to spadne, tak jsem se to dozvěděl, takže už nepotřebuju nic kontrolovat. # Jestli to funguje správně má řešit jiný test.