mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Standardize ObjectPermissionTest

Browse Source
This commit is contained in:
Jeremy Stretch
2020-06-08 14:01:15 -04:00
parent c1a37db871
commit 62224857f0
3 changed files with 15 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/utilities/testing/api.py
View File

@@ -170,9 +170,6 @@ class APIViewTestCases:
"""
POST a single object with permission.
"""
initial_count = self.model.objects.count()
url = self._get_list_url()
# Add object-level permission
obj_perm = ObjectPermission(
actions=['add']
@@ -181,7 +178,8 @@ class APIViewTestCases:
obj_perm.users.add(self.user)
obj_perm.object_types.add(ContentType.objects.get_for_model(self.model))
response = self.client.post(url, self.create_data[0], format='json', **self.header)
initial_count = self.model.objects.count()
response = self.client.post(self._get_list_url(), self.create_data[0], format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(self.model.objects.count(), initial_count + 1)
self.assertInstanceEqual(self.model.objects.get(pk=response.data['id']), self.create_data[0], api=True)
@@ -190,9 +188,6 @@ class APIViewTestCases:
"""
POST a set of objects in a single request.
"""
initial_count = self.model.objects.count()
url = self._get_list_url()
# Add object-level permission
obj_perm = ObjectPermission(
actions=['add']
@@ -201,7 +196,8 @@ class APIViewTestCases:
obj_perm.users.add(self.user)
obj_perm.object_types.add(ContentType.objects.get_for_model(self.model))
response = self.client.post(url, self.create_data, format='json', **self.header)
initial_count = self.model.objects.count()
response = self.client.post(self._get_list_url(), self.create_data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(self.model.objects.count(), initial_count + len(self.create_data))