[ADD] mrp_lot_number_propagation

mrp_lot_number_propagation/tests/common.py

@@ -0,0 +1,92 @@
+# Copyright 2022 Camptocamp SA
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)
+
+import random
+import string
+
+from odoo import fields
+from odoo.tests import common
+
+
+class Common(common.TransactionCase):
+
+    LOT_NAME = "PROPAGATED-LOT"
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.env = cls.env(context=dict(cls.env.context, tracking_disable=True))
+        cls.bom = cls.env.ref("mrp.mrp_bom_desk")
+        cls.product_tracked_by_lot = cls.env.ref(
+            "mrp.product_product_computer_desk_leg"
+        )
+        cls.product_tracked_by_sn = cls.env.ref(
+            "mrp.product_product_computer_desk_head"
+        )
+        cls.line_tracked_by_lot = cls.bom.bom_line_ids.filtered(
+            lambda o: o.product_id == cls.product_tracked_by_lot
+        )
+        cls.line_tracked_by_sn = cls.bom.bom_line_ids.filtered(
+            lambda o: o.product_id == cls.product_tracked_by_sn
+        )
+        cls.line_no_tracking = fields.first(
+            cls.bom.bom_line_ids.filtered(lambda o: o.product_id.tracking == "none")
+        )
+
+    @classmethod
+    def _update_qty_in_location(
+        cls, location, product, quantity, package=None, lot=None, in_date=None
+    ):
+        quants = cls.env["stock.quant"]._gather(
+            product, location, lot_id=lot, package_id=package, strict=True
+        )
+        # this method adds the quantity to the current quantity, so remove it
+        quantity -= sum(quants.mapped("quantity"))
+        cls.env["stock.quant"]._update_available_quantity(
+            product,
+            location,
+            quantity,
+            package_id=package,
+            lot_id=lot,
+            in_date=in_date,
+        )
+
+    @classmethod
+    def _update_stock_component_qty(cls, order=None, bom=None, location=None):
+        if not order and not bom:
+            return
+        if order:
+            bom = order.bom_id
+        if not location:
+            location = cls.env.ref("stock.stock_location_stock")
+        for line in bom.bom_line_ids:
+            if line.product_id.type != "product":
+                continue
+            lot = None
+            if line.product_id.tracking != "none":
+                lot_name = "".join(
+                    random.choice(string.ascii_lowercase) for i in range(10)
+                )
+                if line.propagate_lot_number:
+                    lot_name = cls.LOT_NAME
+                vals = {
+                    "product_id": line.product_id.id,
+                    "company_id": line.company_id.id,
+                    "name": lot_name,
+                }
+                lot = cls.env["stock.production.lot"].create(vals)
+            cls._update_qty_in_location(
+                location,
+                line.product_id,
+                line.product_qty,
+                lot=lot,
+            )
+
+    @classmethod
+    def _get_lot_quants(cls, lot, location=None):
+        quants = lot.quant_ids.filtered(lambda q: q.quantity > 0)
+        if location:
+            quants = quants.filtered(
+                lambda q: q.location_id.parent_path in location.parent_path
+            )
+        return quants