Added API test for Company app

InvenTree/company/test_api.py

@@ -0,0 +1,61 @@
+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 .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()
+        User.objects.create_user('testuser', 'test@testing.com', 'password')
+        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)