def test_set_role_with_profile(self): """ Set a role that takes a profile. """ set_user_role(self.user, roles.moderator, TestModeratorProfile(stars=5)) self.assertTrue(self.user.role.is_moderator) self.assertEquals(self.user.role.profile.stars, 5)
def form_valid(self, form): # import ipdb; ipdb.set_trace(); # set_user_role(self.user, roles.manager) self.user = form.save(self.request) set_user_role(self.user, roles.owner) self.token, created = self.token_model.objects.get_or_create( user=self.user ) org = Organization(owner=self.user) org.save() if isinstance(self.request, HttpRequest): request = self.request else: request = self.request._request return complete_signup(request, self.user, app_settings.EMAIL_VERIFICATION, self.get_success_url())
def addMobileUser(request): if request.method == 'POST': req = request.POST form = UserCreationForm(req) error_message = validate_user_data(req) if form.is_valid(): new_user = form.save() set_user_role(new_user, roles.mobile) mobileUser = MobileUser(user = new_user, username = new_user.username) mobileUser.save() return HttpResponseRedirect('mobileUsers') else: return render_to_response("addMobileUser.html", locals(), context_instance=RequestContext(request)) else: form = UserCreationForm(request.POST) return render_to_response("addMobileUser.html", locals(), context_instance=RequestContext(request))
def post(self, request, *args, **kwargs): self.initial = {} shop_name= self.request.data.pop('shopName') self.request.POST = self.request.data.copy() form_class = self.get_form_class() # import ipdb; ipdb.set_trace(); # set_user_role(self.user, roles.manager) self.form = self.get_form(form_class) if self.form.is_valid(): #mine self.user = self.form.save(self.request) self.user.save() set_user_role(self.user, roles.employee) org = Organization.objects.get(owner=self.request.user) shop = Shop(owner=org, employee = self.user, shopName=shop_name[0]) shop.save() org.employee.add(self.user) org.save() return self.get_response() else: return self.get_response_with_errors()
def test_set_role_without_profile(self): """ Set a role that does not take a profile. """ set_user_role(self.user, roles.client) self.assertTrue(self.user.role.is_client)
def test_view_with_throttle_role(self): url = '/view_with_throttle_role' self.assertStatus( url, THROTTLING_STATUS_CODE ) self.assertStatus( url, THROTTLING_STATUS_CODE, login=self.user ) set_user_role( self.user, roles.senior_developer ) self.assertStatus( url, 200, login=self.user, repeat=THROTTLING_NUMBER_OF_REQUESTS-2 ) set_user_role( self.user, roles.junior_developer ) self.assertStatus( url, 200, login=self.user ) set_user_role( self.user, roles.developer ) self.assertStatus( url, 200, login=self.user ) set_user_role( self.user, roles.senior_developer ) self.assertStatus( url, THROTTLING_STATUS_CODE, login=self.user ) set_user_role( self.user, roles.junior_developer ) self.assertStatus( url, THROTTLING_STATUS_CODE, login=self.user ) set_user_role( self.user, roles.developer ) self.assertStatus( url, THROTTLING_STATUS_CODE, login=self.user ) set_user_role( self.user, roles.other ) self.assertStatus( url, 200, login=self.user ) user = self.create_user('other') set_user_role( user, roles.developer ) self.assertStatus( url, 200, login=user )
def test_get_disallowed_view(self): set_user_role(self.user, roles.client) resp = self.client.get('/manager_or_moderator') self.assertEquals(resp.status_code, 302)
def setUp(self): super(UserRoleTests, self).setUp() self.user = milkman.deliver(User) set_user_role(self.user, roles.manager)
def setUp(self): super(UserRoleTests, self).setUp() self.user = UserFactory() set_user_role(self.user, roles.manager)
def SetCandidaterole(user): set_user_role(user, roles.candidate)
from userroles.models import set_user_role from userroles import roles from django.contrib.auth.models import User # user = User.objects.all()[0] set_user_role(user, roles.mturk)
def assert_it(name, status): user = self.create_user(password=name) set_user_role(user, name) client = self.create_client(user.username, name) self.assertEqual( client.get('/manager/').status_code, status )
from userroles.models import set_user_role from userroles import roles from django.contrib.auth.models import User # user = User.objects.all()[0] set_user_role(user, roles.staff)
from django.contrib.auth.models import User from webAppUser.models import WebAppUser user = User.objects.create_user('admin', '*****@*****.**', '123qwe') from userroles.models import set_user_role from userroles import roles set_user_role(user,roles.admin) web_app_user = WebAppUser(user = user) web_app_user.save()