Pavel "LEdoian" Turinsky
2 years ago
1 changed files with 53 additions and 0 deletions
@ -0,0 +1,53 @@ |
|||
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. |
|||
|
|||
|
Loading…
Reference in new issue