Scrap reason code module

scrap_reason_code/tests/test_scrap_reason_code.py

@@ -0,0 +1,80 @@
+# Copyright (C) 2019 IBM Corp.
+# Copyright (C) 2019 Open Source Integrators
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo.tests.common import TransactionCase
+
+
+class StockScrap(TransactionCase):
+    def setUp(self):
+        super(StockScrap, self).setUp()
+
+        self.stock_location = self.env.ref('stock.stock_location_stock')
+        self.customer_location = self.env.ref('stock.stock_location_customers')
+        self.scrapped_location = self.env.ref('stock.stock_location_scrapped')
+
+        self.scrap_product = self.env['product.product'].create({
+            'name': 'Scrap Product A',
+            'type': 'product',
+            'categ_id': self.env.ref('product.product_category_all').id,
+        })
+
+        self.reason_code = self.env['reason.code'].create({
+            'name': 'DM300',
+            'description': 'Product is damage',
+            'location_id': self.scrapped_location.id
+        })
+
+        self.uom_unit = self.env.ref('product.product_uom_unit')
+
+    def test_scrap_reason_code(self):
+        """ Scrap the product of a picking. Then modify the
+        done linked stock move and ensure the scrap quantity is also
+        updated and verify scrap reason code
+        """
+        self.env['stock.quant']._update_available_quantity(self.scrap_product,
+                                                           self.stock_location,
+                                                           10)
+        partner = self.env['res.partner'].create({'name': 'BOdedra'})
+        picking = self.env['stock.picking'].create({
+            'name': 'A single picking with one move to scrap',
+            'location_id': self.stock_location.id,
+            'location_dest_id': self.customer_location.id,
+            'partner_id': partner.id,
+            'picking_type_id': self.env.ref('stock.picking_type_out').id,
+        })
+        move1 = self.env['stock.move'].create({
+            'name': 'A move to confirm and scrap its product',
+            'location_id': self.stock_location.id,
+            'location_dest_id': self.customer_location.id,
+            'product_id': self.scrap_product.id,
+            'product_uom': self.uom_unit.id,
+            'product_uom_qty': 1.0,
+            'picking_id': picking.id,
+        })
+        move1._action_confirm()
+
+        self.assertEqual(move1.state, 'confirmed')
+        scrap = self.env['stock.scrap'].create({
+            'product_id': self.scrap_product.id,
+            'product_uom_id': self.scrap_product.uom_id.id,
+            'scrap_qty': 5,
+            'picking_id': picking.id,
+            'reason_code_id': self.reason_code.id
+        })
+        scrap._onchange_reason_code_id()
+        scrap.do_scrap()
+        self.assertEqual(len(picking.move_lines), 2)
+        scrapped_move = picking.move_lines.filtered(
+            lambda m: m.state == 'done')
+        self.assertTrue(scrapped_move, 'No scrapped move created.')
+        self.assertEqual(scrapped_move.scrap_ids.ids, [scrap.id],
+                         'Wrong scrap linked to the move.')
+        self.assertEqual(scrap.scrap_qty, 5,
+                         'Scrap quantity has been modified and is not '
+                         'correct anymore.')
+        move = scrap.move_id
+        self.assertEqual(move.reason_code_id.id, self.reason_code.id)
+
+        scrapped_move.quantity_done = 8
+        self.assertEqual(scrap.scrap_qty, 8, 'Scrap quantity is not updated.')