shopfloor_dangerous_goods: fix tests

OCA · Feb 24, 2023 · 64933f3 · 64933f3
1 parent 17598a1
commit 64933f3
Showing 1 changed file with 17 additions and 20 deletions.
diff --git a/shopfloor_dangerous_goods/tests/test_actions_data.py b/shopfloor_dangerous_goods/tests/test_actions_data.py
@@ -19,40 +19,37 @@ def setUpClass(cls):
         cls._set_product_lq()
 
     def test_data_stock_move_line(self):
-        move_line = self.move_a.move_line_ids
+        move_line = self.move_a.move_line_ids[0]
         result_package = self.env["stock.quant.package"].create(
-            {"packaging_id": self.packaging.id}
+            {"packaging_id": self.packaging.id, "pack_weight": 10}
         )
+        # make weight stable
+        move_line.package_id.pack_weight = 10
         move_line.write({"qty_done": 3.0, "result_package_id": result_package.id})
-        data = self.data.move_line(move_line)
-        self.assert_schema(self.schema.move_line(), data)
+        package_src_data = self.data.package(
+            move_line.package_id, picking=self.move_a.picking_id
+        )
+        self.assertTrue(package_src_data["has_lq_products"])
+        package_dest_data = self.data.package(
+            move_line.result_package_id, picking=self.move_a.picking_id
+        )
+        self.assertFalse(package_dest_data["has_lq_products"])
         expected = {
             "id": move_line.id,
             "qty_done": 3.0,
+            "progress": 30.0,
             "quantity": move_line.product_uom_qty,
             "product": self._expected_product(self.product_a),
             "lot": None,
-            "package_src": {
-                "id": move_line.package_id.id,
-                "name": move_line.package_id.name,
-                "move_line_count": 1,
-                "weight": 20.0,
-                "storage_type": None,
-                "has_lq_products": True,
-            },
-            "package_dest": {
-                "id": result_package.id,
-                "name": result_package.name,
-                "move_line_count": 0,
-                "weight": 6.0,
-                "storage_type": None,
-                "has_lq_products": False,
-            },
+            "package_src": package_src_data,
+            "package_dest": package_dest_data,
             "location_src": self._expected_location(move_line.location_id),
             "location_dest": self._expected_location(move_line.location_dest_id),
             "priority": "1",
             "has_lq_products": True,
         }
+        data = self.data.move_line(move_line)
+        self.assert_schema(self.schema.move_line(), data)
         self.assertDictEqual(data, expected)
 
     def test_data_package(self):