From 2bbad1d38717b63925b9f8a0c39a6bbf6362726c Mon Sep 17 00:00:00 2001
From: Matthias <matthias.mair@oewf.org>
Date: Mon, 28 Mar 2022 23:52:26 +0200
Subject: [PATCH] add unit test for migration

---
 InvenTree/order/test_migrations.py | 78 ++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)

diff --git a/InvenTree/order/test_migrations.py b/InvenTree/order/test_migrations.py
index 3afba65223..5516a733c5 100644
--- a/InvenTree/order/test_migrations.py
+++ b/InvenTree/order/test_migrations.py
@@ -122,3 +122,81 @@ class TestShipmentMigration(MigratorTestCase):
         # Check that the correct number of Shipments have been created
         self.assertEqual(SalesOrder.objects.count(), 5)
         self.assertEqual(Shipment.objects.count(), 5)
+
+
+class TestAdditionalLineMigration(MigratorTestCase):
+    """
+    Test entire schema migration
+    """
+
+    migrate_from = ('order', '0063_alter_purchaseorderlineitem_unique_together')
+    migrate_to = ('order', '0064_purchaseorderextraline_salesorderextraline')
+
+    def prepare(self):
+        """
+        Create initial data set
+        """
+
+        # Create a purchase order from a supplier
+        Company = self.old_state.apps.get_model('company', 'company')
+        PurchaseOrder = self.old_state.apps.get_model('order', 'purchaseorder')
+        Part = self.old_state.apps.get_model('part', 'part')
+        Supplierpart = self.old_state.apps.get_model('company', 'supplierpart')
+
+        supplier = Company.objects.create(
+            name='Supplier A',
+            description='A great supplier!',
+            is_supplier=True,
+            is_customer=True,
+        )
+
+        part = Part.objects.create(
+            name='Bob',
+            description='Can we build it?',
+            assembly=True,
+            salable=True,
+            purchaseable=False,
+            tree_id=0,
+            level=0,
+            lft=0,
+            rght=0,
+        )
+        supplierpart = Supplierpart.objects.create(
+            part=part,
+            supplier=supplier
+        )
+
+        # Create some orders
+        for ii in range(10):
+
+            order = PurchaseOrder.objects.create(
+                supplier=supplier,
+                reference=f"{ii}-abcde",
+                description="Just a test order"
+            )
+            order.lines.create(
+                part=supplierpart,
+                quantity=12,
+                received=1
+            )
+            order.lines.create(
+                quantity=12,
+                received=1
+            )
+
+
+    def test_ref_field(self):
+        """
+        Test that the 'reference_int' field has been created and is filled out correctly
+        """
+
+        PurchaseOrder = self.new_state.apps.get_model('order', 'purchaseorder')
+
+        for ii in range(10):
+
+            po = PurchaseOrder.objects.get(reference=f"{ii}-abcde")
+
+            # The integer reference field must have been correctly updated
+            self.assertEqual(po.extra_lines.count(), 1)
+            self.assertEqual(po.lines.count(), 1)
+