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 personalni.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.