Tests for POLineItem creation form

InvenTree/order/test_views.py

@@ -6,7 +6,7 @@ from django.contrib.auth import get_user_model
 
 from InvenTree.status_codes import OrderStatus
 
-from .models import PurchaseOrder
+from .models import PurchaseOrder, PurchaseOrderLineItem
 
 import json
 
@@ -107,3 +107,57 @@ class POTests(OrderViewTestCase):
         # Test that the order was actually placed
         order = PurchaseOrder.objects.get(pk=1)
         self.assertEqual(order.status, OrderStatus.PLACED)
+
+    def test_line_item_create(self):
+        """ Test the form for adding a new LineItem to a PurchaseOrder """
+
+        # Record the number of line items in the PurchaseOrder
+        po = PurchaseOrder.objects.get(pk=1)
+        n = po.lines.count()
+        self.assertEqual(po.status, OrderStatus.PENDING)
+
+        url = reverse('po-line-item-create')
+
+        # GET the form (pass the correct info)
+        response = self.client.get(url, {'order': 1}, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        
+        post_data = {
+            'part': 100,
+            'quantity': 45,
+            'reference': 'Test reference field',
+            'notes': 'Test notes field'
+        }
+
+        # POST with an invalid purchase order
+        post_data['order'] = 99
+        response = self.client.post(url, post_data, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        data = json.loads(response.content)
+        self.assertFalse(data['form_valid'])
+        self.assertIn('Invalid Purchase Order', str(data['html_form']))
+
+        # POST with a part that does not match the purchase order
+        post_data['order'] = 1
+        post_data['part'] = 3
+        response = self.client.post(url, post_data, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        data = json.loads(response.content)
+        self.assertFalse(data['form_valid'])
+        self.assertIn('must match for Part and Order', str(data['html_form']))
+
+        # POST with an invalid part
+        post_data['part'] = 12345
+        response = self.client.post(url, post_data, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        data = json.loads(response.content)
+        self.assertFalse(data['form_valid'])
+        self.assertIn('Invalid SupplierPart selection', str(data['html_form']))
+
+        # POST the form with valid data
+        post_data['part'] = 100
+        response = self.client.post(url, post_data, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.content)
+        self.assertTrue(data['form_valid'])
+
+        self.assertEqual(n+1, PurchaseOrder.objects.get(pk=1).lines.count())
+
+        line = PurchaseOrderLineItem.objects.get(order=1, part=100)
+        self.assertEqual(line.quantity, 45)