mamweb/personalni/tests.py

from django.test import TestCase

from django.contrib.auth.models import User, Group
# 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

import logging
logger = logging.getLogger(__name__)

class DelaniOrguTest(TestCase):
	def setUp(self):
		self.org_group = Group.objects.get(name='org')

		novy_user = User.objects.create(username='osoba')
		self.nova_osoba = m.Osoba.objects.create(
			jmeno='Milada',
			prijmeni='Von Kolej',
			user = novy_user,
			# Snad nic dalšího nepotřebujeme, kdyžtak se doplní…
			)
		stary_user = User.objects.create(username='stary_user')
		stara_osoba = m.Osoba.objects.create(user=stary_user)
		self.stary_org = m.Organizator.objects.create(osoba=stara_osoba)

	def test_pridani_orga(self):
		# Nejdřív to není org…
		self.assertFalse(m.Organizator.objects.filter(osoba=self.nova_osoba).exists())
		self.assertNotIn(self.org_group, self.nova_osoba.user.groups.all())
		self.assertFalse(self.nova_osoba.user.has_perm('auth.org'))
		self.assertFalse(self.nova_osoba.user.is_staff)
		breakpoint

		# Pak orga uděláme…
		from personalni.admin import OsobaAdmin
		qs = m.Osoba.objects.filter(id=self.nova_osoba.id)
		OsobaAdmin.udelej_orgem(None, None, qs)

		# A pak už to org má být.
		logger.info(f'Nová osoba je staff: {self.nova_osoba.user.is_staff}')
		self.assertTrue(self.nova_osoba.user.is_staff)
		self.assertTrue(self.nova_osoba.user.has_perm('auth.org'))
		self.assertIn(self.org_group, self.nova_osoba.user.groups.all())
		self.assertTrue(m.Organizator.objects.filter(osoba=self.nova_osoba).exists())
		novy_org = m.Organizator.objects.get(osoba=self.nova_osoba)
		self.assertIsNotNone(novy_org.organizuje_od)

	def test_pridani_stareho_orga(self):
		from personalni.admin import OsobaAdmin
		OsobaAdmin.udelej_orgem(None, None, 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.