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)
