def setUp(self): self.view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.leadership = LeadershipFactory.create( user=self.user, group=self.group ) self.staff_user = UserFactory.create(is_staff=True) self.provider = ProviderFactory.create() self.quota = QuotaFactory.create() self.identity = IdentityFactory.create( provider=self.provider, quota=self.quota, created_by=self.user) self.allocation = AllocationFactory.create() IdentityMembershipFactory.create( member=self.group, identity=self.identity, ) factory = APIRequestFactory() url = reverse('api:v2:identity-detail', args=(self.identity.id,)) self.request = factory.get(url) force_authenticate(self.request, user=self.user) self.response = self.view(self.request, pk=self.identity.id)
def setUp(self): self.list_view = ViewSet.as_view({'get': 'list'}) self.detailed_view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.staff_user = UserFactory.create(is_staff=True) self.provider = ProviderFactory.create() self.quota = QuotaFactory.create() self.identity = IdentityFactory.create(provider=self.provider, quota=self.quota, created_by=self.user) IdentityMembershipFactory.create( member=self.group, identity=self.identity, ) factory = APIRequestFactory() detail_url = reverse('api:v2:identity-detail', args=(self.identity.id, )) self.detail_request = factory.get(detail_url) list_url = reverse('api:v2:identity-list') self.list_request = factory.get(list_url)
def setUp(self): self.view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.leadership = LeadershipFactory.create( user=self.user, group=self.group ) self.staff_user = UserFactory.create(is_staff=True) self.provider = ProviderFactory.create() self.identity = IdentityFactory.create( provider=self.provider, created_by=self.user) self.quota = QuotaFactory.create() self.allocation = AllocationFactory.create() IdentityMembershipFactory.create( member=self.group, identity=self.identity, quota=self.quota ) factory = APIRequestFactory() url = reverse('api:v2:identity-detail', args=(self.identity.id,)) self.request = factory.get(url) force_authenticate(self.request, user=self.user) self.response = self.view(self.request, pk=self.identity.id)
def test_admin_update_identity_quota(self): # Build a request to patch quota for a user new_quota = QuotaFactory.create(cpu=999) url = "{base_url}?username={username}".format( base_url=reverse('api:v2:identity-detail', args=(self.identity.id, )), username=self.user.username) request = APIRequestFactory().patch(url, {'quota': { 'id': new_quota.id }}, format='json') # Make the patch request as the staff user force_authenticate(request, user=self.staff_user) view = ViewSet.as_view({'patch': 'partial_update'}) response = view(request, pk=self.identity.id) # Assert the response has new quota self.assertEqual(response.data['quota']['cpu'], new_quota.cpu) # Assert the identity has new quota updated_identity = Identity.objects.get(id=self.identity.id) self.assertEqual(updated_identity.quota.id, new_quota.id)
def setUp(self): self.list_view = ViewSet.as_view({'get': 'list'}) self.detailed_view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.staff_user = UserFactory.create(is_staff=True) self.provider = ProviderFactory.create() self.quota = QuotaFactory.create() self.identity = IdentityFactory.create( provider=self.provider, quota=self.quota, created_by=self.user ) IdentityMembershipFactory.create( member=self.group, identity=self.identity, ) factory = APIRequestFactory() detail_url = reverse( 'api:v2:identity-detail', args=(self.identity.id, ) ) self.detail_request = factory.get(detail_url) list_url = reverse('api:v2:identity-list') self.list_request = factory.get(list_url)
def test_admin_update_identity_quota(self): # Build a request to patch quota for a user new_quota = QuotaFactory.create(cpu=999) url = "{base_url}?username={username}".format( base_url=reverse( 'api:v2:identity-detail', args=(self.identity.id, ) ), username=self.user.username ) request = APIRequestFactory().patch( url, {'quota': { 'id': new_quota.id }}, format='json' ) # Make the patch request as the staff user force_authenticate(request, user=self.staff_user) view = ViewSet.as_view({'patch': 'partial_update'}) response = view(request, pk=self.identity.id) # Assert the response has new quota self.assertEqual(response.data['quota']['cpu'], new_quota.cpu) # Assert the identity has new quota updated_identity = Identity.objects.get(id=self.identity.id) self.assertEqual(updated_identity.quota.id, new_quota.id)
def setUp(self): self.view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.staff_user = UserFactory.create(is_staff=True) self.quota = QuotaFactory.create() factory = APIRequestFactory() url = reverse('api:v2:quota-detail', args=(self.quota.id,)) self.request = factory.get(url) force_authenticate(self.request, user=self.user) self.response = self.view(self.request, pk=self.quota.id)
def setUp(self): self.view = ViewSet.as_view({'get': 'retrieve'}) self.anonymous_user = AnonymousUserFactory() self.user = UserFactory.create() self.group = GroupFactory.create(name=self.user.username) self.staff_user = UserFactory.create(is_staff=True) self.quota = QuotaFactory.create() factory = APIRequestFactory() url = reverse('api:v2:quota-detail', args=(self.quota.id, )) self.request = factory.get(url) force_authenticate(self.request, user=self.user) self.response = self.view(self.request, pk=self.quota.id)