mirror of
https://github.com/inventree/InvenTree
synced 2024-08-30 18:33:04 +00:00
73 lines
2.4 KiB
Python
73 lines
2.4 KiB
Python
from rest_framework.test import APITestCase
|
|
from rest_framework import status
|
|
from django.urls import reverse
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from InvenTree.helpers import addUserPermissions
|
|
|
|
from .models import Company
|
|
|
|
|
|
class CompanyTest(APITestCase):
|
|
"""
|
|
Series of tests for the Company DRF API
|
|
"""
|
|
|
|
def setUp(self):
|
|
# Create a user for auth
|
|
User = get_user_model()
|
|
self.user = User.objects.create_user('testuser', 'test@testing.com', 'password')
|
|
|
|
perms = [
|
|
'view_company',
|
|
'change_company',
|
|
'add_company',
|
|
]
|
|
|
|
addUserPermissions(self.user, perms)
|
|
|
|
self.client.login(username='testuser', password='password')
|
|
|
|
Company.objects.create(name='ACME', description='Supplier', is_customer=False, is_supplier=True)
|
|
Company.objects.create(name='Drippy Cup Co.', description='Customer', is_customer=True, is_supplier=False)
|
|
Company.objects.create(name='Sippy Cup Emporium', description='Another supplier')
|
|
|
|
def test_company_list(self):
|
|
url = reverse('api-company-list')
|
|
|
|
# There should be two companies
|
|
response = self.client.get(url, format='json')
|
|
self.assertEqual(len(response.data), 3)
|
|
|
|
data = {'is_customer': True}
|
|
|
|
# There should only be one customer
|
|
response = self.client.get(url, data, format='json')
|
|
self.assertEqual(len(response.data), 1)
|
|
|
|
data = {'is_supplier': True}
|
|
|
|
# There should be two suppliers
|
|
response = self.client.get(url, data, format='json')
|
|
self.assertEqual(len(response.data), 2)
|
|
|
|
def test_company_detail(self):
|
|
url = reverse('api-company-detail', kwargs={'pk': 1})
|
|
response = self.client.get(url, format='json')
|
|
|
|
self.assertEqual(response.data['name'], 'ACME')
|
|
|
|
# Change the name of the company
|
|
data = response.data
|
|
data['name'] = 'ACMOO'
|
|
response = self.client.patch(url, data, format='json')
|
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
|
self.assertEqual(response.data['name'], 'ACMOO')
|
|
|
|
def test_company_search(self):
|
|
# Test search functionality in company list
|
|
url = reverse('api-company-list')
|
|
data = {'search': 'cup'}
|
|
response = self.client.get(url, data, format='json')
|
|
self.assertEqual(len(response.data), 2)
|