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:
@ -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)
|
||||
|
Reference in New Issue
Block a user