1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
from functools import partial
from django.contrib.auth.models import User
from django.test import TestCase
from django.test.utils import override_settings
from organizations.models import Organization
from organizations.utils import create_organization
from organizations.utils import model_field_attr
from test_abstract.models import CustomOrganization
from test_accounts.models import Account
@override_settings(USE_TZ=True)
class CreateOrgTests(TestCase):
fixtures = ["users.json", "orgs.json"]
def setUp(self):
self.user = User.objects.get(username="dave")
def test_create_organization(self):
acme = create_organization(
self.user, "Acme", org_defaults={"slug": "acme-slug"}
)
self.assertTrue(isinstance(acme, Organization))
self.assertEqual(self.user, acme.owner.organization_user.user)
self.assertTrue(acme.owner.organization_user.is_admin)
def test_create_custom_org(self):
custom = create_organization(self.user, "Custom", model=Account)
self.assertTrue(isinstance(custom, Account))
self.assertEqual(self.user, custom.owner.organization_user.user)
def test_create_custom_org_from_abstract(self):
custom = create_organization(self.user, "Custom", model=CustomOrganization)
self.assertTrue(isinstance(custom, CustomOrganization))
self.assertEqual(self.user, custom.owner.organization_user.user)
def test_defaults(self):
"""Ensure models are created with defaults as specified"""
# Default models
org = create_organization(
self.user,
"Is Admin",
org_defaults={"slug": "is-admin-212", "is_active": False},
org_user_defaults={"is_admin": False},
)
self.assertFalse(org.is_active)
self.assertFalse(org.owner.organization_user.is_admin)
# Custom models
create_account = partial(
create_organization,
model=Account,
org_defaults={"monthly_subscription": 99},
org_user_defaults={"user_type": "B"},
)
myaccount = create_account(self.user, name="My New Account")
self.assertEqual(myaccount.monthly_subscription, 99)
def test_backwards_compat(self):
"""Ensure old optional arguments still work"""
org = create_organization(self.user, "Is Admin", "my-slug", is_active=False)
self.assertFalse(org.is_active)
custom = create_organization(self.user, "Custom org", org_model=Account)
self.assertTrue(isinstance(custom, Account))
class AttributeUtilTests(TestCase):
def test_present_field(self):
self.assertTrue(model_field_attr(User, "username", "max_length"))
def test_absent_field(self):
self.assertRaises(KeyError, model_field_attr, User, "blahblah", "max_length")
def test_absent_attr(self):
self.assertRaises(
AttributeError, model_field_attr, User, "username", "mariopoints"
)
|
