def test_pkg_edit_no_shipment(self): # Provide a package to edit that has a shipment package = PackageFactory() data = { 'name': package.name, 'description': 'New description', } form = PackageEditForm(instance=package, data=data) self.assertTrue(form.is_valid(), msg=form.errors) pkg = form.save() self.assertEqual(data['description'], pkg.description)
def test_pkg_edit_specify_shipment(self): # Creating a new instance, specify shipment shipment = ShipmentFactory() data = { 'name': 'New name', 'description': 'New description', } form = PackageEditForm(data=data, shipment=shipment) pkg = form.save() self.assertEqual(shipment, pkg.shipment) self.assertEqual(data['name'], pkg.name) self.assertEqual(data['description'], pkg.description)