mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
5ba7aeaa27
- Use DRF ImageField, not FileField - Ensure that permissions get updated correctly in 'test' mode - Allow file upload in the APITester class
123 lines
3.0 KiB
Python
123 lines
3.0 KiB
Python
"""
|
|
Helper functions for performing API unit tests
|
|
"""
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.auth.models import Group
|
|
from rest_framework.test import APITestCase
|
|
|
|
|
|
class InvenTreeAPITestCase(APITestCase):
|
|
"""
|
|
Base class for running InvenTree API tests
|
|
"""
|
|
|
|
# User information
|
|
username = 'testuser'
|
|
password = 'mypassword'
|
|
email = 'test@testing.com'
|
|
|
|
superuser = False
|
|
auto_login = True
|
|
|
|
# Set list of roles automatically associated with the user
|
|
roles = []
|
|
|
|
def setUp(self):
|
|
|
|
super().setUp()
|
|
|
|
# Create a user to log in with
|
|
self.user = get_user_model().objects.create_user(
|
|
username=self.username,
|
|
password=self.password,
|
|
email=self.email
|
|
)
|
|
|
|
# Create a group for the user
|
|
self.group = Group.objects.create(name='my_test_group')
|
|
self.user.groups.add(self.group)
|
|
|
|
if self.superuser:
|
|
self.user.is_superuser = True
|
|
self.user.save()
|
|
|
|
for role in self.roles:
|
|
self.assignRole(role)
|
|
|
|
if self.auto_login:
|
|
self.client.login(username=self.username, password=self.password)
|
|
|
|
def assignRole(self, role):
|
|
"""
|
|
Set the user roles for the registered user
|
|
"""
|
|
|
|
# role is of the format 'rule.permission' e.g. 'part.add'
|
|
|
|
rule, perm = role.split('.')
|
|
|
|
for ruleset in self.group.rule_sets.all():
|
|
|
|
if ruleset.name == rule:
|
|
|
|
if perm == 'view':
|
|
ruleset.can_view = True
|
|
elif perm == 'change':
|
|
ruleset.can_change = True
|
|
elif perm == 'delete':
|
|
ruleset.can_delete = True
|
|
elif perm == 'add':
|
|
ruleset.can_add = True
|
|
|
|
ruleset.save()
|
|
break
|
|
|
|
def get(self, url, data={}, expected_code=200):
|
|
"""
|
|
Issue a GET request
|
|
"""
|
|
|
|
response = self.client.get(url, data, format='json')
|
|
|
|
if expected_code is not None:
|
|
self.assertEqual(response.status_code, expected_code)
|
|
|
|
return response
|
|
|
|
def post(self, url, data, expected_code=None):
|
|
"""
|
|
Issue a POST request
|
|
"""
|
|
|
|
response = self.client.post(url, data=data, format='json')
|
|
|
|
if expected_code is not None:
|
|
self.assertEqual(response.status_code, expected_code)
|
|
|
|
return response
|
|
|
|
def delete(self, url, expected_code=None):
|
|
"""
|
|
Issue a DELETE request
|
|
"""
|
|
|
|
response = self.client.delete(url)
|
|
|
|
if expected_code is not None:
|
|
self.assertEqual(response.status_code, expected_code)
|
|
|
|
return response
|
|
|
|
def patch(self, url, data, files=None, expected_code=None):
|
|
"""
|
|
Issue a PATCH request
|
|
"""
|
|
|
|
response = self.client.patch(url, data=data, files=files, format='json')
|
|
|
|
if expected_code is not None:
|
|
self.assertEqual(response.status_code, expected_code)
|
|
|
|
return response
|