def test_get_orgs_for_role(self):
self.assertEqual(
set(APIToken.get_orgs_for_role(self.admin, self.admins_group)),
{self.org})
self.assertEqual(
set(APIToken.get_orgs_for_role(self.admin, self.surveyors_group)),
{self.org, self.org2})
def form_valid(self, form, *args, **kwargs):
username = form.cleaned_data.get("email")
password = form.cleaned_data.get("password")
role_code = form.cleaned_data.get("role")
user = authenticate(username=username, password=password)
if user and user.is_active:
login(self.request, user)
role = APIToken.get_role_from_code(role_code)
orgs = []
if role:
valid_orgs = APIToken.get_orgs_for_role(user, role)
for org in valid_orgs:
token = APIToken.get_or_create(org, user, role)
orgs.append(dict(id=org.pk, name=org.name,
token=token.key))
else: # pragma: needs cover
return HttpResponse(status=403)
return JsonResponse(orgs, safe=False)
else: # pragma: needs cover
return HttpResponse(status=403)
def form_valid(self, form, *args, **kwargs):
username = form.cleaned_data.get("email")
password = form.cleaned_data.get("password")
role_code = form.cleaned_data.get("role")
user = authenticate(username=username, password=password)
if user and user.is_active:
login(self.request, user)
role = APIToken.get_role_from_code(role_code)
orgs = []
if role:
valid_orgs = APIToken.get_orgs_for_role(user, role)
for org in valid_orgs:
token = APIToken.get_or_create(org, user, role)
orgs.append(dict(id=org.pk, name=org.name, token=token.key))
else: # pragma: needs cover
return HttpResponse(status=403)
return JsonResponse(orgs, safe=False)
else: # pragma: needs cover
return HttpResponse(status=403)
def test_get_orgs_for_role(self):
self.assertEqual(set(APIToken.get_orgs_for_role(self.admin, self.admins_group)), {self.org})
self.assertEqual(set(APIToken.get_orgs_for_role(self.admin, self.surveyors_group)), {self.org, self.org2})
