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 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 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 )
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)
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)
def setUp(self):
super(UserRoleTests, self).setUp()
self.user = UserFactory()
set_user_role(self.user, roles.manager)
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()
