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))
def form_valid(self, form):
"""
"""
user = self.backend.register_by_email(form.cleaned_data['email'])
create_organization(user=user, name=form.cleaned_data['name'],
slug=form.cleaned_data['slug'], is_active=False)
return redirect(self.get_success_url())
def form_valid(self, form):
"""
"""
user = self.backend.register_by_email(form.cleaned_data['email'])
create_organization(user=user,
name=form.cleaned_data['name'],
slug=form.cleaned_data['slug'],
is_active=False)
return redirect(self.get_success_url())
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))
def form_valid(self, form):
"""
Register user and create the organization
"""
user = self.backend.register_by_email(form.cleaned_data["email"])
create_organization(
user=user,
name=form.cleaned_data["name"],
slug=form.cleaned_data["slug"],
is_active=False,
)
return redirect(self.get_success_url())
def create_view(self, request):
"""
Initiates the organization and user account creation process
"""
if request.user.is_authenticated:
return redirect("organization_add")
form = org_registration_form(self.org_model)(request.POST or None)
if form.is_valid():
try:
user = self.user_model.objects.get(email=form.cleaned_data["email"])
except self.user_model.DoesNotExist:
user = self.user_model.objects.create(
username=self.get_username(),
email=form.cleaned_data["email"],
password=self.user_model.objects.make_random_password(),
)
user.is_active = False
user.save()
else:
return redirect("organization_add")
organization = create_organization(
user,
form.cleaned_data["name"],
form.cleaned_data["slug"],
is_active=False,
)
return render(
request,
self.activation_success_template,
{"user": user, "organization": organization},
)
return render(request, self.registration_form_template, {"form": form})
def create_view(self, request):
"""
Initiates the organization and user account creation process
"""
try:
if request.user.is_authenticated():
return redirect("organization_add")
except TypeError:
if request.user.is_authenticated:
return redirect("organization_add")
form = org_registration_form(self.org_model)(request.POST or None)
if form.is_valid():
try:
user = self.user_model.objects.get(
email=form.cleaned_data['email'])
except self.user_model.DoesNotExist:
user = self.user_model.objects.create(
username=self.get_username(),
email=form.cleaned_data['email'],
password=self.user_model.objects.make_random_password())
user.is_active = False
user.save()
else:
return redirect("organization_add")
organization = create_organization(user,
form.cleaned_data['name'],
form.cleaned_data['slug'],
is_active=False)
return render(request, self.activation_success_template, {
'user': user,
'organization': organization
})
return render(request, self.registration_form_template, {'form': form})
def save(self, **kwargs):
image = self.request.POST.get("image_field")
is_active = True
user = self.request.user
return create_organization(
user, self.cleaned_data["name"], self.cleaned_data["description"], image, is_active=is_active
)
def create_view(self, request):
"""
Initiates the organization and user account creation process
"""
if callable(request.user.is_authenticated):
is_authenticated = request.user.is_authenticated()
else:
is_authenticated = request.user.is_authenticated
if is_authenticated:
return redirect("organization_add")
form = OrganizationRegistrationForm(request.POST or None)
if form.is_valid():
try:
user = self.user_model.objects.get(email=form.cleaned_data['email'])
except self.user_model.DoesNotExist:
user = self.user_model.objects.create(username=self.get_username(),
email=form.cleaned_data['email'],
password=self.user_model.objects.make_random_password())
user.is_active = False
user.save()
else:
return redirect("organization_add")
organization = create_organization(user, form.cleaned_data['name'],
form.cleaned_data['slug'], is_active=False)
return render(request, 'organizations/register_success.html',
{'user': user, 'organization': organization})
return render(request, 'organizations/register_form.html',
{'form': form})
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 create_view(self, request):
"""
Initiates the organization and user account creation process
"""
if request.user.is_authenticated():
return redirect("organization_add")
form = OrganizationRegistrationForm(request.POST or None)
if form.is_valid():
try:
user = self.user_model.objects.get(email=form.cleaned_data["email"])
except self.user_model.DoesNotExist:
user = self.user_model.objects.create(
username=self.get_username(),
email=form.cleaned_data["email"],
password=self.user_model.objects.make_random_password(),
)
user.is_active = False
user.save()
else:
return redirect("organization_add")
organization = create_organization(
user, form.cleaned_data["name"], form.cleaned_data["slug"], is_active=False
)
return render(request, "organizations/register_success.html", {"user": user, "organization": organization})
return render(request, "organizations/register_form.html", {"form": form})
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 setUp(self):
OrganizationFactory.create_batch(NO_ORG)
# WorkflowFactory.create_batch(NO_WF)
valid_user_name = "valid"
valid_user_password = "******"
self.valid_user = User.objects.create_user(
username=valid_user_name, email='*****@*****.**', password=valid_user_password
)
self.valid_user.save()
self.organization = create_organization(self.valid_user, 'org', 'org')
self.valid_client = Client()
self.assertTrue(
self.valid_client.login(username=valid_user_name, password=valid_user_password),
'Logging in user "{}" failed.'.format(valid_user_name)
)
for i in range(0, 9):
Organization.objects.create(name="org_{}".format(i))
# WorkflowFactory.create_batch(1)
WorkflowLogFactory.create_batch(NO_WFL)
def save(self, **kwargs):
"""
Create the organization, then get the user, then make the owner.
"""
is_active = True
try:
user = get_user_model().objects.get(email=self.cleaned_data["email"])
except get_user_model().DoesNotExist:
# TODO(bennylope): look into hooks for alt. registration systems here
user = invitation_backend().invite_by_email(
self.cleaned_data["email"],
**{
"domain": get_current_site(self.request),
"organization": self.cleaned_data["name"],
"sender": self.request.user,
"created": True,
}
)
is_active = False
return create_organization(
user,
self.cleaned_data["name"],
self.cleaned_data["slug"],
is_active=is_active,
)
def save(self, **kwargs):
"""
Create the organization, then get the user, then make the owner.
"""
user = self.request.user
return create_organization(user,
self.cleaned_data['name'],
self.cleaned_data['slug'],
is_active=True)
def save(self, **kwargs):
is_active = True
user = self.request.user
organization = create_organization(user,
self.cleaned_data['name'],
'', # Empty slug to be autofilled
is_active=is_active)
assign_perm('edit_decisions_feedback',
organization.owner.organization_user.user,
organization)
return organization
def save(self, **kwargs):
is_active = True
user = self.request.user
organization = create_organization(user,
self.cleaned_data['name'],
'', # Empty slug to be autofilled
is_active=is_active)
assign('edit_decisions_feedback',
organization.owner.organization_user.user,
organization)
return organization
def setUp(self):
User = get_user_model()
self.mw = OrganizationsMiddleware()
self.factory = RequestFactory()
self.unknown_user = User.objects.create_user(
username='******', email='*****@*****.**', password='******'
)
self.known_user = User.objects.create_user(
username='******', email='*****@*****.**', password='******'
)
self.org = create_organization(
self.known_user, 'Test', 'test', is_active=True
)
def save(self):
"""
Create the organization, then get the user, then make the owner.
"""
try:
user = User.objects.get(email=self.cleaned_data['email'])
except User.DoesNotExist:
user = invitation_backend().create_invitation(
self.cleaned_data['email'],
**{'domain': get_current_site(self.request),
'organization': self.cleaned_data['name'],
'sender': self.request.user, 'created': True})
return create_organization(self.cleaned_data['name'], 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 save(self, **kwargs):
"""
Create the organization, then get the user, then make the owner.
"""
is_active = True
try:
user = get_user_model().objects.get(email=self.cleaned_data['email'])
except get_user_model().DoesNotExist:
user = invitation_backend().invite_by_email(
self.cleaned_data['email'],
**{'domain': get_current_site(self.request),
'organization': self.cleaned_data['name'],
'sender': self.request.user, 'created': True})
is_active = False
return create_organization(user, self.cleaned_data['name'],
self.cleaned_data['slug'], is_active=is_active)
def save(self, **kwargs):
"""
Create the organization, then get the user, then make the owner.
"""
is_active = True
try:
user = get_user_model().objects.get(email=self.cleaned_data['email'])
except get_user_model().DoesNotExist:
user = invitation_backend().invite_by_email(
self.cleaned_data['email'],
**{'domain': get_current_site(self.request),
'organization': self.cleaned_data['name'],
'sender': self.request.user, 'created': True})
is_active = False
return create_organization(user, self.cleaned_data['name'],
self.cleaned_data['slug'], is_active=is_active)
def save(self, **kwargs):
"""
Create the organization, then get the user, then make the owner.
"""
is_active = True
try:
user = get_user_model().objects.get(email=self.cleaned_data["email"])
except get_user_model().DoesNotExist:
user = invitation_backend().invite_by_email(
self.cleaned_data["email"],
**{
"domain": get_current_site(self.request),
"organization": self.cleaned_data["name"],
"sender": self.request.user,
"created": True,
}
)
is_active = False
return create_organization(user, self.cleaned_data["name"], self.cleaned_data["slug"], is_active=is_active)
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 create_view(self, request):
"""
Initiates the organization and user account creation process
"""
if callable(request.user.is_authenticated):
is_authenticated = request.user.is_authenticated()
else:
is_authenticated = request.user.is_authenticated
if is_authenticated:
return redirect("organization_add")
form = OrganizationRegistrationForm(request.POST or None)
if form.is_valid():
try:
user = self.user_model.objects.get(
email=form.cleaned_data['email'])
except self.user_model.DoesNotExist:
user = self.user_model.objects.create(
username=self.get_username(),
email=form.cleaned_data['email'],
password=self.user_model.objects.make_random_password())
user.is_active = False
user.save()
else:
return redirect("organization_add")
organization = create_organization(user,
form.cleaned_data['name'],
form.cleaned_data['slug'],
is_active=False)
return render(request, 'organizations/register_success.html', {
'user': user,
'organization': organization
})
return render(request, 'organizations/register_form.html',
{'form': form})
def account_account(account_user):
vendor = create_organization(account_user, "Acme", org_model=Account)
yield vendor
def save(self, **kwargs):
image = self.request.POST.get('image_field')
is_active = True
user = self.request.user
return create_organization(user, self.cleaned_data['name'], self.cleaned_data['description'], image, is_active=is_active)
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 org_organization(account_user):
yield create_organization(account_user, "Acme", org_model=Organization)
def account_account(account_user):
vendor = create_organization(account_user, "Acme", org_model=Account)
yield vendor
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_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_organization(self):
user = User.objects.get(username="******")
acme = create_organization(user, "Acme", "acme")
self.assertTrue(isinstance(acme, Organization))
self.assertEqual(user, acme.owner.organization_user.user)
def account_account(account_user):
yield create_organization(account_user, "Acme", org_model=Account)
def test_create_organization(self):
user = User.objects.get(username="******")
acme = create_organization(user, "Acme", "acme")
self.assertTrue(isinstance(acme, Organization))
self.assertEqual(user, acme.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)
