2
0
mirror of https://github.com/inventree/InvenTree.git synced 2025-06-17 12:35:46 +00:00

More test refactoring

This commit is contained in:
Oliver Walters
2021-02-26 22:08:06 +11:00
parent d76b873c00
commit 5a536be22d
3 changed files with 37 additions and 44 deletions

View File

@ -10,10 +10,12 @@ from rest_framework import status
from django.urls import reverse
from django.contrib.auth import get_user_model
from InvenTree.api_tester import InvenTreeAPITestCase
from .models import PurchaseOrder, SalesOrder
class OrderTest(APITestCase):
class OrderTest(InvenTreeAPITestCase):
fixtures = [
'category',
@ -26,25 +28,20 @@ class OrderTest(APITestCase):
'sales_order',
]
roles = [
'purchase_order.change',
'sales_order.change',
]
def setUp(self):
# Create a user for auth
get_user_model().objects.create_user('testuser', 'test@testing.com', 'password')
self.client.login(username='testuser', password='password')
def doGet(self, url, data={}):
return self.client.get(url, data=data, format='json')
def doPost(self, url, data={}):
return self.client.post(url, data=data, format='json')
super().setUp()
def filter(self, filters, count):
"""
Test API filters
"""
response = self.doGet(
response = self.get(
self.LIST_URL,
filters
)
@ -98,7 +95,7 @@ class PurchaseOrderTest(OrderTest):
url = '/api/order/po/1/'
response = self.doGet(url)
response = self.get(url)
self.assertEqual(response.status_code, 200)
@ -111,7 +108,7 @@ class PurchaseOrderTest(OrderTest):
url = reverse('api-po-attachment-list')
response = self.doGet(url)
response = self.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)
@ -161,7 +158,7 @@ class SalesOrderTest(OrderTest):
url = '/api/order/so/1/'
response = self.doGet(url)
response = self.get(url)
self.assertEqual(response.status_code, 200)
@ -173,6 +170,6 @@ class SalesOrderTest(OrderTest):
url = reverse('api-so-attachment-list')
response = self.doGet(url)
response = self.get(url)
self.assertEqual(response.status_code, status.HTTP_200_OK)