[ADD] stock_archive_constraint: Allows to block archiving products and locations with associated stock.quant or stock.move

stock_archive_constraint/tests/test_location_archive_constraint.py

@@ -0,0 +1,178 @@
+# Copyright 2020 Tecnativa - Víctor Martínez
+# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html).
+
+from odoo.tests.common import SavepointCase, Form
+from odoo.exceptions import ValidationError
+
+
+class TestLocationArchiveConstraint(SavepointCase):
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+        cls.product_1 = cls._create_product(cls, 'Product 1')
+        cls.product_2 = cls._create_product(cls, 'Product 2')
+        stock_location_stock = cls.env.ref('stock.stock_location_stock')
+        cls.stock_location = cls._create_stock_location(
+            cls, "%s (Copy)" % (stock_location_stock.name)
+        )
+        cls.stock_location_child = cls._create_stock_location(
+            cls, "%s (Child)" % (cls.stock_location.name)
+        )
+        cls.stock_location_child.location_id = cls.stock_location
+
+    def _create_product(self, name):
+        product_form = Form(self.env['product.product'])
+        product_form.name = name
+        product_form.type = 'product'
+        return product_form.save()
+
+    def _create_stock_location(self, name):
+        stock_location_form = Form(self.env['stock.location'])
+        stock_location_form.name = name
+        stock_location_form.usage = self.env.ref('stock.stock_location_stock').usage
+        return stock_location_form.save()
+
+    def _create_stock_inventory(self, location_id, product_id, qty):
+        stock_inventory_form = Form(self.env['stock.inventory'])
+        stock_inventory_form.name = 'INV: %s' % product_id.display_name
+        stock_inventory_form.filter = 'product'
+        stock_inventory_form.product_id = product_id
+        stock_inventory_form.location_id = location_id
+        stock_inventory = stock_inventory_form.save()
+        stock_inventory.action_start()
+        for line_id in stock_inventory.line_ids:
+            line_id.product_qty = qty
+        stock_inventory.action_validate()
+
+    def _create_stock_move(self, location_id, location_dest_id, product_id, qty):
+        stock_move_form = Form(self.env['stock.move'])
+        stock_move_form.name = product_id.display_name
+        stock_move_form.location_id = location_id
+        stock_move_form.location_dest_id = location_dest_id
+        stock_move_form.product_id = product_id
+        stock_move_form.product_uom_qty = qty
+        stock_move = stock_move_form.save()
+        stock_move._action_done()
+
+    def _create_stock_move_line(self, location_id, location_dest_id, product_id, qty):
+        stock_move_line_form = Form(self.env['stock.move.line'])
+        stock_move_line_form.location_id = location_id
+        stock_move_line_form.location_dest_id = location_dest_id
+        stock_move_line_form.product_id = product_id
+        stock_move_line_form.product_uom_qty = qty
+        stock_move_line_form.qty_done = qty
+        stock_move_line_form.state = 'done'
+        stock_move_line_form.save()
+
+    def _create_stock_picking(self, location_id, location_dest_id, product_id, qty):
+        stock_picking_form = Form(self.env['stock.picking'])
+        stock_picking_form.picking_type_id = self.env.ref('stock.picking_type_in')
+        with stock_picking_form.move_ids_without_package.new() as line:
+            line.product_id = product_id
+            line.product_uom_qty = qty
+        stock_picking = stock_picking_form.save()
+        stock_picking.write({
+            'location_id': location_id.id,
+            'location_dest_id': location_dest_id.id,
+        })
+        stock_picking.action_confirm()
+        for line in stock_picking.move_ids_without_package:
+            line.quantity_done = line.product_uom_qty
+        stock_picking.button_validate()
+
+    def test_archive_product_ok(self):
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        self.product_2.active = False
+        self.assertFalse(self.product_2.active)
+
+    def test_archive_unarchive_product(self):
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        self.product_1.active = True
+        self.assertTrue(self.product_1.active)
+
+    def test_archive_product_with_stock_move_in(self):
+        self._create_stock_move(
+            self.env.ref('stock.stock_location_suppliers'),
+            self.stock_location, self.product_2, 20.00
+        )
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        with self.assertRaises(ValidationError):
+            self.product_2.active = False
+
+    def test_archive_product_with_stock_move_line_in(self):
+        self._create_stock_move_line(
+            self.env.ref('stock.stock_location_suppliers'),
+            self.stock_location, self.product_2, 20.00
+        )
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        with self.assertRaises(ValidationError):
+            self.product_2.active = False
+
+    def test_archive_product_with_stock_picking_in(self):
+        self._create_stock_picking(
+            self.env.ref('stock.stock_location_suppliers'),
+            self.stock_location, self.product_2, 20.00
+        )
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        with self.assertRaises(ValidationError):
+            self.product_2.active = False
+
+    def test_archive_product_with_stock_picking_in_out(self):
+        self._create_stock_picking(
+            self.env.ref('stock.stock_location_suppliers'),
+            self.stock_location, self.product_2, 20.00
+        )
+        self._create_stock_picking(
+            self.stock_location,
+            self.env.ref('stock.stock_location_customers'), self.product_2, 20.00
+        )
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        self.product_2.active = False
+        self.assertFalse(self.product_2.active)
+
+    def test_archive_product_stock_location(self):
+        self._create_stock_inventory(self.stock_location, self.product_2, 20.00)
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        with self.assertRaises(ValidationError):
+            self.product_2.active = False
+
+    def test_archive_product_stock_location_child(self):
+        self._create_stock_inventory(self.stock_location_child, self.product_2, 20.00)
+        self.product_1.active = False
+        self.assertFalse(self.product_1.active)
+        with self.assertRaises(ValidationError):
+            self.product_2.active = False
+
+    def test_archive_unarchive_stock_location(self):
+        self.stock_location.active = False
+        self.assertFalse(self.stock_location.active)
+        self.stock_location.active = True
+        self.assertTrue(self.stock_location.active)
+
+    def test_archive_stock_location_ok(self):
+        self.stock_location.active = False
+        self.assertFalse(self.stock_location.active)
+
+    def test_archive_stock_location(self):
+        self._create_stock_inventory(self.stock_location, self.product_2, 20.00)
+        with self.assertRaises(ValidationError):
+            self.stock_location.active = False
+
+    def test_archive_unarchive_stock_location_child(self):
+        self.stock_location_child.active = False
+        self.assertFalse(self.stock_location_child.active)
+        self.stock_location_child.active = True
+        self.assertTrue(self.stock_location_child.active)
+
+    def test_archive_stock_location_child(self):
+        self._create_stock_inventory(self.stock_location_child, self.product_2, 20.00)
+        with self.assertRaises(ValidationError):
+            self.stock_location.active = False