Link to transport

modules/purchase_trade/tests/test_module.py

@@ -223,7 +223,7 @@ class PurchaseTradeTestCase(ModuleTestCase):
         'link to transport creates shipment in from planned locations'
         wizard = lot_module.LotShipping()
         supplier = Mock(id=30)
-        created_shipment = Mock()
+        created_shipment = Mock(id=40)
         ShipmentIn = Mock()
         ShipmentIn.create.return_value = [created_shipment]
         wizard.records = [Mock(r_supplier=supplier)]
@@ -244,7 +244,7 @@ class PurchaseTradeTestCase(ModuleTestCase):
             'from_location': 10,
             'to_location': 20,
         }])
-        self.assertEqual(wizard.ship.shipment_in, created_shipment)
+        self.assertEqual(wizard.default_ship(None)['shipment_in'], 40)
 
     @with_transaction()
     def test_get_mtm_applies_component_ratio_as_percentage(self):