[IMP] fieldservice_stage_validation: Add test

fieldservice_stage_validation/tests/__init__.py

@@ -0,0 +1,4 @@
+# Copyright (C) 2020, Brian McMaster <brian@mcmpest.com>
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from . import test_fsm_stage_validation

fieldservice_stage_validation/tests/test_fsm_stage_validation.py

@@ -0,0 +1,189 @@
+# Copyright 2020, Brian McMaster <brian@mcmpest.com>
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html)
+
+from odoo.exceptions import ValidationError
+from odoo.tests import SavepointCase
+
+
+class TestFSMStageValidation(SavepointCase):
+    @classmethod
+    def setUpClass(cls):
+        super(TestFSMStageValidation, cls).setUpClass()
+
+        cls.stage = cls.env["fsm.stage"]
+        cls.fsm_order = cls.env["fsm.order"]
+        cls.fsm_person = cls.env["fsm.person"]
+        cls.fsm_location = cls.env["fsm.location"]
+        cls.fsm_equipment = cls.env["fsm.equipment"]
+        cls.ir_model_fields = cls.env["ir.model.fields"]
+
+        # Get some fields to use in the stages
+        cls.order_field = cls.ir_model_fields.search(
+            [("model", "=", "fsm.order"), ("name", "=", "description")]
+        )
+        cls.person_field = cls.ir_model_fields.search(
+            [("model", "=", "fsm.person"), ("name", "=", "mobile")]
+        )
+        cls.location_field = cls.ir_model_fields.search(
+            [("model", "=", "fsm.location"), ("name", "=", "direction")]
+        )
+        cls.equipment_field = cls.ir_model_fields.search(
+            [("model", "=", "fsm.equipment"), ("name", "=", "notes")]
+        )
+
+        # For each model type, create a default stage and a stage
+        # which will apply field validation
+        # Order Stages
+        cls.stage_order_default = cls.stage.create(
+            {
+                "name": "Order Stage Default",
+                "stage_type": "order",
+                "is_default": True,
+                "sequence": "1",
+            }
+        )
+        cls.stage_order = cls.stage.create(
+            {
+                "name": "Order Stage Validate",
+                "stage_type": "order",
+                "validate_field_ids": [(6, 0, [cls.order_field.id])],
+                "sequence": "2",
+            }
+        )
+        # Person Stages
+        cls.stage_person_default = cls.stage.create(
+            {
+                "name": "Person Stage Default",
+                "stage_type": "worker",
+                "is_default": True,
+                "sequence": "1",
+            }
+        )
+        cls.stage_person = cls.stage.create(
+            {
+                "name": "Person Stage Validate",
+                "stage_type": "worker",
+                "validate_field_ids": [(6, 0, [cls.person_field.id])],
+                "sequence": "2",
+            }
+        )
+        # Location Stages
+        cls.stage_location_default = cls.stage.create(
+            {
+                "name": "Location Stage Default",
+                "stage_type": "location",
+                "is_default": True,
+                "sequence": "1",
+            }
+        )
+        cls.stage_location = cls.stage.create(
+            {
+                "name": "Location Stage Validate",
+                "stage_type": "location",
+                "validate_field_ids": [(6, 0, [cls.location_field.id])],
+                "sequence": "2",
+            }
+        )
+        # Equipment Stages
+        cls.stage_equipment_default = cls.stage.create(
+            {
+                "name": "Equipment Stage Default",
+                "stage_type": "equipment",
+                "is_default": True,
+                "sequence": "1",
+            }
+        )
+        cls.stage_equipment = cls.stage.create(
+            {
+                "name": "Equipment Stage Validate",
+                "stage_type": "equipment",
+                "validate_field_ids": [(6, 0, [cls.equipment_field.id])],
+                "sequence": "2",
+            }
+        )
+
+        # Create a person
+        cls.person_01 = cls.fsm_person.create(
+            {
+                "name": "FSM Worker 01",
+                "partner_id": cls.env["res.partner"]
+                .create({"name": "Worker 01 Partner"})
+                .id,
+            }
+        )
+        # Create a location
+        cls.location_01 = cls.fsm_location.create(
+            {
+                "name": "Location 01",
+                "owner_id": cls.env["res.partner"]
+                .create({"name": "Location 01 Partner"})
+                .id,
+            }
+        )
+        # Create an Equipment
+        cls.equipment_01 = cls.fsm_equipment.create(
+            {"name": "Equipment 01", "current_location_id": cls.location_01.id}
+        )
+        # Create an Order
+        cls.order_01 = cls.fsm_order.create({"location_id": cls.location_01.id})
+
+    def test_fsm_stage_validation(self):
+
+        # Validate the stage computes the correct model type
+        self.assertEqual(
+            self.stage_order.stage_type_model_id,
+            self.fsm_order,
+            "FSM Stage model is not computed correctly",
+        )
+
+        # Validate the Equipment cannot move to next stage
+        with self.assertRaises(ValidationError):
+            self.equipment_01.next_stage()
+
+        # Update the Equipment notes field and validate it goes to next stage
+        self.equipment_01.notes = "Equipment service note"
+        self.equipment_01.next_stage()
+        self.assertEqual(
+            self.equipment_01.stage_id,
+            self.stage_equipment,
+            "FSM Equipment did not progress to correct stage",
+        )
+
+        # Validate the Location cannot move to next stage
+        with self.assertRaises(ValidationError):
+            self.location_01.next_stage()
+
+        # Update the Location directions field and validate it goes to next stage
+        self.location_01.direction = "Location direction note"
+        self.location_01.next_stage()
+        self.assertEqual(
+            self.location_01.stage_id,
+            self.stage_location,
+            "FSM Location did not progress to correct stage",
+        )
+
+        # Validate the Person cannot move to next stage
+        with self.assertRaises(ValidationError):
+            self.person_01.next_stage()
+
+        # Update the Person mobile field and validate it goes to next stage
+        self.person_01.mobile = "1-888-888-8888"
+        self.person_01.next_stage()
+        self.assertEqual(
+            self.person_01.stage_id,
+            self.stage_person,
+            "FSM Person did not progress to correct stage",
+        )
+
+        # Validate the Order cannot move to next stage
+        with self.assertRaises(ValidationError):
+            self.order_01.write({"stage_id": self.stage_order.id})
+
+        # Update the Order description field and validate it goes to next stage
+        self.order_01.description = "Complete the work order"
+        self.order_01.write({"stage_id": self.stage_order.id})
+        self.assertEqual(
+            self.order_01.stage_id,
+            self.stage_order,
+            "FSM Order did not progress to correct stage",
+        )