diff --git a/stock_move_location/tests/test_common.py b/stock_move_location/tests/test_common.py index 61a1865e7..ff02a388f 100644 --- a/stock_move_location/tests/test_common.py +++ b/stock_move_location/tests/test_common.py @@ -15,6 +15,7 @@ class TestsCommon(common.TransactionCase): cls.wizard_obj = cls.env["wiz.stock.move.location"] cls.quant_obj = cls.env["stock.quant"] cls.company = cls.env.ref("base.main_company") + cls.partner = cls.env.ref("base.res_partner_category_0") cls.internal_loc_1 = cls.location_obj.create( { @@ -79,7 +80,15 @@ class TestsCommon(common.TransactionCase): "company_id": cls.company.id, } ) - cls.package = cls.env["stock.quant.package"].create({}) + cls.lot5 = cls.env["stock.production.lot"].create( + { + "name": "lot5", + "product_id": cls.product_package.id, + "company_id": cls.company.id, + } + ) + cls.package1 = cls.env["stock.quant.package"].create({}) + cls.package2 = cls.env["stock.quant.package"].create({}) def setup_product_amounts(self): self.set_product_amount(self.product_no_lots, self.internal_loc_1, 123) @@ -97,22 +106,39 @@ class TestsCommon(common.TransactionCase): self.internal_loc_1, 1.0, lot_id=self.lot4, - package_id=self.package, + package_id=self.package1, + ) + self.set_product_amount( + self.product_package, + self.internal_loc_1, + 1.0, + lot_id=self.lot5, + package_id=self.package2, + owner_id=self.partner, ) def set_product_amount( - self, product, location, amount, lot_id=None, package_id=None + self, product, location, amount, lot_id=None, package_id=None, owner_id=None ): self.env["stock.quant"]._update_available_quantity( - product, location, amount, lot_id=lot_id, package_id=package_id + product, + location, + amount, + lot_id=lot_id, + package_id=package_id, + owner_id=owner_id, ) def check_product_amount( - self, product, location, amount, lot_id=None, package_id=None + self, product, location, amount, lot_id=None, package_id=None, owner_id=None ): self.assertEqual( self.env["stock.quant"]._get_available_quantity( - product, location, lot_id=lot_id, package_id=package_id + product, + location, + lot_id=lot_id, + package_id=package_id, + owner_id=owner_id, ), amount, ) diff --git a/stock_move_location/tests/test_move_location.py b/stock_move_location/tests/test_move_location.py index f6631ce5c..9c29eb9de 100644 --- a/stock_move_location/tests/test_move_location.py +++ b/stock_move_location/tests/test_move_location.py @@ -22,14 +22,30 @@ class TestMoveLocation(TestsCommon): self.check_product_amount(self.product_lots, self.internal_loc_1, 0, self.lot2) self.check_product_amount(self.product_lots, self.internal_loc_1, 0, self.lot3) self.check_product_amount( - self.product_package, self.internal_loc_1, 0, self.lot4, self.package + self.product_package, self.internal_loc_1, 0, self.lot4, self.package1 + ) + self.check_product_amount( + self.product_package, + self.internal_loc_1, + 0, + self.lot5, + self.package2, + self.partner, ) self.check_product_amount(self.product_no_lots, self.internal_loc_2, 123) self.check_product_amount(self.product_lots, self.internal_loc_2, 1, self.lot1) self.check_product_amount(self.product_lots, self.internal_loc_2, 1, self.lot2) self.check_product_amount(self.product_lots, self.internal_loc_2, 1, self.lot3) self.check_product_amount( - self.product_package, self.internal_loc_2, 1, self.lot4, self.package + self.product_package, self.internal_loc_2, 1, self.lot4, self.package1 + ) + self.check_product_amount( + self.product_package, + self.internal_loc_2, + 1, + self.lot5, + self.package2, + self.partner, ) def test_move_location_wizard_amount(self): @@ -60,9 +76,9 @@ class TestMoveLocation(TestsCommon): """Test lines getting cleared properly.""" wizard = self._create_wizard(self.internal_loc_1, self.internal_loc_2) wizard.onchange_origin_location() - self.assertEqual(len(wizard.stock_move_location_line_ids), 5) + self.assertEqual(len(wizard.stock_move_location_line_ids), 6) wizard._onchange_destination_location_id() - self.assertEqual(len(wizard.stock_move_location_line_ids), 5) + self.assertEqual(len(wizard.stock_move_location_line_ids), 6) dest_location_line = wizard.stock_move_location_line_ids.mapped( "destination_location_id" ) @@ -102,9 +118,10 @@ class TestMoveLocation(TestsCommon): wizard.action_move_location() picking = wizard.picking_id self.assertEqual(picking.state, "assigned") - self.assertEqual(len(picking.move_line_ids), 5) + self.assertEqual(len(picking.move_line_ids), 6) self.assertEqual( - sorted(picking.move_line_ids.mapped("product_uom_qty")), [1, 1, 1, 1, 123] + sorted(picking.move_line_ids.mapped("product_uom_qty")), + [1, 1, 1, 1, 1, 123], ) def test_quant_transfer(self): diff --git a/stock_move_location/wizard/stock_move_location.py b/stock_move_location/wizard/stock_move_location.py index 9ff3dd845..221aab144 100644 --- a/stock_move_location/wizard/stock_move_location.py +++ b/stock_move_location/wizard/stock_move_location.py @@ -104,6 +104,7 @@ class StockMoveLocationWizard(models.TransientModel): "origin_location_id": quant.location_id.id, "lot_id": quant.lot_id.id, "package_id": quant.package_id.id, + "owner_id": quant.owner_id.id, "product_uom_id": quant.product_uom_id.id, "custom": False, }, @@ -131,6 +132,7 @@ class StockMoveLocationWizard(models.TransientModel): "origin_location_id": quant.location_id.id, "lot_id": quant.lot_id.id, "package_id": quant.package_id.id, + "owner_id": quant.owner_id.id, "product_uom_id": quant.product_uom_id.id, "custom": False, }, @@ -233,6 +235,7 @@ class StockMoveLocationWizard(models.TransientModel): ("location_id", "=", line.origin_location_id.id), ("lot_id", "=", line.lot_id.id), ("package_id", "=", line.package_id.id), + ("owner_id", "=", line.owner_id.id), ("product_uom_qty", ">", 0.0), ] ) @@ -274,11 +277,11 @@ class StockMoveLocationWizard(models.TransientModel): # Using sql as search_group doesn't support aggregation functions # leading to overhead in queries to DB query = """ - SELECT product_id, lot_id, package_id, SUM(quantity) AS quantity, + SELECT product_id, lot_id, package_id, owner_id, SUM(quantity) AS quantity, SUM(reserved_quantity) AS reserved_quantity FROM stock_quant WHERE location_id = %s - GROUP BY product_id, lot_id, package_id + GROUP BY product_id, lot_id, package_id, owner_id """ self.env.cr.execute(query, (location_id.id,)) return self.env.cr.dictfetchall() @@ -305,6 +308,7 @@ class StockMoveLocationWizard(models.TransientModel): # cursor returns None instead of False "lot_id": group.get("lot_id") or False, "package_id": group.get("package_id") or False, + "owner_id": group.get("owner_id") or False, "product_uom_id": product.uom_id.id, "custom": False, } diff --git a/stock_move_location/wizard/stock_move_location.xml b/stock_move_location/wizard/stock_move_location.xml index f294582fa..766ad1b13 100644 --- a/stock_move_location/wizard/stock_move_location.xml +++ b/stock_move_location/wizard/stock_move_location.xml @@ -80,6 +80,7 @@ domain="[('location_id', '=', origin_location_id)]" options="{'no_create': True}" /> +