def test_finalize(self):
shipment = ShipmentFactory()
pkg2 = PackageFactory(shipment=shipment, name='pkg2')
pkg1 = PackageFactory(shipment=shipment, name='pkg1')
self.assertEqual(pkg1.status, Shipment.STATUS_IN_PROGRESS)
shipment.finalize()
pkg1 = Package.objects.get(pk=pkg1.pk)
pkg2 = Package.objects.get(pk=pkg2.pk)
self.assertEqual(pkg1.status, Shipment.STATUS_READY)
self.assertEqual(pkg2.number_in_shipment, 1) # in order of creation
self.assertEqual(pkg1.number_in_shipment, 2)
shipment = Shipment.objects.get(pk=shipment.pk)
self.assertEqual(shipment.status, Shipment.STATUS_READY)
shipment.reopen()
pkg1 = Package.objects.get(pk=pkg1.pk)
self.assertEqual(pkg1.status, Shipment.STATUS_IN_PROGRESS)
